Update app.py

app.py

@@ -1,6 +1,7 @@
 import speech_recognition as sr  # type: ignore
 import os
 import pygame  # type: ignore
+import gradio as gr
 from gtts import gTTS  # type: ignore
 from pydub import AudioSegment
 from pydub.playback import play
@@ -17,87 +18,79 @@ dog_sounds = {
 
 # Initialize speech recognizer
 recognizer = sr.Recognizer()
-
-# Initialize pygame mixer
 pygame.mixer.init()
 
-
 def recognize_speech():
     """Recognizes speech from the microphone input."""
     with sr.Microphone() as source:
-        print("Listening for command...")
         try:
             audio = recognizer.listen(source, timeout=5)
             command = recognizer.recognize_google(audio)
-            print(f"Human said: {command}")
             return command.lower()
         except sr.UnknownValueError:
-            print("Sorry, I could not understand your speech.")
-            return None
+            return "Sorry, I could not understand your speech."
         except sr.RequestError:
-            print("Sorry, the speech service is unavailable.")
-            return None
+            return "Sorry, the speech service is unavailable."
         except sr.WaitTimeoutError:
-            print("No speech detected. Please try again.")
-            return None
-
+            return "No speech detected. Please try again."
 
 def dog_response(command):
     """Plays the corresponding dog sound based on the recognized command."""
     if command:
-        # Check if command matches known keywords
         for key in dog_sounds:
             if key in command:
                 play_dog_sound(dog_sounds[key])
-                return
-        
-        # Default to bark sound if command is unknown
-        print("No specific dog command recognized. Playing default bark sound.")
+                return f"Playing sound for {key}"
         play_dog_sound(dog_sounds["bark"])
-    else:
-        print("No command to process.")
-
+        return "No specific dog command recognized. Playing default bark sound."
+    return "No command to process."
 
 def play_dog_sound(sound_file):
     """Plays an audio file using Pygame."""
     if os.path.exists(sound_file):
         pygame.mixer.music.load(sound_file)
         pygame.mixer.music.play()
-        while pygame.mixer.music.get_busy():  # Wait until sound finishes playing
+        while pygame.mixer.music.get_busy():
             continue
     else:
-        print(f"Error: Sound file '{sound_file}' not found.")
+        return f"Error: Sound file '{sound_file}' not found."
 
-
-def make_dog_response(text):
+def make_dog_response(command):
     """Generates a playful dog response using text-to-speech."""
     try:
-        tts = gTTS(text=text, lang='en')
+        tts = gTTS(text=f"Woof! I heard you say {command}", lang='en')
         response_file = "dog_response.mp3"
         tts.save(response_file)
-        
         pygame.mixer.music.load(response_file)
         pygame.mixer.music.play()
         while pygame.mixer.music.get_busy():
             continue
-
-        os.remove(response_file)  # Clean up the temporary file
+        os.remove(response_file)
+        return f"Generated response: Woof! I heard you say {command}"
     except Exception as e:
-        print(f"Error generating speech: {e}")
-
-
-def main():
-    """Main loop to listen for speech commands and respond accordingly."""
-    print("Say 'exit' or 'quit' to stop.")
-    while True:
-        command = recognize_speech()
-        if command in ["exit", "quit"]:
-            print("Exiting program. Goodbye!")
-            break
-
-        dog_response(command)  # Process command
-        make_dog_response(f"Woof! I heard you say {command}.")  # Playful dog response
-
+        return f"Error generating speech: {e}"
+
+def process_command():
+    command = recognize_speech()
+    response = dog_response(command)
+    tts_response = make_dog_response(command)
+    return command, response, tts_response
+
+iface = gr.Interface(
+    fn=process_command,
+    inputs=[],
+    outputs=["text", "text", "text"],
+    title="🐶 Dog Command Recognition 🐶",
+    description="🎤 Speak a command and let the dog respond! 🐕\n\nTry commands like 'sit', 'come', 'fetch', 'treat', 'play', or anything else!",
+    theme="default",
+    live=True,
+    css="""
+    body { background-color: #f0f8ff; }
+    .output-text { color: #ff4500; font-size: 18px; font-weight: bold; }
+    .interface-title { color: #008080; font-size: 24px; font-weight: bold; }
+    .interface-description { color: #2f4f4f; font-size: 16px; }
+    """
+)
 
 if __name__ == "__main__":
-    main()
+    iface.launch()