Update app.py

app.py

@@ -6,7 +6,7 @@ from gtts import gTTS  # type: ignore
 from pydub import AudioSegment
 from pydub.playback import play
 
-# Dog sound files (Ensure these files exist)
+# Dog sound files (Ensure these files exist in the same directory)
 dog_sounds = {
     "sit": "dog_sit.mp3",
     "come": "dog_come.mp3",
@@ -40,11 +40,16 @@ def dog_response(command):
     if command:
         for key in dog_sounds:
             if key in command:
-                play_dog_sound(dog_sounds[key])
-                return key, f"Playing sound for {key}", generate_speech(f"Woof! I heard you say {key}")
-        play_dog_sound(dog_sounds["bark"])
-        return "bark", "No specific dog command recognized. Playing default bark sound.", generate_speech("Woof! I didn't recognize that, so I'll just bark!")
-    return "unknown", "No command to process.", None
+                sound_file = dog_sounds[key]
+                play_dog_sound(sound_file)
+                return sound_file, f"Playing sound for {key}", generate_speech(f"Woof! I heard you say {key}")
+        
+        # If no specific command is recognized, play bark
+        sound_file = dog_sounds["bark"]
+        play_dog_sound(sound_file)
+        return sound_file, "No specific dog command recognized. Playing default bark sound.", generate_speech("Woof! I didn't recognize that, so I'll just bark!")
+    
+    return None, "No command to process.", None
 
 def play_dog_sound(sound_file):
     """Plays an audio file using Pygame."""
@@ -64,25 +69,21 @@ def generate_speech(text):
     return speech_file  # Return audio file for Gradio
 
 def process_command(audio_file):
+    """Processes the command from the user."""
     command = recognize_speech(audio_file)
-    keyword, response_text, speech_file = dog_response(command)
-    return command, response_text, speech_file
+    sound_file, response_text, speech_file = dog_response(command)
+    
+    # Return the dog sound file in the first output
+    return sound_file, response_text, speech_file
 
 # Gradio UI
 iface = gr.Interface(
     fn=process_command,
     inputs=gr.Audio(sources=["microphone"], type="filepath"),
-    outputs=["text", "text", "audio"],  # Added audio output
+    outputs=["audio", "text", "audio"],  # First box now plays dog sound
     title="🐶 Dog Command Recognition 🐶",
     description="🎤 Speak a command and let the dog respond! 🐕\n\nTry commands like 'sit', 'come', 'fetch', 'treat', 'play'",
-    theme="default",
     live=True,
-    css="""
-    body { background-color: #f8f9fa; text-align: center; }
-    .output-text { color: #ff4500; font-size: 20px; font-weight: bold; }
-    .interface-title { color: #008080; font-size: 26px; font-weight: bold; }
-    .interface-description { color: #2f4f4f; font-size: 18px; }
-    """
 )
 
 if __name__ == "__main__":