Update app.py

app.py

@@ -29,55 +29,46 @@ from utils import (
     get_debug_messages,   # ✅ make sure this is imported
 )
 
-def process_image(image, use_gps=True):
-    if image is None:
+from PIL import Image
+
+def process_image(image_path, use_gps=True):  # Now we get a filepath, not a PIL image
+    if not image_path:
         return None, "❌ Please upload an image!", None, "📊 No statistics yet", ""
 
+    # Load image for YOLO detection
+    image = Image.open(image_path)
+
     # Run YOLO detection
     detections = detector.detect_items(image)
     annotated = detector.draw_boxes(image, detections)
     severity = detector.calculate_severity(detections)
 
-    # ---------------- GPS Extraction ----------------
+    # GPS extraction from original file
     gps_data = {"coords": None, "address": None}
     if use_gps:
         try:
-            if hasattr(image, "name"):  # file path
-                gps_data = extract_gps_from_image(image.name)
-            else:  # fallback: pass the PIL image
-                gps_data = extract_gps_from_image(image)
+            gps_data = extract_gps_from_image(image_path)  # ✅ pass file path
         except Exception as e:
             add_debug(f"❌ GPS extraction failed: {e}")
 
-    # ---------------- Save Detection ----------------
-    detection_data = {
-        "timestamp": datetime.now().strftime("%Y-%m-%d %H:%M"),
-        "severity": severity,
+    # Save to history
+    detection_entry = {
+        "timestamp": datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
         "count": len(detections),
-        "types": [d["type"] for d in detections],
+        "severity": severity,
         "gps": gps_data,
     }
-    save_detection_to_history(detection_data)
-
-    # ---------------- Summary ----------------
-    summary = f"🔍 Severity: {severity}\n📦 Items: {len(detections)}"
-    if gps_data["coords"]:
-        lat, lon = gps_data["coords"]
-        summary += f"\n📍 Location: {lat:.6f}, {lon:.6f}"
-    elif gps_data["address"]:
-        summary += f"\n📍 Location: {gps_data['address']}"
-    else:
-        summary += "\n📍 No GPS/location data available"
-
-    # ---------------- Map + Stats ----------------
+    save_detection_to_history(detection_entry)
+
+    # Build map + stats
     history = load_detection_history()
     map_html = create_detection_map(history)
-    stats = format_statistics_text(generate_statistics())
+    stats = generate_statistics()
+    stats_text = format_statistics_text(stats)
+    debug_log = get_debug_messages()
 
-    # ---------------- Debug Logs ----------------
-    debug_output = get_debug_messages()
+    return annotated, f"🔍 Severity: {severity}\n📦 Items: {len(detections)}", map_html, stats_text, debug_log
 
-    return annotated, summary, map_html, stats, debug_output  # ✅ now returns 5 values
 
 
 
@@ -91,10 +82,12 @@ def create_interface():
                 with gr.Row():
                     with gr.Column():
                         input_img = gr.Image(
-                            type="pil",
+                            type="filepath",
                             label="📷 Take/Upload Photo",
                             sources=["upload", "webcam"],
                         )
+                
+
                         gps_toggle = gr.Checkbox("📍 Extract GPS", value=True)
                         detect_btn = gr.Button("🔍 Detect", variant="primary")