Update app.py

app.py

@@ -95,8 +95,18 @@ def infer(
     height,
     guidance_scale,
     num_inference_steps,
-    latent_file,
-    ip_scale,
+    latent_file = gr.File(),  # Add latents file input
+    latent_file_2 = gr.File(),  # Add latents file input
+    latent_file_3 = gr.File(),  # Add latents file input
+    latent_file_4 = gr.File(),  # Add latents file input
+    latent_file_5 = gr.File(),  # Add latents file input
+    text_scale: float = 1.0,
+    ip_scale: float = 1.0,
+    latent_file_1_scale: float = 1.0,
+    latent_file_2_scale: float = 1.0,
+    latent_file_3_scale: float = 1.0,
+    latent_file_4_scale: float = 1.0,
+    latent_file_5_scale: float = 1.0,
     image_encoder_path,
     progress=gr.Progress(track_tqdm=True),
 ):
@@ -194,9 +204,49 @@ with gr.Blocks(theme=gr.themes.Origin(),css=css) as demo:
                 container=False,
             )
             run_button = gr.Button("Run", scale=0, variant="primary")
-        result = gr.Image(label="Result", show_label=False)
+            result = gr.Image(label="Result", show_label=False)
         with gr.Accordion("Advanced Settings", open=True):
-            latent_file = gr.File(label="Image File (optional)")
+        with gr.Row():
+            latent_file = gr.File(label="Image Prompt (Required)")
+            file_1_strength =  gr.Slider(
+                label="Img 1 %",
+                minimum=0.0,
+                maximum=16.0,
+                step=0.01,
+                value=1.0,
+            )
+            latent_file_2 = gr.File(label="Image Prompt 2 (Optional)")
+            file_2_strength =  gr.Slider(
+                label="Img 2 %",
+                minimum=0.0,
+                maximum=16.0,
+                step=0.01,
+                value=1.0,
+            )
+            latent_file_3 = gr.File(label="Image Prompt 3 (Optional)")
+            file_3_strength =  gr.Slider(
+                label="Img 3 %",
+                minimum=0.0,
+                maximum=16.0,
+                step=0.01,
+                value=1.0,
+            )
+            latent_file_4 = gr.File(label="Image Prompt 4 (Optional)")
+            file_4_strength =  gr.Slider(
+                label="Img 4 %",
+                minimum=0.0,
+                maximum=16.0,
+                step=0.01,
+                value=1.0,
+                )
+            latent_file_5 = gr.File(label="Image Prompt 5 (Optional)")
+                file_5_strength =  gr.Slider(
+                label="Img 5 %",
+                minimum=0.0,
+                maximum=16.0,
+                step=0.01,
+                value=1.0,
+            )
             image_encoder_path = gr.Dropdown(
                 ["google/siglip-so400m-patch14-384", "laion/CLIP-ViT-H-14-laion2B-s32B-b79K"], 
                 label="CLIP Model", 
@@ -272,7 +322,17 @@ with gr.Blocks(theme=gr.themes.Origin(),css=css) as demo:
             guidance_scale,
             num_inference_steps,
             latent_file,
+            latent_file_2,
+            latent_file_3,
+            latent_file_4,
+            latent_file_5,  
+            text_strength,
             ip_scale,
+            file_1_strength,
+            file_2_strength,
+            file_3_strength,
+            file_4_strength,
+            file_5_strength,
             image_encoder_path,
         ],
         outputs=[result, expanded_prompt_output],