Update app.py

app.py

@@ -5,21 +5,18 @@ from gtts import gTTS
 from groq import Groq
 
 GROQ_API_KEY = os.getenv("GORQ_API_KEY")
-client = Groq(api_key= GROQ_API_KEY)
+client = Groq(api_key=GROQ_API_KEY)
 
 # Load Whisper model
 model = whisper.load_model("base")
 
-def chatbot(audio=None, text_input=""):
-    if audio is None and not text_input.strip():
-        return "No input detected. Please provide either an audio or text input.", None
+def chatbot(audio=None):
+    if audio is None:
+        return "No input detected. Please provide an audio input.", None
 
-    if audio:
-        # Transcribe the audio input using Whisper
-        transcription = model.transcribe(audio)
-        user_input = transcription.get("text", "")
-    else:
-        user_input = text_input.strip()
+    # Transcribe the audio input using Whisper
+    transcription = model.transcribe(audio)
+    user_input = transcription.get("text", "")
 
     if not user_input:
         return "Could not understand input.", None
@@ -42,27 +39,25 @@ def chatbot(audio=None, text_input=""):
 
 # Create a custom interface
 def build_interface():
-    with gr.Blocks() as demo:
+    with gr.Blocks(css=".gradio-container {background-color: #F5F5F5;}") as demo:
         gr.Markdown(
             """
             <h1 style="text-align: center; color: #000000;">VoiceToVoice Chatbot</h1>
             """
         )
         with gr.Row():
-            with gr.Column(scale=1):
-                audio_input = gr.Audio(type="filepath", label="Record Your Voice")
-            with gr.Column(scale=2):
-                text_input = gr.Textbox(label="Enter your prompt")
-                chatbot_output_text = gr.Textbox(label="Chatbot Response")
-                chatbot_output_audio = gr.Audio(label="Audio Response")
+            audio_input = gr.Audio(type="filepath", label="Record Your Voice")
+            chatbot_output_text = gr.Textbox(label="Chatbot Response")
+            chatbot_output_audio = gr.Audio(label="Audio Response")
 
-        submit_button = gr.Button("Submit")
+        # Add a clear button
+        clear_button = gr.Button("Clear")
 
-        submit_button.click(
-            fn=chatbot,
-            inputs=[audio_input, text_input],
-            outputs=[chatbot_output_text, chatbot_output_audio]
-        )
+        # Handle the clear button click
+        clear_button.click(lambda: None, None, [audio_input, chatbot_output_text, chatbot_output_audio])
+
+        # Make it real-time
+        audio_input.change(fn=chatbot, inputs=audio_input, outputs=[chatbot_output_text, chatbot_output_audio])
 
     return demo