Update app.py

app.py

@@ -475,6 +475,7 @@ def generate_video(
     flow_shift=6.0,
     frame_multiplier=16,
     safe_mode=False,
+    # original 3
     lora1_preset="(None)",
     lora1_high_scale=1.0,
     lora1_low_scale=1.0,
@@ -484,11 +485,12 @@ def generate_video(
     lora3_preset="(None)",
     lora3_high_scale=1.0,
     lora3_low_scale=1.0,
-    my_repo_link="",
-    my_file_high="(None)",
-    my_high_scale=1.0,
-    my_file_low="(None)",
-    my_low_scale=1.0,
+    # my 5
+    my1_repo="", my1_high="(None)", my1_hs=1.0, my1_low="(None)", my1_ls=1.0,
+    my2_repo="", my2_high="(None)", my2_hs=1.0, my2_low="(None)", my2_ls=1.0,
+    my3_repo="", my3_high="(None)", my3_hs=1.0, my3_low="(None)", my3_ls=1.0,
+    my4_repo="", my4_high="(None)", my4_hs=1.0, my4_low="(None)", my4_ls=1.0,
+    my5_repo="", my5_high="(None)", my5_hs=1.0, my5_low="(None)", my5_ls=1.0,
     video_component=True,
     progress=gr.Progress(track_tqdm=True),
 ):
@@ -507,13 +509,11 @@ def generate_video(
         {"preset": lora1_preset, "high_scale": lora1_high_scale, "low_scale": lora1_low_scale},
         {"preset": lora2_preset, "high_scale": lora2_high_scale, "low_scale": lora2_low_scale},
         {"preset": lora3_preset, "high_scale": lora3_high_scale, "low_scale": lora3_low_scale},
-        {
-            "repo": my_repo_link,
-            "file_high": my_file_high,
-            "file_low": my_file_low,
-            "high_scale": my_high_scale,
-            "low_scale": my_low_scale,
-        },
+        {"repo": my1_repo, "file_high": my1_high, "file_low": my1_low, "high_scale": my1_hs, "low_scale": my1_ls},
+        {"repo": my2_repo, "file_high": my2_high, "file_low": my2_low, "high_scale": my2_hs, "low_scale": my2_ls},
+        {"repo": my3_repo, "file_high": my3_high, "file_low": my3_low, "high_scale": my3_hs, "low_scale": my3_ls},
+        {"repo": my4_repo, "file_high": my4_high, "file_low": my4_low, "high_scale": my4_hs, "low_scale": my4_ls},
+        {"repo": my5_repo, "file_high": my5_high, "file_low": my5_low, "high_scale": my5_hs, "low_scale": my5_ls},
     ]
 
     video_path, task_n = run_inference(
@@ -540,6 +540,14 @@ def generate_video(
     return (video_path if video_component else None), video_path, current_seed
 
 
+def _load_repo_files(repo_link):
+    files = lora_loader.list_loras_from_repo(repo_link)
+    return (
+        gr.update(choices=files, value="(None)"),
+        gr.update(choices=files, value="(None)"),
+    )
+
+
 CSS = """
 #hidden-timestamp {
     opacity: 0;
@@ -591,59 +599,87 @@ with gr.Blocks(delete_cache=(3600, 10800)) as demo:
                 )
                 flow_shift_slider = gr.Slider(minimum=0.5, maximum=15.0, step=0.1, value=3.0, label="Flow Shift")
 
-                gr.Markdown("### 1) Original LoRA (NSFW) + strength")
+                gr.Markdown("### 1) Original LoRA x3 + strength")
                 with gr.Row():
-                    lora1_preset = gr.Dropdown(choices=LORA_CHOICES, value="(None)", label="LoRA 1", filterable=True)
+                    lora1_preset = gr.Dropdown(choices=LORA_CHOICES, value="(None)", label="Original LoRA 1", filterable=True)
                     lora1_high_scale = gr.Slider(0.0, 2.0, value=1.0, step=0.05, label="High strength")
                     lora1_low_scale = gr.Slider(0.0, 2.0, value=1.0, step=0.05, label="Low strength")
                 with gr.Row():
-                    lora2_preset = gr.Dropdown(choices=LORA_CHOICES, value="(None)", label="LoRA 2", filterable=True)
+                    lora2_preset = gr.Dropdown(choices=LORA_CHOICES, value="(None)", label="Original LoRA 2", filterable=True)
                     lora2_high_scale = gr.Slider(0.0, 2.0, value=1.0, step=0.05, label="High strength")
                     lora2_low_scale = gr.Slider(0.0, 2.0, value=1.0, step=0.05, label="Low strength")
                 with gr.Row():
-                    lora3_preset = gr.Dropdown(choices=LORA_CHOICES, value="(None)", label="LoRA 3", filterable=True)
+                    lora3_preset = gr.Dropdown(choices=LORA_CHOICES, value="(None)", label="Original LoRA 3", filterable=True)
                     lora3_high_scale = gr.Slider(0.0, 2.0, value=1.0, step=0.05, label="High strength")
                     lora3_low_scale = gr.Slider(0.0, 2.0, value=1.0, step=0.05, label="Low strength")
 
-                gr.Markdown("### 2) My LoRA from HF repo link")
-                gr.Markdown(
-                    "Paste repo link, click **Load files**, then pick High/Low file.\n\n"
-                    "Example: `username/my-lora-repo` or `https://huggingface.co/username/my-lora-repo`"
-                )
+                gr.Markdown("### 2) My LoRA x5 from HF repo + strength")
+                gr.Markdown("For each slot: paste repo → Load files → pick High/Low → set strength. Example: `username/my-loras`")
+
+                # --- My LoRA 1 ---
+                gr.Markdown("**My LoRA 1**")
+                with gr.Row():
+                    my1_repo = gr.Textbox(label="Repo link", placeholder="username/repo", lines=1, scale=4)
+                    my1_load = gr.Button("Load files", scale=1)
                 with gr.Row():
-                    my_repo_link = gr.Textbox(
-                        label="HF repo link",
-                        placeholder="username/my-loras",
-                        lines=1,
-                        scale=4,
-                    )
-                    load_repo_btn = gr.Button("Load files", scale=1)
+                    my1_high = gr.Dropdown(choices=["(None)"], value="(None)", label="High file", filterable=True)
+                    my1_hs = gr.Slider(0.0, 2.0, value=1.0, step=0.05, label="High strength")
+                with gr.Row():
+                    my1_low = gr.Dropdown(choices=["(None)"], value="(None)", label="Low file", filterable=True)
+                    my1_ls = gr.Slider(0.0, 2.0, value=1.0, step=0.05, label="Low strength")
+                my1_load.click(fn=_load_repo_files, inputs=[my1_repo], outputs=[my1_high, my1_low])
 
+                # --- My LoRA 2 ---
+                gr.Markdown("**My LoRA 2**")
                 with gr.Row():
-                    my_file_high = gr.Dropdown(
-                        choices=["(None)"], value="(None)",
-                        label="My High LoRA file", filterable=True,
-                    )
-                    my_high_scale = gr.Slider(0.0, 2.0, value=1.0, step=0.05, label="My High strength")
+                    my2_repo = gr.Textbox(label="Repo link", placeholder="username/repo", lines=1, scale=4)
+                    my2_load = gr.Button("Load files", scale=1)
                 with gr.Row():
-                    my_file_low = gr.Dropdown(
-                        choices=["(None)"], value="(None)",
-                        label="My Low LoRA file", filterable=True,
-                    )
-                    my_low_scale = gr.Slider(0.0, 2.0, value=1.0, step=0.05, label="My Low strength")
-
-                def _load_my_repo_files(repo_link):
-                    files = lora_loader.list_loras_from_repo(repo_link)
-                    return (
-                        gr.update(choices=files, value="(None)"),
-                        gr.update(choices=files, value="(None)"),
-                    )
-
-                load_repo_btn.click(
-                    fn=_load_my_repo_files,
-                    inputs=[my_repo_link],
-                    outputs=[my_file_high, my_file_low],
-                )
+                    my2_high = gr.Dropdown(choices=["(None)"], value="(None)", label="High file", filterable=True)
+                    my2_hs = gr.Slider(0.0, 2.0, value=1.0, step=0.05, label="High strength")
+                with gr.Row():
+                    my2_low = gr.Dropdown(choices=["(None)"], value="(None)", label="Low file", filterable=True)
+                    my2_ls = gr.Slider(0.0, 2.0, value=1.0, step=0.05, label="Low strength")
+                my2_load.click(fn=_load_repo_files, inputs=[my2_repo], outputs=[my2_high, my2_low])
+
+                # --- My LoRA 3 ---
+                gr.Markdown("**My LoRA 3**")
+                with gr.Row():
+                    my3_repo = gr.Textbox(label="Repo link", placeholder="username/repo", lines=1, scale=4)
+                    my3_load = gr.Button("Load files", scale=1)
+                with gr.Row():
+                    my3_high = gr.Dropdown(choices=["(None)"], value="(None)", label="High file", filterable=True)
+                    my3_hs = gr.Slider(0.0, 2.0, value=1.0, step=0.05, label="High strength")
+                with gr.Row():
+                    my3_low = gr.Dropdown(choices=["(None)"], value="(None)", label="Low file", filterable=True)
+                    my3_ls = gr.Slider(0.0, 2.0, value=1.0, step=0.05, label="Low strength")
+                my3_load.click(fn=_load_repo_files, inputs=[my3_repo], outputs=[my3_high, my3_low])
+
+                # --- My LoRA 4 ---
+                gr.Markdown("**My LoRA 4**")
+                with gr.Row():
+                    my4_repo = gr.Textbox(label="Repo link", placeholder="username/repo", lines=1, scale=4)
+                    my4_load = gr.Button("Load files", scale=1)
+                with gr.Row():
+                    my4_high = gr.Dropdown(choices=["(None)"], value="(None)", label="High file", filterable=True)
+                    my4_hs = gr.Slider(0.0, 2.0, value=1.0, step=0.05, label="High strength")
+                with gr.Row():
+                    my4_low = gr.Dropdown(choices=["(None)"], value="(None)", label="Low file", filterable=True)
+                    my4_ls = gr.Slider(0.0, 2.0, value=1.0, step=0.05, label="Low strength")
+                my4_load.click(fn=_load_repo_files, inputs=[my4_repo], outputs=[my4_high, my4_low])
+
+                # --- My LoRA 5 ---
+                gr.Markdown("**My LoRA 5**")
+                with gr.Row():
+                    my5_repo = gr.Textbox(label="Repo link", placeholder="username/repo", lines=1, scale=4)
+                    my5_load = gr.Button("Load files", scale=1)
+                with gr.Row():
+                    my5_high = gr.Dropdown(choices=["(None)"], value="(None)", label="High file", filterable=True)
+                    my5_hs = gr.Slider(0.0, 2.0, value=1.0, step=0.05, label="High strength")
+                with gr.Row():
+                    my5_low = gr.Dropdown(choices=["(None)"], value="(None)", label="Low file", filterable=True)
+                    my5_ls = gr.Slider(0.0, 2.0, value=1.0, step=0.05, label="Low strength")
+                my5_load.click(fn=_load_repo_files, inputs=[my5_repo], outputs=[my5_high, my5_low])
 
                 play_result_video = gr.Checkbox(label="Display result", value=True, interactive=True)
 
@@ -664,10 +700,16 @@ with gr.Blocks(delete_cache=(3600, 10800)) as demo:
         guidance_scale_input, guidance_scale_2_input, seed_input, randomize_seed_checkbox,
         quality_slider, scheduler_dropdown, flow_shift_slider, frame_multi,
         safe_mode_checkbox,
+        # original 3
         lora1_preset, lora1_high_scale, lora1_low_scale,
         lora2_preset, lora2_high_scale, lora2_low_scale,
         lora3_preset, lora3_high_scale, lora3_low_scale,
-        my_repo_link, my_file_high, my_high_scale, my_file_low, my_low_scale,
+        # my 5
+        my1_repo, my1_high, my1_hs, my1_low, my1_ls,
+        my2_repo, my2_high, my2_hs, my2_low, my2_ls,
+        my3_repo, my3_high, my3_hs, my3_low, my3_ls,
+        my4_repo, my4_high, my4_hs, my4_low, my4_ls,
+        my5_repo, my5_high, my5_hs, my5_low, my5_ls,
         play_result_video,
     ]