Update app.py

app.py

@@ -103,7 +103,7 @@ DEFAULT_NEGATIVE_PROMPT = (
 )
 DEFAULT_FRAME_RATE = 24.0
 MIN_DIM, MAX_DIM, STEP = 256, 1280, 64
-MIN_FRAMES, MAX_FRAMES = 9, 257
+MIN_FRAMES, MAX_FRAMES = 9, 721
 
 # Resolution presets with high/low tiers
 RESOLUTIONS = {
@@ -201,7 +201,7 @@ class HQPipelineWithCachedLoRA:
             [prompt, negative_prompt],
             self.model_ledger,
             enhance_first_prompt=enhance_prompt,
-            enhance_prompt_image=images[0][0] if len(images) > 0 else None,
+            enhance_prompt_image=images[0].path if len(images) > 0 else None,
             enhance_prompt_seed=seed,
         )
         
@@ -677,7 +677,10 @@ def detect_aspect_ratio(image) -> str:
 
 
 def on_image_upload(first_image, last_image, high_res):
-    ref_image = first_image if first_image is not None else last_image
+    # Always use first_image for resolution calculation
+    ref_image = first_image
+    if ref_image is None:
+        return gr.update(), gr.update()
     aspect = detect_aspect_ratio(ref_image)
     tier = "high" if high_res else "low"
     w, h = RESOLUTIONS[tier][aspect]
@@ -685,7 +688,10 @@ def on_image_upload(first_image, last_image, high_res):
 
 
 def on_highres_toggle(first_image, last_image, high_res):
-    ref_image = first_image if first_image is not None else last_image
+    # Always use first_image for resolution calculation
+    ref_image = first_image
+    if ref_image is None:
+        return gr.update(), gr.update()
     aspect = detect_aspect_ratio(ref_image)
     tier = "high" if high_res else "low"
     w, h = RESOLUTIONS[tier][aspect]
@@ -783,7 +789,7 @@ def generate_video(
         images = []
         output_dir = Path("outputs")
         output_dir.mkdir(exist_ok=True)
-
+        
         if first_image is not None:
             temp_first_path = output_dir / f"temp_first_{current_seed}.jpg"
             if hasattr(first_image, "save"):
@@ -791,8 +797,8 @@ def generate_video(
             else:
                 import shutil
                 shutil.copy(first_image, temp_first_path)
-            images.append((str(temp_first_path), 1.0))
-
+            images.append(ImageConditioningInput(path=str(temp_first_path), frame_idx=0, strength=1.0))
+        
         if last_image is not None:
             temp_last_path = output_dir / f"temp_last_{current_seed}.jpg"
             if hasattr(last_image, "save"):
@@ -800,7 +806,7 @@ def generate_video(
             else:
                 import shutil
                 shutil.copy(last_image, temp_last_path)
-            images.append((str(temp_last_path), 1.0))
+            images.append(ImageConditioningInput(path=str(temp_last_path), frame_idx=num_frames - 1, strength=1.0))
 
         tiling_config = TilingConfig.default()
         video_chunks_number = get_video_chunks_number(num_frames, tiling_config)