Update app.py

app.py

@@ -79,7 +79,7 @@ def inpaint(
     return result.convert("RGBA"), prompt, seed
 
 # ========================
-# API CON FIRMA ESTÁTICA
+# API Y UI UNIFICADAS
 # ========================
 def inpaint_api(
     image,
@@ -88,8 +88,8 @@ def inpaint_api(
     seed: int,
     num_inference_steps: int,
     guidance_scale: int,
-    flux_keywords: list[str] = None,
-    loras_selected: list[Tuple[str, float]] = None
+    flux_keywords: List[str] = None,
+    loras_selected: List[Tuple[str, float]] = None
 ):
     flux_keywords = flux_keywords or []
     loras_selected = loras_selected or []
@@ -129,53 +129,7 @@ def inpaint_api(
     )
 
 # ========================
-# ADAPTADOR UI -> API
-# ========================
-def ui_to_api_adapter(*ui_args):
-    # Desempaquetar argumentos del UI
-    main_inputs = ui_args[:6]  # image, mask, prompt, seed, steps, guidance
-    flux_keyword_values = ui_args[6:6 + len(flux_keywords_available)]
-    lora_args = ui_args[6 + len(flux_keywords_available):]
-
-    # Construir lista de keywords seleccionadas
-    flux_keywords = [
-        kw for kw, selected in zip(flux_keywords_available, flux_keyword_values) if selected
-    ]
-
-    # Construir lista de (nombre, peso) para LoRAs seleccionados
-    loras_selected = []
-    for i, lora_obj in enumerate(loras):
-        enabled = lora_args[i * 2]
-        weight = lora_args[i * 2 + 1]
-        if enabled:
-            loras_selected.append((lora_obj.display_name, weight))
-
-    return inpaint_api(*main_inputs, flux_keywords=flux_keywords, loras_selected=loras_selected)
-
-# ========================
-# CREACIÓN DE COMPONENTES
-# ========================
-def toggle_input(checked):
-    return gr.Number(interactive=checked)
-
-def create_flux_keyword_components():
-    components = []
-    for keyword in flux_keywords_available:
-        components.append(gr.Checkbox(label=keyword, value=False))
-    return components
-
-def create_lora_components():
-    components = []
-    for lora in loras:
-        checkbox = gr.Checkbox(label=lora.display_name, value=False, info=lora.note or "")
-        number_input = gr.Number(value=0.5, minimum=-1.0, maximum=1.0, step=0.1,
-                                 interactive=False, label="Adapter weight")
-        checkbox.change(fn=toggle_input, inputs=[checkbox], outputs=[number_input], api_name=False)
-        components.append((checkbox, number_input))
-    return components
-
-# ========================
-# UI
+# UI DIRECTA A inpaint_api
 # ========================
 with gr.Blocks(title="Flux.1 Fill dev Inpainting with LoRAs", theme=gr.themes.Soft()) as demo:
     with gr.Row():
@@ -187,12 +141,17 @@ with gr.Blocks(title="Flux.1 Fill dev Inpainting with LoRAs", theme=gr.themes.So
             guidance_scale_input = gr.Number(label="Guidance scale", value=28, interactive=True)
 
             gr.Markdown("### Flux Keywords")
-            flux_keyword_components = create_flux_keyword_components()
-
-            lora_components = []
-            if loras:
-                gr.Markdown("### Available LoRAs")
-                lora_components = create_lora_components()
+            flux_keywords_input = gr.CheckboxGroup(choices=flux_keywords_available, label="Flux Keywords")
+
+            gr.Markdown("### Available LoRAs")
+            lora_names = [l.display_name for l in loras]
+            loras_selected_input = gr.Dataframe(
+                headers=["LoRA", "Weight"],
+                value=[[name, 0.5] for name in lora_names],
+                type="array",
+                interactive=True,
+                label="LoRA selection (set Weight to 0 to disable)"
+            )
 
         with gr.Column(scale=3):
             image_input = gr.Image(label="Image", type="pil")
@@ -204,10 +163,8 @@ with gr.Blocks(title="Flux.1 Fill dev Inpainting with LoRAs", theme=gr.themes.So
             used_prompt_box = gr.Text(label="Used prompt", lines=4)
             used_seed_box = gr.Number(label="Used seed")
 
-    flat_lora_components = [comp for pair in lora_components for comp in pair]
-
     run_btn.click(
-        fn=ui_to_api_adapter,
+        fn=inpaint_api,
         inputs=[
             image_input,
             mask_input,
@@ -215,7 +172,9 @@ with gr.Blocks(title="Flux.1 Fill dev Inpainting with LoRAs", theme=gr.themes.So
             seed_slider,
             num_inference_steps_input,
             guidance_scale_input,
-        ] + flux_keyword_components + flat_lora_components,
+            flux_keywords_input,
+            loras_selected_input
+        ],
         outputs=[result_image, used_prompt_box, used_seed_box],
         api_name="inpaint"
     )