Update app.py

app.py

@@ -121,19 +121,25 @@ def predict(text, url_params):
     print (mod_url)
     return ["" + text + "", mod_url]
 
-def dl(inp):
+def dl(inp,img):
     out = None
     out_file=[]
-    try:
-        inp_out=inp.replace("https://","")
-        inp_out=inp_out.replace("/","_").replace(".","_").replace("=","_").replace("?","_")
-        os.system(f'yt-dlp "{inp}" --trim-filenames 160 -o "{uid}/{inp_out}.mp4" -S res,mp4 --recode mp4')  
-        out = f"{uid}/{inp_out}.mp4"
-        capture = cv2.VideoCapture(out)
+    if img == None and inp !="":
+        try:
+            inp_out=inp.replace("https://","")
+            inp_out=inp_out.replace("/","_").replace(".","_").replace("=","_").replace("?","_")
+            os.system(f'yt-dlp "{inp}" --trim-filenames 160 -o "{uid}/{inp_out}.mp4" -S res,mp4 --recode mp4')  
+            out = f"{uid}/{inp_out}.mp4"
+            capture = cv2.VideoCapture(out)
+            fps = capture.get(cv2.CAP_PROP_FPS)
+            capture.release()
+        except Exception as e:
+            out = None
+    elif img !=None and inp == "":
+        capture = cv2.VideoCapture(img)
         fps = capture.get(cv2.CAP_PROP_FPS)
         capture.release()
-    except Exception as e:
-        out = None
+        out = f"{img}"
     return out,out,out,out,fps
 def dl_json(inp):
     out_json={}
@@ -276,14 +282,22 @@ with gr.Blocks() as app:
 
     def echo_fn(inp):
         return inp 
+    def check_load(inp_url,outp_vid,hid_box,start_f,end_f):
+        if outp_vid == None and inp_url !="":
+            out_trim,in_vid,trim_count=trim_vid(hid_box,start_f,end_f)
+        elif outp_vid !=None and inp_url == "":
+            out_trim = None
+            in_vid=outp_vid
+            trim_count = ""
+        return out_trim,in_vid,trim_count      
     #outp_vid.change(echo_fn,outp_vid,[out_trim])
     info_btn.click(dl_json,inp_url,info_json)
     speed_btn.click(update_speed,[hid_box,clip_speed,hid_fps],[outp_vid,hid_box])
     gif_btn.click(make_gif,None,[gif_show,gif_file])
     trim_btn.click(trim_vid,[hid_box,start_f,end_f],[out_trim,in_vid,trim_count])
-    outp_vid.change(load_video,outp_vid,[frame_count,fps,end_f]).then(trim_vid,[hid_box,start_f,end_f],[out_trim,in_vid,trim_count])
+    outp_vid.change(load_video,outp_vid,[frame_count,fps,end_f]).then(check_load,[hid_box,start_f,end_f],[out_trim,in_vid,trim_count])
     frame_btn.click(capture_function,[out_trim],[frame_gal,frame_file,frame_stat])
     rem_btn.click(rem_bg,[out_trim],[rem_vid,frame_num,rem_bg_vid])
-    go_btn.click(dl,inp_url,[outp_vid,outp_file,out_trim,hid_box,hid_fps])
+    go_btn.click(dl,[inp_url,outp_vid],[outp_vid,outp_file,out_trim,hid_box,hid_fps])
     app.load(fn=predict, inputs=[text_input,url_params], outputs=[text_output,text_input],_js=load_js)
 app.launch()