Update app.py

app.py

@@ -151,6 +151,8 @@ def generate_video(
     prompt: str,
     end_frame_upload=None,
     end_frame_generated=None,
+    strength_start: float = 1.,
+    strength_end: float = 1.,
     duration: float = 5,
     enhance_prompt: bool = True,
     negative_prompt: str = DEFAULT_NEGATIVE_PROMPT,
@@ -194,7 +196,7 @@ def generate_video(
             else:
                 temp_start_path = Path(start_frame)
             temp_paths.append(temp_start_path)
-            images.append((str(temp_start_path), 0, 1.0))
+            images.append((str(temp_start_path), 0, strength_start))
 
         # Save end frame (last frame index) if provided
         if end_frame is not None:
@@ -204,7 +206,7 @@ def generate_video(
             else:
                 temp_end_path = Path(end_frame)
             temp_paths.append(temp_end_path)
-            images.append((str(temp_end_path), num_frames - 1, 1.0))
+            images.append((str(temp_end_path), num_frames - 1, strength_end))
 
         # Get embeddings from text encoder space
         print(f"Encoding prompt: {prompt}")
@@ -341,6 +343,21 @@ with gr.Blocks(title="LTX-2 Keyframe Interpolation 🎥🔈") as demo:
                         label="Enhance Prompt",
                         value=True
                     )
+                with gr.Row():
+                    strength_start = gr.Slider(
+                        label="strength - start frame",
+                        minimum=0.0,
+                        maximum=1.0,
+                        value=1.0,
+                        step=0.05
+                    )
+                    strength_end = gr.Slider(
+                        label="strength - end frame",
+                        minimum=0.0,
+                        maximum=1.0,
+                        value=1.0,
+                        step=0.05
+                    )
                 
                 negative_prompt = gr.Textbox(
                     label="Negative Prompt",
@@ -412,6 +429,8 @@ with gr.Blocks(title="LTX-2 Keyframe Interpolation 🎥🔈") as demo:
             prompt,
             end_frame_upload,
             end_frame_generated,
+            strength_start,
+            strength_end,
             duration,
             enhance_prompt,
             negative_prompt,