Update app.py

app.py

@@ -40,7 +40,12 @@ def monitor_feed():
         frame = last_frame.copy()
         metrics = last_metrics.copy()
     else:
-        frame = get_next_video_frame()
+        try:
+            frame = get_next_video_frame()
+        except RuntimeError as e:
+            log_entries.append(f"Error: {str(e)}")
+            return None, last_metrics, "\n".join(log_entries[-10:]), None, None, last_detected_images, None
+
         detected_boxes = detect_cracks(frame)
         frame = overlay_boxes(frame, detected_boxes)
         cv2.imwrite(TEMP_IMAGE_PATH, frame, [int(cv2.IMWRITE_JPEG_QUALITY), 95])
@@ -48,7 +53,7 @@ def monitor_feed():
 
         frame_count += 1
         last_timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-        gps_coord = simulate_gps_coordinates(frame_count)  # Simulate GPS
+        gps_coord = simulate_gps_coordinates(frame_count)
         gps_coordinates.append(gps_coord)
 
         if detected_boxes:
@@ -89,6 +94,8 @@ def monitor_feed():
 
 # Line chart
 def generate_line_chart():
+    if not crack_counts:
+        return None
     fig, ax = plt.subplots(figsize=(4, 2))
     ax.plot(crack_counts[-50:], marker='o')
     ax.set_title("Cracks Over Time")
@@ -162,7 +169,10 @@ with gr.Blocks(theme=gr.themes.Soft()) as app:
     def streaming_loop():
         while True:
             frame, metrics, logs, chart, pie, captured, map_path = monitor_feed()
-            yield frame, str(metrics), logs, chart, pie, captured, map_path
+            if frame is None:
+                yield None, str(metrics), logs, chart, pie, captured, map_path
+            else:
+                yield frame, str(metrics), logs, chart, pie, captured, map_path
             time.sleep(frame_rate)
 
     app.load(streaming_loop, outputs=[video_output, metrics_output, logs_output, chart_output, pie_output, captured_images, map_output])