Update utils.py

utils.py

@@ -1,70 +1,139 @@
+"""
+utils.py - GPS, Maps, and Statistics helpers
+"""
+
+import os, json
+from pathlib import Path
+from datetime import datetime
+from PIL import ExifTags
 import exifread
-import requests
-
-def dms_to_decimal(dms, ref):
-    """Convert [deg, min, sec] to decimal degrees."""
-    d, m, s = [float(x) for x in dms]
-    decimal = d + m / 60.0 + s / 3600.0
-    if ref in ["S", "W"]:
-        decimal = -decimal
-    return decimal
-
-def reverse_geocode(lat, lon):
-    """Convert coordinates into human-readable address using OpenStreetMap Nominatim."""
-    try:
-        url = f"https://nominatim.openstreetmap.org/reverse"
-        params = {"lat": lat, "lon": lon, "format": "json", "zoom": 18, "addressdetails": 1}
-        headers = {"User-Agent": "roadside-clean-app"}
-        resp = requests.get(url, params=params, headers=headers, timeout=5)
-        if resp.status_code == 200:
-            data = resp.json()
-            return data.get("display_name")
-    except Exception as e:
-        print(f"⚠️ Reverse geocoding failed: {e}")
-    return None
+import folium
+from folium import plugins
+from config import HISTORY_FILE
+
 
+# ---------------- GPS Extraction ----------------
 def extract_gps_from_image(image):
-    """
-    Extract GPS coordinates and human-readable address from EXIF metadata.
-    Tries both PIL and exifread for Xiaomi/Android phones.
-    """
-    # --- First try PIL ---
+    """Try GPS extraction via PIL, fallback to exifread"""
+    # 1️⃣ Try PIL
     try:
-        exifdata = image.getexif()
-        if exifdata and "GPSInfo" in exifdata:
-            gps_info = {}
-            for t, val in exifdata["GPSInfo"].items():
-                gps_info[GPSTAGS.get(t, t)] = val
-            lat = gps_info.get("GPSLatitude")
-            lon = gps_info.get("GPSLongitude")
+        exif = image.getexif()
+        gps_info = {}
+        for tag_id, val in exif.items():
+            tag = ExifTags.TAGS.get(tag_id, tag_id)
+            if tag == "GPSInfo":
+                gps_info = {ExifTags.GPSTAGS.get(t, t): v for t, v in val.items()}
+
+        if gps_info:
+            lat = _to_decimal(gps_info.get("GPSLatitude"), gps_info.get("GPSLatitudeRef"))
+            lon = _to_decimal(gps_info.get("GPSLongitude"), gps_info.get("GPSLongitudeRef"))
             if lat and lon:
-                lat = dms_to_decimal(lat, gps_info.get("GPSLatitudeRef", "N"))
-                lon = dms_to_decimal(lon, gps_info.get("GPSLongitudeRef", "E"))
-                return {"coords": (lat, lon), "address": reverse_geocode(lat, lon)}
-    except Exception as e:
-        print(f"⚠️ PIL GPS parse failed: {e}")
+                return {"coords": (lat, lon), "address": None}
+    except:
+        pass
 
-    # --- Fallback: use exifread ---
+    # 2️⃣ Fallback: exifread
     try:
         import io
+        import numpy as np
+        from PIL import Image
+
         buf = io.BytesIO()
         image.save(buf, format="JPEG")
         buf.seek(0)
+
         tags = exifread.process_file(buf, details=False)
 
-        lat = tags.get("GPS GPSLatitude")
-        lon = tags.get("GPS GPSLongitude")
-        lat_ref = tags.get("GPS GPSLatitudeRef")
-        lon_ref = tags.get("GPS GPSLongitudeRef")
-
-        if lat and lon and lat_ref and lon_ref:
-            # Convert to [d, m, s]
-            lat_vals = [x.num / x.den for x in lat.values]
-            lon_vals = [x.num / x.den for x in lon.values]
-            lat_dd = dms_to_decimal(lat_vals, str(lat_ref))
-            lon_dd = dms_to_decimal(lon_vals, str(lon_ref))
-            return {"coords": (lat_dd, lon_dd), "address": reverse_geocode(lat_dd, lon_dd)}
+        lat = _to_decimal(tags.get("GPS GPSLatitude"), tags.get("GPS GPSLatitudeRef"))
+        lon = _to_decimal(tags.get("GPS GPSLongitude"), tags.get("GPS GPSLongitudeRef"))
+
+        if lat and lon:
+            return {"coords": (lat, lon), "address": None}
     except Exception as e:
-        print(f"⚠️ exifread GPS parse failed: {e}")
+        print(f"⚠️ exifread failed: {e}")
 
     return {"coords": None, "address": None}
+
+
+def _to_decimal(dms, ref):
+    """Convert GPS dms to decimal"""
+    if not dms:
+        return None
+    try:
+        d, m, s = [float(x.num) / float(x.den) if hasattr(x, "num") else float(x) for x in dms]
+        decimal = d + (m / 60.0) + (s / 3600.0)
+        if isinstance(ref, str) and ref in ["S", "W"]:
+            decimal = -decimal
+        return decimal
+    except Exception as e:
+        print(f"⚠️ GPS decode failed: {e}")
+        return None
+
+
+# ---------------- History ----------------
+def save_detection_to_history(detection):
+    Path("data").mkdir(exist_ok=True)
+    history = []
+    if os.path.exists(HISTORY_FILE):
+        with open(HISTORY_FILE, "r") as f:
+            history = json.load(f)
+
+    history.append(detection)
+    if len(history) > 100:
+        history = history[-100:]
+
+    with open(HISTORY_FILE, "w") as f:
+        json.dump(history, f, indent=2)
+
+
+def load_detection_history():
+    if os.path.exists(HISTORY_FILE):
+        with open(HISTORY_FILE, "r") as f:
+            return json.load(f)
+    return []
+
+
+# ---------------- Map ----------------
+def create_detection_map(detections):
+    if not detections:
+        return "<p>🗺️ No GPS data yet</p>"
+
+    center = detections[-1]["gps"]["coords"] if detections[-1]["gps"]["coords"] else (41.015, 28.979)  # Istanbul default
+    m = folium.Map(location=center, zoom_start=12)
+    plugins.Fullscreen().add_to(m)
+
+    for d in detections:
+        gps = d.get("gps", {})
+        if gps and gps.get("coords"):
+            lat, lon = gps["coords"]
+            sev = d.get("severity", "LOW")
+            color = {"HIGH": "red", "MEDIUM": "orange", "LOW": "yellow"}.get(sev, "blue")
+            popup = f"<b>{sev}</b> | {d['count']} items"
+            folium.Marker([lat, lon], popup=popup, icon=folium.Icon(color=color)).add_to(m)
+
+    return m._repr_html_()
+
+
+# ---------------- Stats ----------------
+def generate_statistics():
+    history = load_detection_history()
+    if not history:
+        return {"total": 0, "items": 0, "high": 0, "medium": 0, "low": 0}
+
+    return {
+        "total": len(history),
+        "items": sum(h["count"] for h in history),
+        "high": sum(1 for h in history if h["severity"] == "HIGH"),
+        "medium": sum(1 for h in history if h["severity"] == "MEDIUM"),
+        "low": sum(1 for h in history if h["severity"] == "LOW"),
+    }
+
+
+def format_statistics_text(stats):
+    if stats["total"] == 0:
+        return "📊 No detections yet"
+    return (
+        f"📊 Reports: {stats['total']}\n"
+        f"📦 Items: {stats['items']}\n"
+        f"🔴 High: {stats['high']} | 🟠 Medium: {stats['medium']} | 🟡 Low: {stats['low']}"
+    )