Update app.py

app.py

@@ -61,7 +61,7 @@ with tabs[0]:
 with tabs[1]:
     col1, col2 = st.columns([1, 1])
     with col1:
-        st.markdown("**Webcam snapshot or stream**")
+        st.markdown("**Webcam snapshot**")
         webcam_url = st.text_input("URL", "http://<your_webcam_ip>/current.jpg", label_visibility="collapsed")
         confidence = st.slider("Confidence", 0.25, 1.0, 0.4, key="webcam_conf")
         mode = st.radio("", ["Snapshot", "Stream"], label_visibility="collapsed")
@@ -69,28 +69,50 @@ with tabs[1]:
     with col2:
         if start:
             if mode == "Snapshot":
-                placeholder = st.empty()
+                image_placeholder = st.empty()
                 timer_placeholder = st.empty()
-                refresh_interval = 5  # seconds
+                refresh_interval = 30  # Refresh every 30 seconds
                 while True:
                     start_time = time.time()
                     try:
+                        # Fetch the webcam image
                         response = requests.get(webcam_url, timeout=5)
+                        if response.status_code != 200:
+                            st.error(f"Failed to fetch image: HTTP {response.status_code}")
+                            break
                         image_array = np.asarray(bytearray(response.content), dtype=np.uint8)
                         frame = cv2.imdecode(image_array, cv2.IMREAD_COLOR)
+                        if frame is None:
+                            st.error("Failed to decode image from URL.")
+                            break
+
+                        # Check for fire/smoke using the YOLO model
                         results = model.predict(frame, conf=confidence)
-                        detected_frame = results[0].plot()[:, :, ::-1]
-                        placeholder.image(detected_frame, use_column_width=True)
+                        detected_frame = results[0].plot()[:, :, ::-1]  # Convert BGR to RGB for display
+
+                        # Display the processed image
+                        image_placeholder.image(detected_frame, use_column_width=True)
+
+                        # Update timer
                         elapsed = time.time() - start_time
                         remaining = max(0, refresh_interval - elapsed)
                         timer_placeholder.write(f"Next refresh in: {int(remaining)}s")
-                        time.sleep(1)
-                        if remaining <= 0:
-                            st.experimental_rerun()
+
+                        # Wait until 30 seconds have passed, updating the timer
+                        while remaining > 0:
+                            time.sleep(1)
+                            elapsed = time.time() - start_time
+                            remaining = max(0, refresh_interval - elapsed)
+                            timer_placeholder.write(f"Next refresh in: {int(remaining)}s")
+
+                        # Refresh the page to fetch a new image
+                        st.experimental_rerun()
+
                     except Exception as e:
                         st.error(f"Error: {e}")
                         break
             else:
+                # Stream mode (unchanged from previous)
                 cap = cv2.VideoCapture(webcam_url)
                 frame_placeholder = st.empty()
                 while cap.isOpened():