Update app.py

app.py

@@ -21,6 +21,9 @@ def get_llm_response(prompt):
     return response.choices[0].message.content.strip()
 
 def process_voice_or_text(input_audio, input_text, output_lang):
+    if output_lang not in ["ta", "en"]:
+        return "Invalid output language selected. Please choose either Tamil ('ta') or English ('en').", "", "", ""
+
     if input_audio is not None:
         # Process audio input
         recognizer = sr.Recognizer()
@@ -47,6 +50,18 @@ def process_voice_or_text(input_audio, input_text, output_lang):
 
     return input_text, english_query, llm_response, final_response
 
+
+    # Translate input to English
+    english_query = translate_text(input_text, "en")
+
+    # Get response from LLM
+    llm_response = get_llm_response(english_query)
+
+    # Translate LLM response to desired output language
+    final_response = translate_text(llm_response, output_lang)
+
+    return input_text, english_query, llm_response, final_response
+
 def text_to_speech(response, lang):
     tts = gTTS(text=response, lang=lang)
     tts.save("response.mp3")