Update app.py

app.py

@@ -1,15 +1,21 @@
 import gradio as gr
 from core_pipeline import extract_frames, detect_trees, plot_detections
-import numpy as np
 
-def process_video(video_path):
-    if not video_path:
+def process_video(video_file):
+    import tempfile
+    import numpy as np
+
+    if video_file is None:
         return None
 
+    with tempfile.NamedTemporaryFile(delete=False, suffix=".mp4") as tmp:
+        tmp.write(video_file.read())
+        video_path = tmp.name
+
     frames = extract_frames(video_path)
     results = []
 
-    for i, frame in enumerate(frames[:3]):  # limit to 3 sample frames for preview
+    for i, frame in enumerate(frames[:3]):
         detected, bboxes, confs, labels = detect_trees(frame)
         annotated = plot_detections(detected, bboxes)
         results.append(annotated)
@@ -21,8 +27,8 @@ def process_video(video_path):
 
 gr.Interface(
     fn=process_video,
-    inputs=gr.Video(label="Upload Drone Video", type="filepath"),
+    inputs=gr.File(label="Upload Drone Video", file_types=[".mp4"]),  # ✅ FIXED
     outputs=gr.Image(label="Tree Detections (Sample Frames)"),
     title="🌳 Tree Height Detection from Drone Video",
-    description="Upload top-down drone video to detect trees and visualize sample frames. Height estimation is possible if SfM data is provided."
+    description="Upload drone video to detect trees. Sample frames will be shown with bounding boxes."
 ).launch()