Update app.py

app.py

@@ -43,6 +43,7 @@ try:
     from services.plantation.missing_patch_check import process_missing_patches
 except ImportError as e:
     print(f"Failed to import service modules: {str(e)}")
+    logging.error(f"Import error: {str(e)}")
     exit(1)
 
 # Configure logging
@@ -79,15 +80,17 @@ os.makedirs(OUTPUT_DIR, exist_ok=True)
 def initialize_video(video_file: Optional[Any] = None) -> str:
     global video_loaded, log_entries
     release_video()
-    video_path = DEFAULT_VIDEO_PATH
+    video_path = DEFAULT_VIDEO_PATH if video_file is None else video_file.name
+
+    if not os.path.exists(video_path):
+        status = f"Error: Video file '{video_path}' not found."
+        log_entries.append(status)
+        logging.error(status)
+        video_loaded = False
+        return status
 
-    if video_file is not None:
-        video_path = video_file.name
-        log_entries.append(f"Using uploaded video: {video_path}")
-        logging.info(f"Using uploaded video: {video_path}")
-    
     try:
-        preload_video()
+        preload_video(video_path)
         video_loaded = True
         status = f"Successfully loaded video: {video_path}"
         log_entries.append(status)
@@ -133,35 +136,50 @@ def set_active_service(
     logging.info("No service category enabled.")
     return None, "No Service Category Enabled"
 
-def generate_line_chart() -> Optional[str]:
+def generate_line_chart():
     if not crack_counts:
         return None
-    fig, ax = plt.subplots(figsize=(4, 2))
-    ax.plot(crack_counts[-50:], marker='o', color='#4682B4')
-    ax.set_title("Cracks/Holes Over Time")
-    ax.set_xlabel("Frame")
-    ax.set_ylabel("Count")
-    ax.grid(True)
-    fig.tight_layout()
-    chart_path = "chart_temp.png"
-    fig.savefig(chart_path)
-    plt.close(fig)
-    return chart_path
-
-def generate_pie_chart() -> Optional[str]:
-    if not crack_severity_all:
-        return None
-    fig, ax = plt.subplots(figsize=(4, 2))
-    count = Counter(crack_severity_all[-200:])
-    labels, sizes = zip(*count.items())
-    colors = ['#FF6347', '#FFA500', '#32CD32', '#800080']
-    ax.pie(sizes, labels=labels, autopct='%1.1f%%', startangle=140, colors=colors[:len(labels)])
-    ax.axis('equal')
-    fig.tight_layout()
-    pie_path = "pie_temp.png"
-    fig.savefig(pie_path)
-    plt.close(fig)
-    return pie_path
+    ```chartjs
+    {
+        "type": "line",
+        "data": {
+            "labels": [i for i in range(len(crack_counts[-50:]))],
+            "datasets": [{
+                "label": crack_counts[-50:],
+                "data": crack_counts[-50:],
+                "borderColor": "#4682B4",
+                "backgroundColor": "#4682B4",
+                "fill": false,
+                "pointBackgroundColor": "#3CB371",
+                "pointRadius": 5
+            }]
+        },
+        "options": {
+            "responsive": false,
+            "scales": {
+                "x": {
+                    "title": {
+                        "display": true,
+                        "text": "Frame"
+                    }
+                },
+                "y": {
+                    "title": {
+                        "display": true,
+                        "text": "Count of Cracks/Holes"
+                    }
+                }
+            },
+            "title": {
+                "display": true,
+                "text": "Cracks/Holes Over Time"
+            }
+        }
+    }
+    ```
+
+def generate_map(gps_coordinates: List[List[float]], items: List[Dict]]) -> Optional[str]:
+    return generate_map(gps_coordinates, items)
 
 def monitor_feed() -> Tuple[
     Optional[np.ndarray],
@@ -170,13 +188,19 @@ def monitor_feed() -> Tuple[
     List[str],
     List[str],
     Optional[str],
-    Optional[str]
-]:
-    global paused, frame_count, last_frame, last_metrics, last_timestamp
-    global gps_coordinates, last_detected_cracks, last_detected_holes, video_loaded
+    global
+    paused,
+    frame_count,
+    last_frame,
+    last_metrics,
+    last_timestamp,
+    gps_coordinates,
+    last_detected_cracks,
+    last_detected_holes,
+    video_loaded
 
     if not video_loaded:
-        log_entries.append("Cannot start streaming: Video not loaded successfully.")
+        log_entries.append(("Cannot start streaming: Video not loaded successfully.")
         logging.error("Video not loaded successfully.")
         return (
             None,
@@ -241,6 +265,7 @@ def monitor_feed() -> Tuple[
                 all_detected_items.extend(generic_dets)
 
             # Apply shadow detection
+            cv2.imwrite(TEMP_IMAGE_PATH, frame)
             shadow_issue = detect_shadow_coverage(TEMP_IMAGE_PATH)
 
             # Apply thermal processing if frame is grayscale
@@ -373,7 +398,7 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", secondary_hue="green"))
 
     with gr.Row():
         with gr.Column(scale=3):
-            video_input = gr.File(label="Upload Video File (e.g., sample.mp4)", file_types=["video"])
+            video_input = gr.File(label="Upload Video File (e.g., sample.mp4)", file_types=[".mp4", ".avi"])
             load_button = gr.Button("Load Video", variant="primary")
         with gr.Column(scale=1):
             video_status = gr.Textbox(