Update app.py

app.py

@@ -2,21 +2,18 @@ from ultralytics import YOLO
 import gradio as gr
 import tempfile, os, shutil, glob
 
-model = YOLO("best.pt")  # make sure model file name matches
+model = YOLO("best.pt")
 
 def detect_potholes(video):
-
     if video is None:
-        return "Please upload a video."
-
-    # ✅ video is already a file path string from Gradio
-    video_path = video  
+        return "Please upload a video.", None
+    
+    video_path = video  # gr.Video gives file path
 
-    # ✅ Reset runs folder (optional but clean)
+    # Remove old results
     if os.path.exists("runs"):
         shutil.rmtree("runs")
 
-    # ✅ YOLO prediction
     model.predict(
         source=video_path,
         save=True,
@@ -24,22 +21,25 @@ def detect_potholes(video):
         iou=0.5
     )
 
-    # ✅ Find output video path from YOLO runs
     output_files = glob.glob("runs/detect/*/*.mp4")
 
     if len(output_files) == 0:
-        return "❌ Error: No output generated."
+        return "❌ No output generated.", None
 
-    return output_files[0]
+    output_video = output_files[0]
+
+    return output_video, output_video   # ⬅️ return video for preview AND download
 
 
 app = gr.Interface(
     fn=detect_potholes,
     inputs=gr.Video(label="Upload Dashcam Video"),
-    outputs=gr.Video(label="Detected Video Output"),
+    outputs=[
+        gr.Video(label="Processed Video ✅"),
+        gr.File(label="Download Video ⬇️")
+    ],
     title="🚧 Pothole Detection AI",
     description="Upload a dashcam video and detect potholes using YOLO."
 )
 
 app.launch()
-