Upload 5 files

README.md

@@ -9,16 +9,7 @@ app_file: app.py
 pinned: true
 ---
 
-# NHAI DAMS Phase 1: Video to Frame Extraction
+# NHAI DAMS Phase 1: Video to Frame Extractor
 
-This Gradio app allows you to upload a drone video and extract frames every 1 second.
-
-### Features:
-- Supports `.mp4` files
-- Saves extracted frames to `frames/<video_name>/`
-- Works on Gradio 4.24+ with `type="filepath"` support
-
-### How to Use:
-1. Upload your video
-2. Click "Start Frame Extraction"
-3. View output path and frame count
+Upload a `.mp4` video and extract frames every 1 second into the `/frames/<video_name>/` folder.
+Tested with Gradio 4.24.0 using only valid schema-safe return types.

app.py

@@ -2,16 +2,20 @@ import gradio as gr
 from backend.video_processor import extract_frames
 
 def process_video(video_path):
-    saved, folder = extract_frames(video_path)
-    return f"✅ Extracted {saved} frames to {folder}"
+    try:
+        saved, folder = extract_frames(video_path)
+        return f"✅ {saved} frames extracted and saved to folder: {folder}"
+    except Exception as e:
+        return f"❌ Error during frame extraction: {str(e)}"
 
 with gr.Blocks() as demo:
-    gr.Markdown("# 🎥 NHAI DAMS Phase 1: Video to Frame Extractor")
+    gr.Markdown("## 🎥 NHAI DAMS Phase 1: Video to Frame Extractor")
 
-    video_input = gr.File(label="Upload Drone Video (.mp4)", type="filepath")
-    result_output = gr.Textbox(label="Status")
+    with gr.Row():
+        video_input = gr.File(label="Upload Drone Video (.mp4)", type="filepath")
+        output = gr.Textbox(label="Frame Extraction Status", lines=4)
 
-    process_button = gr.Button("Start Frame Extraction")
-    process_button.click(fn=process_video, inputs=[video_input], outputs=[result_output])
+    extract_button = gr.Button("Start Extraction")
+    extract_button.click(fn=process_video, inputs=[video_input], outputs=[output])
 
 demo.launch()

backend/video_processor.py

@@ -8,6 +8,9 @@ def extract_frames(video_path, output_dir='frames', every_n_sec=1):
     os.makedirs(save_path, exist_ok=True)
 
     cap = cv2.VideoCapture(video_path)
+    if not cap.isOpened():
+        raise ValueError("Failed to open video.")
+
     fps = cap.get(cv2.CAP_PROP_FPS)
     interval = max(1, int(fps * every_n_sec))
     frame_idx = 0
@@ -18,8 +21,8 @@ def extract_frames(video_path, output_dir='frames', every_n_sec=1):
         if not ret:
             break
         if frame_idx % interval == 0:
-            out_file = os.path.join(save_path, f"frame_{saved:04d}.jpg")
-            cv2.imwrite(out_file, frame)
+            filename = os.path.join(save_path, f"frame_{saved:04d}.jpg")
+            cv2.imwrite(filename, frame)
             saved += 1
         frame_idx += 1
 

requirements.txt

@@ -1,2 +1,2 @@
-gradio==4.44.1
+gradio==4.24.0
 opencv-python