Update app.py

app.py

@@ -5,6 +5,9 @@ import json
 from rembg import remove as rm
 import cv2
 import uuid
+from moviepy.video.io.ffmpeg_tools import ffmpeg_extract_subclip
+
+
 uid=uuid.uuid4()
 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')
@@ -133,13 +136,34 @@ def dl(inp):
         print (e)
         out = None
     return out,out,out,out
-def trim_clip(inp):
+
+def trim_vid(vid,start_time,end_time):
+    start_hr=int(start_time.split(":",2)[0])*360
+    start_min=int(start_time.split(":",2)[1])*60
+    start_sec=int(start_time.split(":",2)[2])
+    end_hr=int(end_time.split(":",2)[0])*360
+    end_min=int(end_time.split(":",2)[1])*60
+    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"
+    ffmpeg_extract_subclip(vid, start, end, targetname=f"{uid}-clip.mp4")
+    return f"{uid}-clip.mp4"
+    
+def other():    
     new_video_in = str(vid)
     capture = cv2.VideoCapture(new_video_in)    
     frame_count = int(capture.get(cv2.CAP_PROP_FRAME_COUNT))
     fps = capture.get(cv2.CAP_PROP_FPS)
     fbox3=[]
     frame_count1= int(frame_count)
+
+    start = (int(fps*int(start_f)))
+
+    capture.set(cv2.CAP_PROP_POS_FRAMES, i-1)
+
+
+    
     for i in range(int(frame_count1)):
         capture.set(cv2.CAP_PROP_POS_FRAMES, i)
         ret, frame_f = capture.read(i)
@@ -169,8 +193,8 @@ with gr.Blocks() as app:
                 outp_file=gr.Files()
     with gr.Tab("Trim"):
         with gr.Row():
-            start_f=gr.Number(value=0)
-            end_f=gr.Number(value=1)
+            start_f = gr.Textbox(label = "Start", value = "0:00:00", placeholder = "0:00:23")
+            end_f = gr.Textbox(label = "End", value = "0:00:05", placeholder = "0:00:54")
             trim_btn=gr.Button("Trim")
         with gr.Row():
             in_trim = gr.Video()
@@ -196,8 +220,13 @@ with gr.Blocks() as app:
         text_input=gr.Textbox()
         text_output=gr.Textbox()
         url_params=gr.JSON()
-    in_trim.change(None,in_trim,out_trim)
-    trim_btn.click(trim_clip,in_trim,out_trim)
+
+    def echo_fn(inp):
+        return inp 
+        
+    in_trim.change(echo_fn,in_trim,out_trim)
+    trim_btn.click(trim_vid,[in_trim,start_f,end_f],out_trim)
+    
     outp_vid.change(load_video,outp_vid,[frame_count,fps])
     frame_btn.click(capture_function,[outp_vid],[frame_gal,frame_file])
     rem_btn.click(rem_bg,[outp_vid],[rem_vid,frame_num])