Update app.py

app.py

@@ -50,11 +50,18 @@ def run_inference(prompt, video_path, condition, video_length):
     chunk_size = 12
     chunks = cut_mp4_into_chunks(video_path, chunk_size)
 
+    processed_chunks = []
+
     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):
+        # Check if the file already exists
+        if os.path.exists(os.path.join(output_path, f"{prompt}.mp4")):
+            # Delete the existing file
+            os.remove(video_path_output)
+
         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}")
@@ -62,6 +69,26 @@ def run_inference(prompt, video_path, condition, video_length):
         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)
 
+        # Construct the video path
+        video_path_output = os.path.join(output_path, f"{prompt}.mp4")
+
+        # rename
+        new_file_name = os.path.join(output_path, f"new_file_name_{i}.mp4")
+        os.rename(video_path_output, new_file_name)
+        processed_chunks.append(new_file_name)
+
+    output_path = "final_video.mp4"
+    clips = []
+    for path in processed_chunks:
+        clip = VideoFileClip(path)
+        clips.append(clip)
+    
+    final_clip = concatenate_videoclips(clips)
+    final_clip.write_videofile(output_path, codec="libx264")
+    final_clip.close()
+
+    return "done", "final_video.mp4"
+        
 def working_run_inference(prompt, video_path, condition, video_length):