Update app.py

app.py

@@ -45,7 +45,7 @@ def resize_video(input_path, output_path, width):
     height = int(video.size[1] * (width / video.size[0]))
 
     # Resize the video
-    resized_video = video.resize(width=width, height=height)
+    resized_video = video.resize(width=width, height=512)
 
     # Write the resized video to the output path
     resized_video.write_videofile(output_path, codec='libx264')
@@ -86,6 +86,48 @@ def run_inference(prompt, video_path, condition, video_length):
 
         
 
+    return "done", video_path_output 
+
+def run_inference_chunks(prompt, video_path, condition, video_length):
+
+    # Specify the input and output paths
+    input_vid = video_path
+    resized_vid = 'resized.mp4'
+
+    # Call the function to resize the video
+    video_path = resize_video(input_vid, resized_vid, width=512)
+    width, height, fps = get_video_dimension(video_path)
+
+    print(f"{width} x {height} | {fps}")
+
+    # Split the video into chunks mp4 of 12 frames at video fps
+    # Store chunks as mp4 paths in an array
+
+    # For each mp4 chunks in chunks arrays, run command
+    # store video result in processed chunks array
+    
+    output_path = 'output/'
+    os.makedirs(output_path, exist_ok=True)
+
+    # Construct the final video path
+    video_path_output = os.path.join(output_path, f"{prompt}.mp4")
+
+    # Check if the file already exists
+    if os.path.exists(video_path_output):
+        # Delete the existing file
+        os.remove(video_path_output)
+
+    if video_length > 12:
+        command = f"python inference.py --prompt '{prompt}' --condition '{condition}' --video_path '{video_path}' --output_path '{output_path}' --width {width} --height {height} --fps {fps} --video_length {video_length} --is_long_video"
+    else:
+        command = f"python inference.py --prompt '{prompt}' --condition '{condition}' --video_path '{video_path}' --output_path '{output_path}' --width {width} --height {height} --fps {fps} --video_length {video_length}"
+    subprocess.run(command, shell=True)
+
+    # Construct the video path
+    video_path_output = os.path.join(output_path, f"{prompt}.mp4")
+
+        
+
     return "done", video_path_output 
 
 css="""