Update app.py

app.py

@@ -29,6 +29,16 @@ def get_frame_count_in_duration(filepath):
     video.release()
     return gr.update(maximum=frame_count)
 
+# Function to process a video chunk and return the processed video
+def process_video_chunk(chunk, index):
+    # Count the frame number of the video chunk
+    frame_count = len(chunk)
+
+    # Export the processed chunk to a file with the index in the file name
+    processed_chunk_filename = f'processed_chunk_{index}.mp4'
+    processed_chunk.write_videofile(processed_chunk_filename)
+
+    return processed_chunk_filename, frame_count
 
 # Function to split video into chunks
 def split_video_into_chunks(video_path, chunk_size):
@@ -93,11 +103,12 @@ def run_inference(prompt, video_path, condition, video_length):
     # Split the video into chunks
     video_chunks = split_video_into_chunks(video_path, chunk_size)
 
-    # Process each chunk and store the processed chunk filenames
-    processed_chunk_filenames = []
-    for i, chunk in enumerate(video_chunks):
+    # Process each chunk and store the processed chunk filenames and frame counts
+    processed_chunk_info = []
+    for i, (chunk, _) in enumerate(video_chunks):
+        processed_chunk_filename, frame_count = process_video_chunk(chunk, i)
         # Count the frame number of the video chunk
-        frame_count = len(chunk)
+        #frame_count = len(chunk)
         command = f"python inference.py --prompt '{prompt}' --condition '{condition}' --video_path '{chunk}' --output_path '{output_path}' --video_length {frame_count} --smoother_steps 19 20"
         subprocess.run(command, shell=True)
 
@@ -105,10 +116,11 @@ def run_inference(prompt, video_path, condition, video_length):
         video_path_output = os.path.join(output_path, f"{prompt}_{i}.mp4")
 
         processed_chunk_filename = video_path_output
-        processed_chunk_filenames.append(processed_chunk_filename)
+        #processed_chunk_filenames.append(processed_chunk_filename)
+        processed_chunk_info.append((processed_chunk_filename, frame_count))
 
     # Load the processed video chunks
-    processed_chunks = [VideoFileClip(filename) for filename in processed_chunk_filenames]
+    processed_chunks = [VideoFileClip(filename) for filename, _ in processed_chunk_info]
 
     # Concatenate the processed video chunks into a final video
     final_video = concatenate_videoclips(processed_chunks)
@@ -117,7 +129,7 @@ def run_inference(prompt, video_path, condition, video_length):
     final_video.write_videofile('final_video.mp4')
 
     # Clean up the temporary processed chunk files (optional)
-    for filename in processed_chunk_filenames:
+    for filename, _ in processed_chunk_info:
         os.remove(filename)
 
     return "done", 'final_video.mp4'