Update app.py

app.py

@@ -39,15 +39,30 @@ LORA_REPO = os.environ.get("H3_LORA_REPO", "dagloop5/LoRA")
 LORA_FILES = {
     "loraa": os.environ.get("H3_LORA_A_FILE", "Mylo_lora_epoch31.safetensors"),
     "lorab": os.environ.get("H3_LORA_B_FILE", "VBVR_H3_attn_only.safetensors"),
+    "lorac": os.environ.get("H3_LORA_C_FILE", "AIO_V2.safetensors"),
+    "lorad": os.environ.get("H3_LORA_D_FILE", "Furry enhancer Video H3 V2.54.safetensors"),
+    "lorae": os.environ.get("H3_LORA_E_FILE", "sb_H3_i2v_v1.1.safetensors"),
+    "loraf": os.environ.get("H3_LORA_F_FILE", "moawxx_000002000.safetensors"),
+    "lorag": os.environ.get("H3_LORA_G_FILE", "H3_ref2va_shot_v1_fp16.safetensors"),
 }
 # Display names, keyed the same as LORA_FILES — used in the UI slider labels, the per-request report line, and
 # the status line's failure list. Keep these two dicts' keys in sync when adding a LoRA.
 LORA_LABELS = {
     "loraa": "Anthro Enhancer",
     "lorab": "Reasoning Enhancer",
+    "lorac": "HM-AIO", # hmmotion
+    "lorad": "Anthro Realism",
+    "lorae": "SB",
+    "loraf": "Moaxx", # moawxx
+    "lorag": "Fluid Enhancer",
 }
 DEFAULT_LORA_A_STRENGTH = 0.0
 DEFAULT_LORA_B_STRENGTH = 0.0
+DEFAULT_LORA_C_STRENGTH = 0.0
+DEFAULT_LORA_D_STRENGTH = 0.0
+DEFAULT_LORA_E_STRENGTH = 0.0
+DEFAULT_LORA_F_STRENGTH = 0.0
+DEFAULT_LORA_G_STRENGTH = 0.0
 # Some `diffusion_model.blocks.*` checkpoints store SwiGLU's fc1 gate/value halves in the opposite order
 # diffusers expects. Leave off first; if the LoRA's effect looks inverted/broken rather than just weak or
 # strong, set H3_LORA_SWAP_FC1=1 and compare.
@@ -529,6 +544,11 @@ def generate(
     upsample=False,
     lora_a_strength=DEFAULT_LORA_A_STRENGTH,
     lora_b_strength=DEFAULT_LORA_B_STRENGTH,
+    lora_c_strength=DEFAULT_LORA_C_STRENGTH,
+    lora_d_strength=DEFAULT_LORA_D_STRENGTH,
+    lora_e_strength=DEFAULT_LORA_E_STRENGTH,
+    lora_f_strength=DEFAULT_LORA_F_STRENGTH,
+    lora_g_strength=DEFAULT_LORA_G_STRENGTH,
     progress=gr.Progress(track_tqdm=True),
 ):
     """One request through the PlagueKind graph. Every parameter but the prompt carries the default its UI
@@ -570,7 +590,7 @@ def generate(
 
     # Every UI LoRA slider gets packed into one dict here — this is the only place a new LoRA's slider value
     # needs wiring in; `_generate`, `set_adapters`, and the report line below are all keyed off this dict.
-    lora_strengths = {"loraa": float(lora_a_strength), "lorab": float(lora_b_strength)}
+    lora_strengths = {"loraa": float(lora_a_strength), "lorab": float(lora_b_strength), "lorac": float(lora_c_strength), "lorad": float(lora_d_strength), "lorae": float(lora_e_strength), "loraf": float(lora_f_strength), "lorag": float(lora_g_strength)}
 
     progress(0.1, desc=f"Denoising {int(steps)} steps at {width}x{height}, {num_frames} frames ...")
     call = (
@@ -727,6 +747,41 @@ with gr.Blocks(title="PlagueKind · MiniMax-H3") as demo:
                     step=0.05,
                     value=DEFAULT_LORA_B_STRENGTH,
                 )
+                lora_c_strength = gr.Slider(
+                   label="HM-AIO LoRA",
+                   minimum=0.0,
+                   maximum=2.0,
+                   step=0.05,
+                   value=DEFAULT_LORA_C_STRENGTH,
+                )
+                lora_d_strength = gr.Slider(
+                   label="Anthro Realism LoRA",
+                   minimum=0.0,
+                   maximum=2.0,
+                   step=0.05,
+                   value=DEFAULT_LORA_D_STRENGTH,
+                )
+                lora_e_strength = gr.Slider(
+                   label="SB LoRA",
+                   minimum=0.0,
+                   maximum=2.0,
+                   step=0.05,
+                   value=DEFAULT_LORA_E_STRENGTH,
+                )
+                lora_f_strength = gr.Slider(
+                   label="Moaxx LoRA",
+                   minimum=0.0,
+                   maximum=2.0,
+                   step=0.05,
+                   value=DEFAULT_LORA_F_STRENGTH,
+                )
+                lora_g_strength = gr.Slider(
+                   label="Fluid Enhancer LoRA",
+                   minimum=0.0,
+                   maximum=2.0,
+                   step=0.05,
+                   value=DEFAULT_LORA_G_STRENGTH,
+                )
                 schedule = gr.Dropdown(
                     label="Sigma schedule",
                     choices=list(SCHEDULES),
@@ -775,6 +830,11 @@ with gr.Blocks(title="PlagueKind · MiniMax-H3") as demo:
         upsample,
         lora_a_strength,
         lora_b_strength,
+        lora_c_strength,
+        lora_d_strength,
+        lora_e_strength,
+        lora_f_strength,
+        lora_g_strength,
     ]
 
     run.click(generate, controls, [video, report], api_name="generate")