Update app.py

app.py

@@ -121,7 +121,7 @@ with tabs[1]:
     with col1:
         st.markdown("**Webcam Feed**")
         st.write("Provide a webcam URL (image or video stream) to monitor for hazards.")
-        webcam_url = st.text_input("Webcam URL", "http://<your_webcam_ip>/current.jpg", label_visibility="collapsed")
+        webcam_url = st.text_input("Webcam URL", "https://wms-prod-3.wetmet.net/live/281-05-01/playlist.m3u8?wmsAuthSign=c2VydmVyX3RpbWU9Mi8yOC8yMDI1IDk6MDE6MTcgUE0maGFzaF92YWx1ZT1jR0JLZnJYbmFVOEdYbDh1M0wyc1FnPT0mdmFsaWRtaW51dGVzPTMw", label_visibility="collapsed")
         confidence = st.slider("Detection Threshold", 0.25, 1.0, 0.4, key="webcam_conf")
         refresh_rate = st.slider("Refresh Rate (seconds)", 1, 60, 30, key="webcam_rate")
         start = st.button("Begin Monitoring", key="webcam_start")
@@ -140,21 +140,44 @@ with tabs[1]:
         timer_placeholder = st.empty()
         
         if st.session_state.monitoring and st.session_state.current_webcam_url:
-            # Try video stream first
+            # Try as video stream first
             cap = cv2.VideoCapture(webcam_url)
             is_video_stream = cap.isOpened()
             
-            while st.session_state.monitoring:
-                try:
-                    start_time = time.time()
-                    if is_video_stream:
+            if is_video_stream:
+                status_placeholder.write("Connected to video stream...")
+                while st.session_state.monitoring:
+                    try:
+                        start_time = time.time()
                         ret, frame = cap.read()
                         if not ret:
                             status_placeholder.error("Video stream interrupted.")
                             break
-                    else:
+                        results = model.predict(frame, conf=confidence)
+                        detected_frame = results[0].plot()[:, :, ::-1]
+                        frame_placeholder.image(detected_frame, use_column_width=True)
+                        status_placeholder.write(f"Objects detected: {len(results[0].boxes)}")
+                        
+                        elapsed = time.time() - start_time
+                        remaining = max(0, refresh_rate - elapsed)
+                        timer_placeholder.write(f"Next scan: {int(remaining)}s")
+                        time.sleep(0.1)
+                    except Exception as e:
+                        status_placeholder.error(f"Video error: {e}")
+                        st.session_state.monitoring = False
+                        break
+                cap.release()
+            else:
+                # Fallback to image-based fetch
+                status_placeholder.write("Trying as image-based webcam...")
+                while st.session_state.monitoring:
+                    try:
+                        start_time = time.time()
                         response = requests.get(webcam_url, timeout=5)
-                        if response.status_code != 200:
+                        if response.status_code == 403:
+                            status_placeholder.error("HTTP 403: Access denied. Token may have expired.")
+                            break
+                        elif response.status_code != 200:
                             status_placeholder.error(f"Fetch failed: HTTP {response.status_code}")
                             break
                         image_array = np.asarray(bytearray(response.content), dtype=np.uint8)
@@ -163,27 +186,19 @@ with tabs[1]:
                             status_placeholder.error("Image decoding failed.")
                             break
 
-                    results = model.predict(frame, conf=confidence)
-                    detected_frame = results[0].plot()[:, :, ::-1]
-                    frame_placeholder.image(detected_frame, use_column_width=True)
-                    status_placeholder.write(f"Objects detected: {len(results[0].boxes)}")
-                    
-                    elapsed = time.time() - start_time
-                    remaining = max(0, refresh_rate - elapsed)
-                    timer_placeholder.write(f"Next scan: {int(remaining)}s")
-                    
-                    if not is_video_stream:
+                        results = model.predict(frame, conf=confidence)
+                        detected_frame = results[0].plot()[:, :, ::-1]
+                        frame_placeholder.image(detected_frame, use_column_width=True)
+                        status_placeholder.write(f"Objects detected: {len(results[0].boxes)}")
+                        
+                        elapsed = time.time() - start_time
+                        remaining = max(0, refresh_rate - elapsed)
+                        timer_placeholder.write(f"Next scan: {int(remaining)}s")
                         time.sleep(remaining)
-                    else:
-                        time.sleep(0.1)
-                    
-                except Exception as e:
-                    status_placeholder.error(f"Error: {e}")
-                    st.session_state.monitoring = False
-                    break
-            
-            if is_video_stream:
-                cap.release()
+                    except Exception as e:
+                        status_placeholder.error(f"Image fetch error: {e}")
+                        st.session_state.monitoring = False
+                        break
 
 # Tab 3: YouTube
 with tabs[2]: