Update app.py

app.py

@@ -69,66 +69,38 @@ def make_dropdown_searchable(dropdown_id):
 
 css = """"""
 
-with gr.Blocks(css=css) as myface:
-    make_dropdown_searchable("model_dropdown")
-
-    with gr.Row():
-        with gr.Row():
-            input_text = gr.Textbox(label="Prompt idea", lines=1)
-            # Model selection dropdown
-            model_name1 = gr.Dropdown(
-                label="Choose Model",
-                choices=[m["name"] for m in models],
-                type="index",
-                value=current_model["name"],
-                interactive=True,
-                key="model_dropdown"
-            )
-        with gr.Row():
-            see_prompts = gr.Button("Generate Prompts")
-            run = gr.Button("Generate Images", variant="primary")
-    with gr.Tab("Main"):
-        with gr.Row():
-            output1 = gr.Image(label=f"{current_model['name']}")
-            output2 = gr.Image(label=f"{current_model['name']}")
-            output3 = gr.Image(label=f"{current_model['name']}")
-            output4 = gr.Image(label=f"{current_model['name']}")
-        with gr.Row():
-            magic1 = gr.Textbox(lines=4)
-            magic2 = gr.Textbox(lines=4)
-            magic3 = gr.Textbox(lines=4)
-            magic4 = gr.Textbox(lines=4)
-
-        with gr.Row():
-            output5 = gr.Image(label=f"{current_model['name']}")
-            output6 = gr.Image(label=f"{current_model['name']}")
-            output7 = gr.Image(label=f"{current_model['name']}")
-            output8 = gr.Image(label=f"{current_model['name']}")
-        with gr.Row():
-            magic5 = gr.Textbox(lines=4)
-            magic6 = gr.Textbox(lines=4)
-            magic7 = gr.Textbox(lines=4)
-            magic8 = gr.Textbox(lines=4)
-
-    model_name1.change(set_model, inputs=model_name1, outputs=[output1, output2, output3, output4, output5, output6, output7, output8])
-
-    run.click(send_it, inputs=[magic1, model_name1], outputs=[output1])
-    run.click(send_it, inputs=[magic2, model_name1], outputs=[output2])
-    run.click(send_it, inputs=[magic3, model_name1], outputs=[output3])
-    run.click(send_it, inputs=[magic4, model_name1], outputs=[output4])
-    run.click(send_it, inputs=[magic5, model_name1], outputs=[output5])
-    run.click(send_it, inputs=[magic6, model_name1], outputs=[output6])
-    run.click(send_it, inputs=[magic7, model_name1], outputs=[output7])
-    run.click(send_it, inputs=[magic8, model_name1], outputs=[output8])
-
-    see_prompts.click(text_it, inputs=[input_text], outputs=[magic1])
-    see_prompts.click(text_it, inputs=[input_text], outputs=[magic2])
-    see_prompts.click(text_it, inputs=[input_text], outputs=[magic3])
-    see_prompts.click(text_it, inputs=[input_text], outputs=[magic4])
-    see_prompts.click(text_it, inputs=[input_text], outputs=[magic5])
-    see_prompts.click(text_it, inputs=[input_text], outputs=[magic6])
-    see_prompts.click(text_it, inputs=[input_text], outputs=[magic7])
-    see_prompts.click(text_it, inputs=[input_text], outputs=[magic8])
-
-myface.queue(concurrency_count=200)
-myface.launch(inline=True, show_api=False, max_threads=400)
+with gr.Interface(blocks=True, css=css) as iface:
+    make_dropdown_searchable(iface.inputs[1].component_id)  # Make the second component searchable (model dropdown)
+
+    input_text = iface.inputs[0]
+    model_name1 = iface.inputs[1]
+    see_prompts = iface.buttons[0]
+    run = iface.buttons[1]
+    outputs = iface.outputs
+
+    iface.title = "Image Generation"
+
+    def generate_prompt(btn):
+        prompt_text = input_text.value
+        outputs[:4].clear()
+        outputs[4:8].clear()
+        for i in range(4):
+            magic_input = outputs[i].owner.inputs[0]
+            magic_input.value = text_it(prompt_text)
+
+    def generate_images(btn):
+        for i in range(4):
+            magic_input = outputs[i].owner.inputs[0]
+            magic_input_value = magic_input.value
+            model_choice = model_name1.value
+            proc = models2[model_choice]
+            outputs[i].update(send_it(magic_input_value, model_choice))
+
+    see_prompts.set_action(generate_prompt)
+    run.set_action(generate_images)
+
+    model_name1.choices = [m["name"] for m in models]
+    model_name1.default = current_model["name"]
+    model_name1.changed(set_model)
+
+    iface.launch(inline=True, show=True)