Update app.py

app.py

@@ -1,9 +1,8 @@
-
 import cv2
 import numpy as np
 import torch
 import streamlit as st
-import os
+import streamlit.components.v1 as components
 from ultralytics import YOLO
 from camera_input_live import camera_input_live
 
@@ -14,31 +13,33 @@ model = YOLO(model_path)
 model.to(device)
 
 # Streamlit app title
-st.title("Live Fire Detection with Camera")
+st.title("🔥 Live Fire Detection with Alarm 🔥")
 st.subheader("Hold the camera towards potential fire sources to detect in real-time.")
 
-# Capture live camera input
-image = camera_input_live()
-fire_detected = False  # Track fire detection state
+# Load alarm sound from a public URL
+alarm_url = "https://huggingface.co/spaces/Prathamesh1420/Fire_detection_streamlit/blob/main/alarm.mp3"  # Replace with your hosted alarm
 
-# HTML & JS for Alarm Sound
-alarm_html = """
-<audio id="fireAlarm" src="alarm.mp3"></audio>
+# JavaScript to control alarm playback
+js_code = f"""
 <script>
-    function playAlarm() {
-        var alarm = document.getElementById("fireAlarm");
-        if (alarm.paused) {
-            alarm.play();
-        }
-    }
-    function stopAlarm() {
-        var alarm = document.getElementById("fireAlarm");
+    var alarm = new Audio("{alarm_url}");
+    function playAlarm() {{
+        alarm.loop = true;
+        alarm.play();
+    }}
+    function stopAlarm() {{
         alarm.pause();
         alarm.currentTime = 0;
-    }
+    }}
 </script>
 """
 
+# Display the 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
     bytes_data = image.getvalue()
@@ -47,7 +48,7 @@ if image is not None:
     # Perform fire detection
     results = model(cv2_img)
 
-    fire_present = False  # Temporary flag for fire detection in this frame
+    fire_present = False  # Flag for fire detection
     
     # Draw bounding boxes for detected fires
     for result in results:
@@ -57,18 +58,15 @@ if image is not None:
             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  # Fire detected
+            fire_present = True
 
     # Display the annotated image
     st.image(cv2_img, channels="BGR", caption="Detected Fire", use_container_width=True)
     
-    # Display logs and trigger alarm
+    # Display logs & trigger alarm using JavaScript
     if fire_present:
         st.error("🔥 Fire Detected! 🔥")
-        st.markdown(alarm_html + "<script>playAlarm();</script>", unsafe_allow_html=True)
-        fire_detected = True
+        components.html("<script>playAlarm();</script>", height=0)
     else:
         st.success("✅ No Fire Detected")
-        if fire_detected:
-            st.markdown(alarm_html + "<script>stopAlarm();</script>", unsafe_allow_html=True)
-        fire_detected = False
+        components.html("<script>stopAlarm();</script>", height=0)