Update app.py

app.py

@@ -121,24 +121,27 @@ def run_inference(prompt, video_path, condition, video_length):
     processed_chunks = []
     
     for index, chunk_path in enumerate(chunks):
-        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}' --temp_chunk_path '{index}'  --width 512 --height 512 --fps 8 --video_length {video_length}"
-        subprocess.run(command, shell=True)
-
-        # Construct the video path
-        video_path_output = os.path.join(output_path, f"{index}.mp4")
-
-        # Append processed chunk to final array
-        processed_chunks.append(video_path_output)
+        if index = 0 :
+            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}' --temp_chunk_path '{index}'  --width 512 --height 512 --fps 8 --video_length {video_length}"
+            subprocess.run(command, shell=True)
+    
+            # Construct the video path
+            video_path_output = os.path.join(output_path, f"{index}.mp4")
+    
+            # Append processed chunk to final array
+            processed_chunks.append(video_path_output)
+        else:
+            print("finished")
 
     print(f"PROCESSED CHUNKS: {processed_chunks}")