Update app.py

app.py

@@ -16,8 +16,8 @@ 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 public URL
-alarm_url = "https://soundcloud.com/trending-music-in/sets/soul?utm_source=clipboard&utm_medium=text&utm_campaign=social_sharing"  # Replace with your hosted alarm
+# Load alarm sound from a direct MP3 URL
+alarm_url = "https://www.soundjay.com/button/beep-07.wav"  # Replace with a working direct MP3 URL
 
 # JavaScript to control alarm playback
 js_code = f"""
@@ -34,36 +34,24 @@ js_code = f"""
 </script>
 """
 
-# Display the JavaScript once
+# Inject JavaScript once
 components.html(js_code, height=0)
 
 # Capture live camera input
 image = camera_input_live()
 
 if image is not None:
-    # Convert the image to OpenCV format
+    # Convert 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
     
-    # Draw bounding boxes for detected fires
-    for result in results:
-        boxes = result.boxes
-        for box in boxes:
-            b = box.xyxy[0].cpu().numpy().astype(int)
-            label = f'Fire {box.conf[0]:.2f}'
-            cv2.rectangle(cv2_img, (b[0], b[1]), (b[2], b[3]), (0, 0, 255), 3)
-            cv2.putText(cv2_img, label, (b[0], b[1] - 10), cv2.FONT_HERSHEY_SIMPLEX, 0.9, (0, 0, 255), 2)
-            fire_present = True
+    # Check if fire is detected
+    fire_present = any(len(result.boxes) > 0 for result in results)
 
-    # Display the annotated image
-    st.image(cv2_img, channels="BGR", caption="Detected Fire", use_container_width=True)
-    
-    # Display logs & trigger alarm using JavaScript
+    # Show logs in UI & trigger alarm
     if fire_present:
         st.error("🔥 Fire Detected! 🔥")
         components.html("<script>playAlarm();</script>", height=0)