Update app.py

app.py

@@ -313,6 +313,8 @@ demopose_lora_path = hf_hub_download(repo_id=LORA_REPO, filename="clapping-cheek
 voice_lora_path = hf_hub_download(repo_id=LORA_REPO, filename="hentai_voice_ltx23.safetensors")
 realism_lora_path = hf_hub_download(repo_id=LORA_REPO, filename="FurryenhancerLTX2.3V1.215.safetensors")
 transition_lora_path = hf_hub_download(repo_id=LORA_REPO, filename="LTX-2_takerpov_lora_v1.2.safetensors") # takerpov1, taker pov
+physics_lora_path = hf_hub_download(repo_id=LORA_REPO, filename="LTX2.3_Better_Physics_PhysLTX.safetensors")
+reasoning_lora_path = hf_hub_download(repo_id="LiconStudio/Ltx2.3-VBVR-lora-I2V", filename="Ltx2.3-Licon-VBVR-I2V-390K-R32.safetensors")
 
 print(f"Pose LoRA: {pose_lora_path}")
 print(f"General LoRA: {general_lora_path}")
@@ -326,6 +328,8 @@ print(f"Demopose LoRA: {demopose_lora_path}")
 print(f"Voice LoRA: {voice_lora_path}")
 print(f"Realism LoRA: {realism_lora_path}")
 print(f"Transition LoRA: {transition_lora_path}")
+print(f"Physics LoRA: {physics_lora_path}")
+print(f"Reasoning LoRA: {reasoning_lora_path}")
 # ----------------------------------------------------------------
 
 print(f"Checkpoint: {checkpoint_path}")
@@ -343,7 +347,7 @@ pipeline = LTX23DistilledA2VPipeline(
 )
 # ----------------------------------------------------------------
 
-def _make_lora_key(pose_strength: float, general_strength: float, motion_strength: float, dreamlay_strength: float, mself_strength: float, dramatic_strength: float, fluid_strength: float, liquid_strength: float, demopose_strength: float, voice_strength: float, realism_strength: float, transition_strength: float) -> tuple[str, str]:
+def _make_lora_key(pose_strength: float, general_strength: float, motion_strength: float, dreamlay_strength: float, mself_strength: float, dramatic_strength: float, fluid_strength: float, liquid_strength: float, demopose_strength: float, voice_strength: float, realism_strength: float, transition_strength: float, physics_strength: float, reasoning_strength: float) -> tuple[str, str]:
     rp = round(float(pose_strength), 2)
     rg = round(float(general_strength), 2)
     rm = round(float(motion_strength), 2)
@@ -356,7 +360,9 @@ def _make_lora_key(pose_strength: float, general_strength: float, motion_strengt
     rv = round(float(voice_strength), 2)
     re = round(float(realism_strength), 2)
     rt = round(float(transition_strength), 2)
-    key_str = f"{pose_lora_path}:{rp}|{general_lora_path}:{rg}|{motion_lora_path}:{rm}|{dreamlay_lora_path}:{rd}|{mself_lora_path}:{rs}|{dramatic_lora_path}:{rr}|{fluid_lora_path}:{rf}|{liquid_lora_path}:{rl}|{demopose_lora_path}:{ro}|{voice_lora_path}:{rv}|{realism_lora_path}:{re}|{transition_lora_path}:{rt}"
+    ry = round(float(physics_strength), 2)
+    ri = round(float(reasoning_strength), 2)
+    key_str = f"{pose_lora_path}:{rp}|{general_lora_path}:{rg}|{motion_lora_path}:{rm}|{dreamlay_lora_path}:{rd}|{mself_lora_path}:{rs}|{dramatic_lora_path}:{rr}|{fluid_lora_path}:{rf}|{liquid_lora_path}:{rl}|{demopose_lora_path}:{ro}|{voice_lora_path}:{rv}|{realism_lora_path}:{re}|{transition_lora_path}:{rt}|{physics_lora_path}:{ry}|{reasoning_lora_path}:{ri}"
     key = hashlib.sha256(key_str.encode("utf-8")).hexdigest()
     return key, key_str
 
@@ -374,6 +380,8 @@ def prepare_lora_cache(
     voice_strength: float,
     realism_strength: float,
     transition_strength: float,
+    physics_strength: float,
+    reasoning_strength: float,
     progress=gr.Progress(track_tqdm=True),
 ):
     """
@@ -386,7 +394,7 @@ def prepare_lora_cache(
     global PENDING_LORA_KEY, PENDING_LORA_STATE, PENDING_LORA_STATUS
 
     ledger = pipeline.model_ledger
-    key, _ = _make_lora_key(pose_strength, general_strength, motion_strength, dreamlay_strength, mself_strength, dramatic_strength, fluid_strength, liquid_strength, demopose_strength, voice_strength, realism_strength, transition_strength)
+    key, _ = _make_lora_key(pose_strength, general_strength, motion_strength, dreamlay_strength, mself_strength, dramatic_strength, fluid_strength, liquid_strength, demopose_strength, voice_strength, realism_strength, transition_strength, physics_strength, reasoning_strength)
     cache_path = LORA_CACHE_DIR / f"{key}.safetensors"
 
     progress(0.05, desc="Preparing LoRA state")
@@ -414,6 +422,8 @@ def prepare_lora_cache(
         (voice_lora_path, round(float(voice_strength), 2)),
         (realism_lora_path, round(float(realism_strength), 2)),
         (transition_lora_path, round(float(transition_strength), 2)),
+        (physics_lora_path, round(float(physics_strength), 2)),
+        (reasoning_lora_path, round(float(reasoning_strength), 2)),
     ]
     loras_for_builder = [
         LoraPathStrengthAndSDOps(path, strength, LTXV_LORA_COMFY_RENAMING_MAP)
@@ -610,6 +620,8 @@ def get_gpu_duration(
     voice_strength: float = 0.0,
     realism_strength: float = 0.0,
     transition_strength: float = 0.0,
+    physics_strength: float = 0.0,
+    reasoning_strength: float = 0.0,
     progress=None,
 ):
     return int(gpu_duration)
@@ -640,6 +652,8 @@ def generate_video(
     voice_strength: float = 0.0,
     realism_strength: float = 0.0,
     transition_strength: float = 0.0,
+    physics_strength: float = 0.0,
+    reasoning_strength: float = 0.0,
     progress=gr.Progress(track_tqdm=True),
 ):
     try:
@@ -796,6 +810,14 @@ with gr.Blocks(title="LTX-2.3 Distilled") as demo:
                         label="POV strength",
                         minimum=0.0, maximum=2.0, value=0.0, step=0.01
                     )
+                    physics_strength = gr.Slider(
+                        label="Physics strength",
+                        minimum=0.0, maximum=2.0, value=0.0, step=0.01
+                    )
+                    reasoning_strength = gr.Slider(
+                        label="Alternative Reasoning strength",
+                        minimum=0.0, maximum=2.0, value=0.0, step=0.01
+                    )
                 prepare_lora_btn = gr.Button("Prepare / Load LoRA Cache", variant="secondary")
                 lora_status = gr.Textbox(
                     label="LoRA Cache Status",
@@ -844,12 +866,14 @@ with gr.Blocks(title="LTX-2.3 Distilled") as demo:
                 0.0,
                 0.0,
                 0.0,
+                0.0,
+                0.0,
             ],
         ],
         inputs=[
             first_image, last_image, input_audio, prompt, duration, gpu_duration,
             enhance_prompt, seed, randomize_seed, height, width,
-            pose_strength, general_strength, motion_strength, dreamlay_strength, mself_strength, dramatic_strength, fluid_strength, liquid_strength, demopose_strength, voice_strength, realism_strength, transition_strength,
+            pose_strength, general_strength, motion_strength, dreamlay_strength, mself_strength, dramatic_strength, fluid_strength, liquid_strength, demopose_strength, voice_strength, realism_strength, transition_strength, physics_strength, reasoning_strength,
         ],
     )
 
@@ -873,7 +897,7 @@ with gr.Blocks(title="LTX-2.3 Distilled") as demo:
 
     prepare_lora_btn.click(
         fn=prepare_lora_cache,
-        inputs=[pose_strength, general_strength, motion_strength, dreamlay_strength, mself_strength, dramatic_strength, fluid_strength, liquid_strength, demopose_strength, voice_strength, realism_strength, transition_strength],
+        inputs=[pose_strength, general_strength, motion_strength, dreamlay_strength, mself_strength, dramatic_strength, fluid_strength, liquid_strength, demopose_strength, voice_strength, realism_strength, transition_strength, physics_strength, reasoning_strength],
         outputs=[lora_status],
     )
     
@@ -882,7 +906,7 @@ with gr.Blocks(title="LTX-2.3 Distilled") as demo:
         inputs=[
             first_image, last_image, input_audio, prompt, duration, gpu_duration, enhance_prompt,
             seed, randomize_seed, height, width,
-            pose_strength, general_strength, motion_strength, dreamlay_strength, mself_strength, dramatic_strength, fluid_strength, liquid_strength, demopose_strength, voice_strength, realism_strength, transition_strength,
+            pose_strength, general_strength, motion_strength, dreamlay_strength, mself_strength, dramatic_strength, fluid_strength, liquid_strength, demopose_strength, voice_strength, realism_strength, transition_strength, physics_strength, reasoning_strength,
         ],
         outputs=[output_video, seed],
     )