Update app.py

app.py

@@ -103,7 +103,7 @@ def calculate_dimensions(width, height):
 @spaces.GPU(duration=120)
 def generate_video(prompt, first_image_path, last_image_path, duration_seconds):
     # Create a temporary directory for resized images
-    temp_dir = f"temp_resized_{uuid.uuid4().hex}"
+    temp_dir = "input"
     os.makedirs(temp_dir, exist_ok=True)
     
     with torch.inference_mode():
@@ -118,13 +118,15 @@ def generate_video(prompt, first_image_path, last_image_path, duration_seconds):
         # Resize first image
         with Image.open(first_image_path) as img:
             img_resized = img.resize((target_width, target_height), Image.Resampling.LANCZOS)
-            resized_first_path = os.path.join(temp_dir, "first_frame_resized.png")
+            resized_first_path = os.path.join(temp_dir, f"first_frame_resized_{uuid.uuid4().hex}.png")
+            print(resized_first_path)
             img_resized.save(resized_first_path)
 
         # Resize second image to match the target dimensions
         with Image.open(last_image_path) as img:
             img_resized = img.resize((target_width, target_height), Image.Resampling.LANCZOS)
-            resized_last_path = os.path.join(temp_dir, "last_frame_resized.png")
+            resized_last_path = os.path.join(temp_dir, f"last_frame_resized_{uuid.uuid4().hex}.png")
+            print(resized_last_path)
             img_resized.save(resized_last_path)
         print(f"Images resized to {target_width}x{target_height} and saved temporarily.")
         # --- End Preprocessing ---