Update app.py

app.py

@@ -115,31 +115,34 @@ def run_inference(prompt, video_path, condition, video_length):
     # Chunkify the video into 12 frames chunks
     chunks = chunkify(resized, target_fps, total_frames)
 
-
-
-
-    
     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")
+    processed_chunks = []
+    
+    for index, chunk_path in enumerate(chunks_array):
+        print(f"Chunk #{index}: {chunk_path}")
+        
+        # Check if the file already exists
+        if os.path.exists(os.path.join(output_path, f"{index}.mp4")):
+            # Delete the existing file
+            os.remove(os.path.join(output_path, f"{index}.mp4"))
+
+    #if video_length > 12:
+    #    command = f"python inference.py --prompt '{prompt}' --condition '{condition}' --video_path '{video_path}' --output_path '{output_path}' --width 512 --height 512 --fps 8 --video_length {video_length} --is_long_video"
+    #else:
+        command = f"python inference.py --prompt '{prompt}' --condition '{condition}' --video_path '{chunk_path}' --output_path '{output_path}' --width 512 --height 512 --fps 8 --video_length {video_length}"
+        subprocess.run(command, shell=True)
 
-    # Check if the file already exists
-    if os.path.exists(video_path_output):
-        # Delete the existing file
-        os.remove(video_path_output)
+        # Construct the video path
+        video_path_output = os.path.join(output_path, f"{index}.mp4")
 
-    if video_length > 12:
-        command = f"python inference.py --prompt '{prompt}' --condition '{condition}' --video_path '{video_path}' --output_path '{output_path}' --width 512 --height 512 --fps 8 --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 512 --height 512 --fps 8 --video_length {video_length}"
-    subprocess.run(command, shell=True)
+        # Append processed chunk to final array
+        processed_chunks.append(video_path_output)
 
-    # Construct the video path
-    video_path_output = os.path.join(output_path, f"{prompt}.mp4")
+    print(f"PROCESSED CHUNKS: {processed_chunks}")
 
-    return "done", video_path_output 
+    return "done", processed_chunks[0]