Update app.py

app.py

@@ -19,6 +19,9 @@ import io
 detector = RoadsideDetector()
 
 
+from PIL import Image
+from datetime import datetime
+from detector import RoadsideDetector
 from utils import (
     extract_gps_from_image,
     save_detection_to_history,
@@ -26,28 +29,29 @@ from utils import (
     create_detection_map,
     generate_statistics,
     format_statistics_text,
-    get_debug_messages,   # ✅ make sure this is imported
+    get_debug_messages,
+    add_debug,   # ✅ include this
 )
 
-from PIL import Image
+detector = RoadsideDetector()
 
-def process_image(image_path, use_gps=True):  # Now we get a filepath, not a PIL image
+def process_image(image_path, use_gps=True):
     if not image_path:
         return None, "❌ Please upload an image!", None, "📊 No statistics yet", ""
 
-    # Load image for YOLO detection
+    # Load image for YOLO
     image = Image.open(image_path)
 
-    # Run YOLO detection
+    # YOLO detection
     detections = detector.detect_items(image)
     annotated = detector.draw_boxes(image, detections)
     severity = detector.calculate_severity(detections)
 
-    # GPS extraction from original file
+    # GPS extraction
     gps_data = {"coords": None, "address": None}
     if use_gps:
         try:
-            gps_data = extract_gps_from_image(image_path)  # ✅ pass file path
+            gps_data = extract_gps_from_image(image_path)  # ✅ pass path
         except Exception as e:
             add_debug(f"❌ GPS extraction failed: {e}")
 
@@ -60,14 +64,21 @@ def process_image(image_path, use_gps=True):  # Now we get a filepath, not a PIL
     }
     save_detection_to_history(detection_entry)
 
-    # Build map + stats
+    # Map + stats
     history = load_detection_history()
     map_html = create_detection_map(history)
     stats = generate_statistics()
     stats_text = format_statistics_text(stats)
     debug_log = get_debug_messages()
 
-    return annotated, f"🔍 Severity: {severity}\n📦 Items: {len(detections)}", map_html, stats_text, debug_log
+    return (
+        annotated, 
+        f"🔍 Severity: {severity}\n📦 Items: {len(detections)}"
+        + (f"\n📍 GPS: {gps_data['coords']}" if gps_data.get("coords") else "\n📍 No GPS/location data available"),
+        map_html,
+        stats_text,
+        debug_log
+    )