add video upload option

app.py

@@ -117,13 +117,16 @@ def detect_beeps(video_path, event_length=30, beep_height=0.8):
     return event_start, event_end
 
 
-def inference(stream_url, start_time, end_time, beep_detection_on, event_length, count_only_api, api_key,
+def inference(in_video, stream_url, start_time, end_time, beep_detection_on, event_length, count_only_api, api_key,
               img_size=256, seq_len=64, stride_length=32, stride_pad=3, batch_size=4,
               miss_threshold=0.8, marks_threshold=0.5, median_pred_filter=True, both_feet=True, 
               api_call=False,
               progress=gr.Progress()):
     progress(0, desc="Downloading clip...")
-    in_video = download_clips(stream_url, os.path.join(os.getcwd(), 'clips'), start_time, end_time)
+    if in_video is None:
+        in_video = download_clips(stream_url, os.path.join(os.getcwd(), 'clips'), start_time, end_time)
+    else:  # local uploaded video (still resize with ffmpeg)
+        in_video = download_clips(in_video, os.path.join(os.getcwd(), 'clips'), start_time, end_time)
     progress(0, desc="Running inference...")
     has_access = False
     if api_call:
@@ -406,10 +409,15 @@ def inference(stream_url, start_time, end_time, beep_detection_on, event_length,
         
 
 with gr.Blocks() as demo:
+    with gr.Row():
+        in_video = gr.PlayableVideo(label="Input Video", elem_id='input-video', format='mp4', 
+                                width=400, height=400, interactive=True, container=True,
+                                max_length=300)
     with gr.Row():
         with gr.Column():
             in_stream_url = gr.Textbox(label="Stream URL", elem_id='stream-url', visible=True)
-            in_stream_start = gr.Textbox(label="Start Time", elem_id='stream-start', visible=True)
+            
+            in_stream_start = gr.Textbox(label="Start Time", elem_id='stream-start', visible=True, value='00:00:00')
             in_stream_end = gr.Textbox(label="End Time", elem_id='stream-end', visible=True)
         with gr.Column():
             beep_detection_on = gr.Checkbox(label="Detect Beeps", elem_id='detect-beeps', visible=True)
@@ -440,19 +448,19 @@ with gr.Blocks() as demo:
 
     demo_inference = partial(inference, count_only_api=False, api_key=None)
     
-    run_button.click(demo_inference, [in_stream_url, in_stream_start, in_stream_end, beep_detection_on, event_length], 
+    run_button.click(demo_inference, [in_video, in_stream_url, in_stream_start, in_stream_end, beep_detection_on, event_length], 
                      outputs=[out_video, out_text, out_plot, out_hist, out_event_type_dist])
     api_inference = partial(inference, api_call=True)
-    api_dummy_button.click(api_inference, [in_stream_url, in_stream_start, in_stream_end, beep_detection_on, event_length, count_only, api_token], 
+    api_dummy_button.click(api_inference, [in_video, in_stream_url, in_stream_start, in_stream_end, beep_detection_on, event_length, count_only, api_token], 
                            outputs=[period_length], api_name='inference')
     examples = [
         #['https://hiemdall-dev2.azurewebsites.net/api/clip/clp_vrpWTyjM/mp4', '00:00:00', '00:01:10', True, 60],
-        ['https://hiemdall-dev2.azurewebsites.net/api/playlist/rec_UGEhqlMh/vod', '00:00:18', '00:00:55', True, 30],
+        [None, 'https://hiemdall-dev2.azurewebsites.net/api/playlist/rec_UGEhqlMh/vod', '00:00:18', '00:00:55', True, 30],
         #['https://hiemdall-dev2.azurewebsites.net/api/playlist/rec_rd2FAyUo/vod', '01:24:22', '01:25:35', True, 60]
         #['https://hiemdall-dev2.azurewebsites.net/api/playlist/rec_PY5Ukaua/vod, '00:52:53', '00:55:00', True, 120]
     ]
     gr.Examples(examples, 
-                inputs=[in_stream_url, in_stream_start, in_stream_end, beep_detection_on, event_length], 
+                inputs=[in_video, in_stream_url, in_stream_start, in_stream_end, beep_detection_on, event_length], 
                 outputs=[out_video, out_text, out_plot, out_hist, out_event_type_dist],
                 fn=demo_inference, cache_examples=os.getenv('SYSTEM') == 'spaces')