Update app.py

app.py

@@ -1,9 +1,12 @@
 import gradio as gr
-from deep_translator import GoogleTranslator
 import nltk
 from nltk.tokenize import word_tokenize
 from nltk.corpus import stopwords
 from nltk.stem import WordNetLemmatizer
+from deep_translator import GoogleTranslator
+from gtts import gTTS
+import tempfile
+import os
 
 # Download necessary NLTK data
 nltk.download('punkt', quiet=True)
@@ -21,39 +24,70 @@ def translate_text(text, target_language):
     translator = GoogleTranslator(source='auto', target=target_language)
     return translator.translate(text)
 
-def process_input(input_text, feature, target_language):
+def text_to_speech(text):
+    tts = gTTS(text)
+    with tempfile.NamedTemporaryFile(delete=False, suffix=".mp3") as fp:
+        tts.save(fp.name)
+        return fp.name
+
+def process_input(input_text, input_audio, feature, target_language, output_language):
+    if input_audio is not None:
+        # For now, we'll just return a message as we can't process audio input
+        return "Audio input processing is not available in this version.", None
+
     if not input_text:
-        return "No input provided"
+        return "No input provided", None
 
     processed_text = natural_language_understanding(input_text)
 
     if feature == "Translation":
         result = translate_text(processed_text, target_language)
+    elif feature == "Voice Command":
+        result = "Voice command feature is not implemented in this version"
     elif feature == "Transcription":
         result = processed_text
     else:
         result = "Invalid feature selected"
 
-    return result
+    if output_language:
+        result = translate_text(result, output_language)
+
+    return result, None
+
+def tts_function(text):
+    if text:
+        return text_to_speech(text)
+    return None
 
 # Create Gradio interface
 with gr.Blocks() as demo:
     gr.Markdown("# The Advanced Multi-Faceted Chatbot")
-    gr.Markdown("Enter text to interact with the chatbot. Choose a feature and specify language for translation if needed.")
+    gr.Markdown("Enter text to interact with the chatbot. Choose a feature and specify languages for translation if needed.")
     
-    input_text = gr.Textbox(label="Input Text")
+    with gr.Row():
+        input_text = gr.Textbox(label="Input Text")
+        input_audio = gr.Audio(label="Input Audio", type="filepath", visible=False)  # Hidden for now
     
     with gr.Row():
-        feature = gr.Radio(["Translation", "Transcription"], label="Feature")
+        feature = gr.Radio(["Translation", "Voice Command", "Transcription"], label="Feature")
         target_language = gr.Textbox(label="Target Language (e.g., 'fr' for French)")
+        output_language = gr.Textbox(label="Output Language (e.g., 'es' for Spanish)")
     
     submit_button = gr.Button("Process")
     result_text = gr.Textbox(label="Result")
+    tts_button = gr.Button("Convert to Speech")
+    audio_output = gr.Audio(label="Audio Output")
 
     submit_button.click(
         process_input,
-        inputs=[input_text, feature, target_language],
-        outputs=result_text
+        inputs=[input_text, input_audio, feature, target_language, output_language],
+        outputs=[result_text, audio_output]
+    )
+    
+    tts_button.click(
+        tts_function,
+        inputs=[result_text],
+        outputs=[audio_output]
     )
 
 # Launch the interface