Update app.py

app.py

@@ -15,7 +15,7 @@ pipe = pipeline("text-generation", model=model_name, device="cpu")
 system_prompt = {"role": "system", "content": "You are an AI assistant. Your task is to generate a detailed and comprehensive response."}
 
 # Define the Gradio interface
-def generate_response(history, user_input, temperature, max_new_tokens):
+def generate_response(history, user_input, temperature, max_new_tokens, language):
     messages = [system_prompt] + history + [{"role": "user", "content": user_input}]
     response = pipe(messages, max_new_tokens=max_new_tokens, temperature=temperature, return_full_text=False)
     generated_text = response[0]['generated_text']
@@ -23,7 +23,7 @@ def generate_response(history, user_input, temperature, max_new_tokens):
     history.append({"role": "assistant", "content": generated_text})
 
     # Generate speech
-    tts = gTTS(text=generated_text, lang='en')
+    tts = gTTS(text=generated_text, lang=language)
     audio_path = "response.mp3"
     tts.save(audio_path)
 
@@ -88,9 +88,10 @@ with gr.Blocks(css=custom_css) as demo:
     with gr.Accordion("Advanced Settings", open=False, elem_id="settings"):
         temperature_slider = gr.Slider(0, 1, 0.7, step=0.1, label="Temperature")
         max_new_tokens_slider = gr.Slider(1, 1000, 100, step=1, label="Max New Tokens")
+        language_dropdown = gr.Dropdown(choices=["en", "ru"], value="en", label="Speech Language")
 
-    send_btn.click(generate_response, [chatbot, txt, temperature_slider, max_new_tokens_slider], [chatbot, audio_output])
-    txt.submit(generate_response, [chatbot, txt, temperature_slider, max_new_tokens_slider], [chatbot, audio_output])
+    send_btn.click(generate_response, [chatbot, txt, temperature_slider, max_new_tokens_slider, language_dropdown], [chatbot, audio_output])
+    txt.submit(generate_response, [chatbot, txt, temperature_slider, max_new_tokens_slider, language_dropdown], [chatbot, audio_output])
     clear_btn.click(clear_chat, outputs=[chatbot, txt])
 
 # Launch the interface