Update app.py

app.py

@@ -9,9 +9,16 @@ import pygame
 import threading
 from datetime import datetime, timedelta
 
-# Initialize pygame mixer
-pygame.mixer.quit()  # Ensure the mixer is fully stopped
-pygame.mixer.init()
+# Check if running in a cloud environment (e.g., Hugging Face Spaces)
+running_in_cloud = "HF_HOME" in os.environ
+
+if running_in_cloud:
+    os.environ["SDL_AUDIODRIVER"] = "dummy"  # Use the dummy audio driver
+    print("Running in a cloud environment: Audio alerts disabled.")
+else:
+    # Initialize pygame mixer
+    pygame.mixer.quit()  # Ensure the mixer is fully stopped
+    pygame.mixer.init()
 
 # Load YOLOv8 model
 yolo = YOLO("yolov8n.pt")
@@ -42,15 +49,6 @@ st.markdown(
     unsafe_allow_html=True,
 )
 
-# Create a container for the buttons
-st.markdown('<div class="button-container">', unsafe_allow_html=True)
-
-# # Render buttons inside the container
-# start_detection = st.button("Start Detection", key="start")
-# stop_detection = st.button("Stop Detection", key="stop")
-
-st.markdown('</div>', unsafe_allow_html=True)
-
 # Display welcome image
 welcome_image_path = "bismillah.png"  # Ensure this image exists in the script's directory
 if os.path.exists(welcome_image_path):
@@ -75,7 +73,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)
+audio_activation = st.checkbox("Enable Audio Alerts", value=False) if not running_in_cloud else False
 
 # Categories for audio alerts (hazardous objects or living things)
 alert_categories = {"person", "cat", "dog", "knife", "fire", "gun"}
@@ -87,6 +85,10 @@ 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
+
     phrases = [
         f"Be careful, there's a {label} on your {position}.",
         f"Watch out! {label} detected on your {position}.",
@@ -188,7 +190,8 @@ if start_detection:
         if 'video_capture' in locals() and video_capture.isOpened():
             video_capture.release()
             cv2.destroyAllWindows()
-            pygame.mixer.quit()
+            if not running_in_cloud:
+                pygame.mixer.quit()
 
 elif stop_detection:
     st.warning("Object detection stopped.")