Update app.py

app.py

@@ -3,14 +3,15 @@ import gradio as gr
 import whisper
 import google.generativeai as genai
 
-# Configure Gemini API
+# Configure Gemini
 genai.configure(
-    api_key=os.getenv("GEMINI_API_KEYS")
+    api_key=os.getenv("GEMINI_API_KEY")
 )
 
-# Load models
+# Load Whisper model
 speech_model = whisper.load_model("tiny")
 
+# Load Gemini model
 ai_model = genai.GenerativeModel(
     "gemini-1.5-flash"
 )
@@ -18,20 +19,31 @@ ai_model = genai.GenerativeModel(
 # Main function
 def voice_assistant(audio):
 
-    # Convert speech to text
-    result = speech_model.transcribe(audio)
+    # Check if audio exists
+    if audio is None:
+        return "Please record your voice first."
 
-    user_text = result["text"]
+    try:
+        # Convert speech to text
+        result = speech_model.transcribe(audio)
 
-    # Generate AI response
-    response = ai_model.generate_content(user_text)
+        user_text = result["text"]
 
-    return f"You said: {user_text}\n\nAI: {response.text}"
+        # Gemini response
+        response = ai_model.generate_content(user_text)
 
-# Gradio interface
+        return f"You said: {user_text}\n\nAI: {response.text}"
+
+    except Exception as e:
+        return f"Error: {str(e)}"
+
+# Gradio UI
 interface = gr.Interface(
     fn=voice_assistant,
-    inputs=gr.Audio(type="filepath"),
+    inputs=gr.Audio(
+        sources=["microphone"],
+        type="filepath"
+    ),
     outputs="text",
     title="AI Voice Assistant",
     description="Speak and get AI responses"