Update app.py

app.py

@@ -16,17 +16,21 @@ model.to(device)
 st.title("🔥 Live Fire Detection with Alarm 🔥")
 st.subheader("Hold the camera towards potential fire sources to detect in real-time.")
 
-# Load alarm sound from a direct MP3 URL
-alarm_url = "https://docs.google.com/uc?export=download&id=16IzsnQDmWkfYSeb_AjOTx79NEgkOpz88"  # Replace with a working direct MP3 URL
+# Load alarm sound (must be a direct MP3 URL)
+alarm_url = "https://www.soundjay.com/button/beep-07.wav"  # Replace with your hosted alarm
 
-# JavaScript to control alarm playback
+# JavaScript to auto-play alarm when fire is detected
 js_code = f"""
 <script>
     var alarm = new Audio("{alarm_url}");
+    alarm.loop = true;
+
     function playAlarm() {{
-        alarm.loop = true;
-        alarm.play();
+        alarm.play().catch(error => {{
+            console.log("Autoplay failed. User interaction required.");
+        }});
     }}
+
     function stopAlarm() {{
         alarm.pause();
         alarm.currentTime = 0;
@@ -34,24 +38,29 @@ js_code = f"""
 </script>
 """
 
-# Inject JavaScript once
+# Inject JavaScript
 components.html(js_code, height=0)
 
 # Capture live camera input
 image = camera_input_live()
 
 if image is not None:
-    # Convert image to OpenCV format
+    # Convert the image to OpenCV format
     bytes_data = image.getvalue()
     cv2_img = cv2.imdecode(np.frombuffer(bytes_data, np.uint8), cv2.IMREAD_COLOR)
 
     # Perform fire detection
     results = model(cv2_img)
+
+    fire_present = False  # Flag for fire detection
     
     # Check if fire is detected
-    fire_present = any(len(result.boxes) > 0 for result in results)
+    for result in results:
+        if len(result.boxes) > 0:
+            fire_present = True
+            break  # No need to check further
 
-    # Show logs in UI & trigger alarm
+    # Display logs & trigger alarm
     if fire_present:
         st.error("🔥 Fire Detected! 🔥")
         components.html("<script>playAlarm();</script>", height=0)