Update app.py

app.py

@@ -21,7 +21,7 @@ for model_path in models:
     except Exception as error:
         def the_fn(txt):
             return None
-        model_functions[model_idx] = gr.Interface(fn=the_fn, inputs=gr.Textbox(), outputs=gr.Image())
+        model_functions[model_idx] = gr.Interface(fn=the_fn, inputs=gr.inputs.Textbox(), outputs=gr.outputs.Image())
     model_idx += 1
 
 def send_it_idx(idx):
@@ -51,65 +51,42 @@ def all_task_start():
     t_stamp = time.time()
     return gr.update(value=t_stamp), gr.update(value=t_stamp), gr.update(value=0)
 
-def clear_fn():
+def clear_fn(*args):
     nn = len(models)
     return (None, *[None for _ in range(nn)])
 
-with gr.Interface(title="SD Models") as my_interface:
-    with gr.Column():
-        with gr.Row():
-            with gr.Column():
-                primary_prompt = gr.Textbox(label="Prompt", default="")
-            with gr.Column():
-                run = gr.Button("Run")
-                clear_btn = gr.Button("Clear")
-        with gr.Row():
-            sd_outputs = {}
-            model_idx = 1
-            for model_path in models:
-                with gr.Column():
-                    sd_outputs[model_idx] = gr.Image(label=model_path)
-                model_idx += 1
-
-        start_box = gr.Number(visible=False)
-        end_box = gr.Number(visible=False)
-        tog_box = gr.Textbox(default=0, visible=False)
-
-        def on_change_start_box(value):
-            return all_task_end(value, end_box.value)
-
-        start_box.change(
-            on_change_start_box,
-            inputs=[start_box],
-            outputs=[start_box, tog_box],
-            every=1
-        )
-
-        primary_prompt.submit(all_task_start, None, [start_box, end_box, tog_box])
-        run.click(all_task_start, None, [start_box, end_box, tog_box])
-
-        runs_dict = {}
-        model_idx = 1
-        for model_path in models:
-            runs_dict[model_idx] = run.click(
-                send_it_idx(model_idx),
-                inputs=[primary_prompt],
-                outputs=[sd_outputs[model_idx]]
-            )
-            model_idx += 1
-
-        clear_btn.click(
-            clear_fn,
-            None,
-            [primary_prompt, *list(sd_outputs.values())],
-            cancels=[*list(runs_dict.values())]
-        )
-        tog_box.change(
-            clear_it,
-            tog_box,
-            tog_box,
-            cancels=[*list(runs_dict.values())]
-        )
-
-my_interface.queue(concurrency_count=1)
-my_interface.launch()
+def run_model(prompt, model_idx):
+    return send_it_idx(model_idx)(prompt)
+
+# Create UI components
+def build_ui():
+    primary_prompt = gr.inputs.Textbox(label="Prompt", default="")
+    run = gr.Button("Run")
+    clear_btn = gr.Button("Clear")
+
+    sd_outputs = {}
+    for idx, model_path in enumerate(models, start=1):
+        sd_outputs[idx] = gr.outputs.Image(label=model_path)
+
+    start_box = gr.inputs.Number(visible=False)
+    end_box = gr.inputs.Number(visible=False)
+    tog_box = gr.inputs.Textbox(default=0, visible=False)
+
+    return primary_prompt, run, clear_btn, sd_outputs, start_box, end_box, tog_box
+
+# Define the function that combines all components
+def main_function(prompt):
+    return {idx: run_model(prompt, idx) for idx in range(1, len(models) + 1)}
+
+primary_prompt, run, clear_btn, sd_outputs, start_box, end_box, tog_box = build_ui()
+
+# Setup the interface
+interface = gr.Interface(
+    fn=main_function,
+    inputs=[primary_prompt],
+    outputs=list(sd_outputs.values())
+)
+
+# Additional settings (e.g., queue, launch)
+interface.queue(concurrency_count=1)
+interface.launch()