Update app.py

app.py

@@ -5,21 +5,24 @@ import cv2
 import random
 import time
 from gtts import gTTS
+import pygame
 import threading
 from datetime import datetime, timedelta
 
-# Detect if running in a cloud environment (e.g., Hugging Face Spaces)
-running_in_cloud = "HF_HOME" in os.environ
+# Check if running in a headless environment
+is_headless = os.getenv("DISPLAY") is None
 
-if not running_in_cloud:
+# Initialize pygame mixer if not in headless mode
+if not is_headless:
     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.")
+    except pygame.error as e:
+        print(f"Pygame mixer initialization failed: {e}")
+        pygame.mixer = None
 else:
-    print("Running in cloud mode: Skipping pygame initialization.")
+    pygame.mixer = None
+    print("Headless environment detected. Audio playback disabled.")
 
 # Load YOLOv8 model
 yolo = YOLO("yolov8n.pt")
@@ -51,7 +54,7 @@ st.markdown(
 )
 
 # Display welcome image
-welcome_image_path = "bismillah.png"  # Ensure this image exists in the script's directory
+welcome_image_path = "bismillah.png"
 if os.path.exists(welcome_image_path):
     st.image(welcome_image_path, use_container_width=True, caption="Bismillah hir Rehman Ar Raheem")
 else:
@@ -74,7 +77,7 @@ with col1:
     start_detection = st.button("Start Detection")
 with col2:
     stop_detection = st.button("Stop Detection")
-audio_activation = st.checkbox("Enable Audio Alerts", value=False) if not running_in_cloud else False
+audio_activation = st.checkbox("Enable Audio Alerts", value=False)
 
 # Categories for audio alerts (hazardous objects or living things)
 alert_categories = {"person", "cat", "dog", "knife", "fire", "gun"}
@@ -86,9 +89,8 @@ alert_cooldown = timedelta(seconds=10)  # 10-second cooldown for alerts
 
 def play_audio_alert(label, position):
     """Generate and play an audio alert."""
-    if running_in_cloud:
-        print(f"Audio alert triggered for {label} on {position}. (Audio playback disabled in cloud)")
-        return
+    if pygame.mixer is None:
+        return  # Skip audio playback in headless mode
 
     phrases = [
         f"Be careful, there's a {label} on your {position}.",
@@ -191,7 +193,7 @@ if start_detection:
         if 'video_capture' in locals() and video_capture.isOpened():
             video_capture.release()
             cv2.destroyAllWindows()
-            if not running_in_cloud:
+            if pygame.mixer:
                 pygame.mixer.quit()
 
 elif stop_detection: