Update app.py

app.py

@@ -8,13 +8,18 @@ from gtts import gTTS
 import threading
 from datetime import datetime, timedelta
 
-# Check if running in a cloud environment (e.g., Hugging Face Spaces)
+# Detect if running in a cloud environment (e.g., Hugging Face Spaces)
 running_in_cloud = "HF_HOME" in os.environ
 
 if not running_in_cloud:
-    import pygame
-    pygame.mixer.quit()  # Ensure the mixer is fully stopped
-    pygame.mixer.init()
+    try:
+        import pygame
+        pygame.mixer.quit()  # Ensure the mixer is fully stopped
+        pygame.mixer.init()
+    except ImportError:
+        print("pygame module not found, skipping audio initialization.")
+else:
+    print("Running in cloud mode: Skipping pygame initialization.")
 
 # Load YOLOv8 model
 yolo = YOLO("yolov8n.pt")
@@ -190,4 +195,4 @@ if start_detection:
                 pygame.mixer.quit()
 
 elif stop_detection:
-    st.warning("Object detection stopped.")
+    st.warning("Object detection stopped.")