Update app.py

app.py

@@ -30,10 +30,39 @@ def get_frame_count_in_duration(filepath):
     return gr.update(maximum=frame_count)
 
 
+def cut_mp4_into_chunks(input_file, chunk_size):
+    video = VideoFileClip(input_file)
+    frame_count = int(video.fps * video.duration)
+    num_chunks = (frame_count + chunk_size - 1) // chunk_size  # Ceiling division
 
+    chunks = []
 
+    for i in range(num_chunks):
+        start_frame = i * chunk_size
+        end_frame = min((i + 1) * chunk_size, frame_count)
+        chunk = video.subclip(start_frame / video.fps, end_frame / video.fps)
+        chunk_frame_count = end_frame - start_frame
+        chunks.append((chunk, chunk_frame_count))
+
+    return chunks
 
 def run_inference(prompt, video_path, condition, video_length):
+    chunk_size = 12
+    chunks = cut_mp4_into_chunks(video_path, chunk_size)
+
+    output_path = 'output/'
+    os.makedirs(output_path, exist_ok=True)
+
+    # Accessing chunks and frame counts by index
+    for i, (chunk, frame_count) in enumerate(chunks):
+        chunk.write_videofile(f'chunk_{i}.mp4')  # Saving the chunk to a file
+        chunk_path = f'chunk_{i}.mp4'
+        print(f"Chunk {i}: Frame Count = {frame_count}")
+
+        command = f"python inference.py --prompt '{prompt}' --condition '{condition}' --video_path '{chunk_path}' --output_path '{output_path}' --video_length {frame_count}"
+        subprocess.run(command, shell=True)
+
+def working_run_inference(prompt, video_path, condition, video_length):
 
     
     output_path = 'output/'