Update app.py

app.py

@@ -3,8 +3,22 @@ import os
 from groq import Groq
 import gradio as gr
 from gtts import gTTS
+import speech_recognition as sr
 
-client = Groq(api_key=os.getenv("groqkey"))
+client = Groq(api_key=os.getenv("GROQ_API_KEY"))
+
+# Speech-to-text function
+def speech_to_text(audio_file):
+    recognizer = sr.Recognizer()
+    with sr.AudioFile(audio_file) as source:
+        audio = recognizer.record(source)
+    try:
+        text = recognizer.recognize_google(audio)
+        return text
+    except sr.UnknownValueError:
+        return "Could not understand audio"
+    except sr.RequestError:
+        return "Could not request results"
 
 def get_chatbot_response(user_message, country, language, conversation_history):
     system_message = (
@@ -154,6 +168,17 @@ with gr.Blocks(theme=theme, css=custom_css) as demo:
         liability_btn = gr.Button("⚖️ Liability", elem_classes="insurance-button")
         pet_btn = gr.Button("🐾 Pet", elem_classes="insurance-button")
 
+    # Add a dropdown for input method (text or voice)
+    input_method = gr.Dropdown(
+        ["Text", "Voice"],
+        label="🎤 Input Method",
+        value="Text",
+        interactive=True
+    )
+
+    # Add a microphone input for voice
+    voice_input = gr.Audio(source="microphone", type="filepath", label="🎤 Speak your query", visible=False)
+
     with gr.Row(equal_height=True):
         scenario_input = gr.Textbox(
             label="💡 Type your message...", 
@@ -176,24 +201,53 @@ with gr.Blocks(theme=theme, css=custom_css) as demo:
     ):
         btn.click(lambda current, insurance=insurance: update_insurance_selection(current, insurance), inputs=scenario_input, outputs=scenario_input)
 
-    def submit(country, custom_country, language, scenario, conversation_state):
-        selected_country = custom_country if country == "Other" else country
-        updated_history, chat_display = get_chatbot_response(
-            scenario, selected_country, language, conversation_state
-        )
-        return updated_history, chat_display, ""
+    # Function to handle input method selection
+    def handle_input_method(input_method, scenario_input, voice_input):
+        if input_method == "Voice":
+            return gr.update(visible=False), gr.update(visible=True)
+        else:
+            return gr.update(visible=True), gr.update(visible=False)
+
+    input_method.change(
+        handle_input_method,
+        inputs=[input_method, scenario_input, voice_input],
+        outputs=[scenario_input, voice_input]
+    )
+
+    # Function to process voice input
+    def process_voice_input(voice_file, country, custom_country, language, conversation_state):
+        if voice_file:
+            user_message = speech_to_text(voice_file)
+            selected_country = custom_country if country == "Other" else country
+            updated_history, chat_display = get_chatbot_response(
+                user_message, selected_country, language, conversation_state
+            )
+            return updated_history, chat_display, ""
+        else:
+            return conversation_state, [], "No audio input detected."
+
+    # Submit function for both text and voice input
+    def submit(country, custom_country, language, scenario, conversation_state, input_method, voice_input):
+        if input_method == "Text":
+            selected_country = custom_country if country == "Other" else country
+            updated_history, chat_display = get_chatbot_response(
+                scenario, selected_country, language, conversation_state
+            )
+            return updated_history, chat_display, ""
+        else:
+            return process_voice_input(voice_input, country, custom_country, language, conversation_state)
 
     country_input.change(lambda c: gr.update(visible=c == "Other"), inputs=country_input, outputs=custom_country_input)
 
     submit_btn.click(
         submit, 
-        inputs=[country_input, custom_country_input, language_input, scenario_input, conversation_state], 
+        inputs=[country_input, custom_country_input, language_input, scenario_input, conversation_state, input_method, voice_input], 
         outputs=[conversation_state, chatbot, scenario_input]
     )
 
     scenario_input.submit(
         fn=submit, 
-        inputs=[country_input, custom_country_input, language_input, scenario_input, conversation_state], 
+        inputs=[country_input, custom_country_input, language_input, scenario_input, conversation_state, input_method, voice_input], 
         outputs=[conversation_state, chatbot, scenario_input]
     )