Update app.py

app.py

@@ -16,6 +16,9 @@ import torch
 from tools.painter import mask_painter
 import psutil
 import time
+import urllib.request
+import tempfile
+
 try: 
     from mmcv.cnn import ConvModule
 except:
@@ -69,7 +72,20 @@ def get_prompt(click_state, click_input):
     }
     return prompt
 
+def download_video(video_url):
+    tmp_dir = tempfile.gettempdir()
+    video_path = f"{tmp_dir}/input_video.mp4"
+    
+    urllib.request.urlretrieve(video_url, video_path)
+    
+    return video_path
+
+@app.route("/api/process_video") 
+def api_process_video(video_url):
 
+    video_path = download_video(video_url)
+    
+    return get_frames_from_video(video_path)
 
 # extract frames from upload video
 def get_frames_from_video(video_input, video_state):
@@ -541,7 +557,7 @@ with gr.Blocks() as iface:
                     outputs=[video_demo_operation],
                     # cache_examples=True,
                 ) 
-
+    iface.route("/api/process_video", api_process_video)
     # first step: get the video information 
     extract_frames_button.click(
         fn=get_frames_from_video,