Update app.py

app.py

@@ -153,9 +153,14 @@ def preview_line(video_file, scale_percent, line_start_x, line_start_y, line_end
     return preview_path
 
 def gradio_app(video, scale_percent, line_start_x, line_start_y, line_end_x, line_end_y, line_thickness, draw_tracks, view_img, view_in_counts, view_out_counts, track_thickness, region_thickness, line_dist_thresh, persist, conf, iou, classes_to_track, verbose):
+    # Save the uploaded video to the main folder
+    main_folder_video_path = "vehicle-counting.mp4"
+    with open(main_folder_video_path, "wb") as f:
+        f.write(open(video.name, "rb").read())
+    
     progress = gr.Progress()
-    total_frames = count_frames(video.name)
-    output_path = process_video(video.name, scale_percent, line_start_x, line_start_y, line_end_x, line_end_y, int(line_thickness), draw_tracks, view_img, view_in_counts, view_out_counts, int(track_thickness), int(region_thickness), line_dist_thresh, persist, conf, iou, classes_to_track, verbose, progress)
+    total_frames = count_frames(main_folder_video_path)
+    output_path = process_video(main_folder_video_path, scale_percent, line_start_x, line_start_y, line_end_x, line_end_y, int(line_thickness), draw_tracks, view_img, view_in_counts, view_out_counts, int(track_thickness), int(region_thickness), line_dist_thresh, persist, conf, iou, classes_to_track, verbose, progress)
     return output_path, output_path
 
 def update_preview(video, scale_percent, line_start_x, line_start_y, line_end_x, line_end_y, line_thickness):
@@ -167,10 +172,15 @@ def set_4k_coordinates():
 def set_1080p_coordinates():
     return 0, 700, 1920, 700
 
+def use_example_video():
+    example_video_path = "vehicle-counting.mp4"
+    return example_video_path
+
 with gr.Blocks() as demo:
     with gr.Row():
         with gr.Column(scale=1):
             video_input = gr.File(label="Upload your video")  # Removed max_size parameter
+            example_button = gr.Button("Use Example Video")
             with gr.Row():
                 set_4k_button = gr.Button("4K")
                 set_1080p_button = gr.Button("1080p")
@@ -214,5 +224,8 @@ with gr.Blocks() as demo:
 
     process_button.click(clear_previous_video, outputs=[video_output, download_button], queue=False)
     process_button.click(gradio_app, inputs=[video_input, scale_percent, line_start_x, line_start_y, line_end_x, line_end_y, line_thickness, draw_tracks, view_img, view_in_counts, view_out_counts, track_thickness, region_thickness, line_dist_thresh, persist, conf, iou, classes_to_track, verbose], outputs=[video_output, download_button])
+    
+    # Add the example button right below the video input
+    example_button.click(use_example_video, outputs=[video_input])
 
-demo.launch()
+demo.launch()