Update app.py

app.py

@@ -9,6 +9,7 @@ from moviepy.video.io.ffmpeg_tools import ffmpeg_extract_subclip
 import moviepy.video.io.ImageSequenceClip
 
 uid=uuid.uuid4()
+if not os.path.exists(f'{uid}'): os.makedirs(f'{uid}')
 if not os.path.exists(f'{uid}-frames'): os.makedirs(f'{uid}-frames')
 if not os.path.exists(f'{uid}-rembg'): os.makedirs(f'{uid}-rembg')
 
@@ -109,11 +110,11 @@ def dl(inp):
         inp_out=inp.replace("https://","")
         inp_out=inp_out.replace("/","_").replace(".","_")
         os.system(f'yt-dlp "{inp}" --trim-filenames 100 -o "{inp_out}.mp4"')  
-        out = f"{inp_out}.mp4"
+        out = f"{uid}/{inp_out}.mp4"
     except Exception as e:
         print (e)
         out = None
-    return out,out,out,out
+    return out,out,out,out,out
 
 def trim_vid(vid,start_time,end_time):
     print (vid)
@@ -125,7 +126,7 @@ def trim_vid(vid,start_time,end_time):
     end_sec=int(end_time.split(":",2)[2])
     start=start_hr+start_min+start_sec
     end=end_hr+end_min+end_sec
-    #vid = f"{uid}-tmp.mp4"
+    #vid = f"{inp_out}.mp4"
     ffmpeg_extract_subclip(vid, start, end, targetname=f"{uid}-clip.mp4")
     out= f"{uid}-clip.mp4"
     return out,out 
@@ -167,6 +168,7 @@ with gr.Blocks() as app:
                     end_f = gr.Textbox(label = "End", value = "0:00:05", placeholder = "0:00:54")
                 trim_btn=gr.Button("Trim")
                 out_trim=gr.Video(format="mp4")
+                hid_box = gr.Textbox(visible=False)
 
     with gr.Tab("Frames"):
         frame_btn = gr.Button("Get Frames")
@@ -194,10 +196,10 @@ with gr.Blocks() as app:
     def echo_fn(inp):
         return inp 
     #outp_vid.change(echo_fn,outp_vid,[out_trim])
-    trim_btn.click(trim_vid,[outp_vid,start_f,end_f],[out_trim,in_vid])
+    trim_btn.click(trim_vid,[hid_box,start_f,end_f],[out_trim,in_vid])
     outp_vid.change(load_video,outp_vid,[frame_count,fps]).then(echo_fn,outp_vid,[out_trim])
     frame_btn.click(capture_function,[out_trim],[frame_gal,frame_file])
     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])
+    go_btn.click(dl,inp_url,[outp_vid,outp_file,out_trim,hid_box])
     app.load(fn=predict, inputs=[text_input,url_params], outputs=[text_output,text_input],_js=load_js)
 app.launch()