Update app.py

app.py

@@ -7,6 +7,12 @@ from moviepy.editor import *
 token = os.environ.get('HF_TOKEN')
 pix2pix = gr.Blocks.load(name="spaces/fffiloni/instruct-pix2pix-clone", api_key=token)
 
+def trim_video(video_in, trim_value):
+    clip = VideoFileClip('Test_video.mp4')
+    clip1 = clip.subclip((0,00),(trim_value))
+    clip1.write_videofile('edited.mp4',codec='libx264')
+    return 'edited.mp4'
+
 def get_frames(video_in):
     frames = []
     # Opens the Video file
@@ -34,8 +40,9 @@ def create_video(frames, fps):
     return 'movie.mp4'
 
 
-def infer(prompt,video_in, seed_in):
-    break_vid = get_frames(video_in)
+def infer(prompt,video_in, seed_in, trim_value):
+    trimmed_vid = trim_video(video_in, trim_value)
+    break_vid = get_frames(trimmed_vid)
     
     frames_list= break_vid[0]
     fps = break_vid[1]
@@ -57,12 +64,14 @@ with gr.Blocks(css='style.css') as demo:
             with gr.Row():
                 prompt = gr.Textbox(placeholder="enter prompt")
                 video_inp = gr.Video(label="Video source", source="upload", type="filepath")
-                seed_inp = gr.Number(label="Seed", value=123456)
+                with gr.Row():
+                    seed_inp = gr.Number(label="Seed", value=123456)
+                    trim_in = gr.Slider(minimun=2.00, maximum=10.00, step=1.00, value=2.00)
             with gr.Column():
                 video_out = gr.Video(label="Pix2pix video result")
                 submit_btn = gr.Button("Generate Pix2Pix video")
 
-    inputs = [prompt,video_inp,seed_inp]
+    inputs = [prompt,video_inp,seed_inp, trim_in]
     outputs = [video_out]
     
     submit_btn.click(infer, inputs, outputs)