Update app.py

app.py

@@ -1,6 +1,26 @@
+"""
+app.py - Hugging Face Gradio App
+"""
+
+import gradio as gr
+from datetime import datetime
+from detector import RoadsideDetector
+from utils import (
+    extract_gps_from_image,
+    save_detection_to_history,
+    load_detection_history,
+    create_detection_map,
+    generate_statistics,
+    format_statistics_text,
+)
+
+
+detector = RoadsideDetector()
+
+
 def process_image(image, use_gps=True):
     if image is None:
-        return None, "❌ Please upload an image!", None, "📊 No statistics yet"
+        return None, "❌ Please upload an image!", "<p>No map</p>", "📊 No statistics yet"
 
     detections = detector.detect_items(image)
     annotated = detector.draw_boxes(image, detections)
@@ -13,7 +33,7 @@ def process_image(image, use_gps=True):
         "severity": severity,
         "count": len(detections),
         "types": [d["type"] for d in detections],
-        "gps": gps_data["coords"],
+        "gps": gps_data,
     }
     save_detection_to_history(detection_data)
 
@@ -21,12 +41,52 @@ def process_image(image, use_gps=True):
     if gps_data["coords"]:
         summary += f"\n📍 Location: {gps_data['coords'][0]:.6f}, {gps_data['coords'][1]:.6f}"
     elif gps_data["address"]:
-        summary += f"\n📍 Address: {gps_data['address']}"
+        summary += f"\n📍 Location: {gps_data['address']}"
     else:
         summary += "\n📍 No GPS/location data available"
 
     history = load_detection_history()
-    map_html = create_detection_map([d for d in history if d.get("gps")])
+    map_html = create_detection_map(history)
     stats = format_statistics_text(generate_statistics())
 
     return annotated, summary, map_html, stats
+
+
+def create_interface():
+    with gr.Blocks(title="🚮 RoadsideClean") as demo:
+        gr.Markdown("# 🚮 RoadsideClean - Illegal Dumping Detector")
+
+        with gr.Tabs():
+            with gr.TabItem("📸 Detect"):
+                with gr.Row():
+                    with gr.Column():
+                        input_img = gr.Image(
+                            type="pil",
+                            label="📷 Take/Upload Photo",
+                            sources=["upload", "webcam"],
+                        )
+                        gps_toggle = gr.Checkbox("📍 Extract GPS", value=True)
+                        detect_btn = gr.Button("🔍 Detect", variant="primary")
+
+                    with gr.Column():
+                        out_img = gr.Image(label="Detection Results", type="pil")
+                        summary = gr.Textbox(label="Summary", lines=6)
+
+            with gr.TabItem("🗺️ Map"):
+                map_display = gr.HTML("<p>🗺️ No detections yet</p>")
+
+            with gr.TabItem("📊 Statistics"):
+                stats_display = gr.Markdown("📊 No statistics yet")
+
+        detect_btn.click(
+            process_image,
+            inputs=[input_img, gps_toggle],
+            outputs=[out_img, summary, map_display, stats_display],
+        )
+
+    return demo
+
+
+if __name__ == "__main__":
+    app = create_interface()
+    app.launch(server_name="0.0.0.0", server_port=7860, ssr_mode=False)