Update app.py

app.py

@@ -26,7 +26,7 @@ def text_it(inputs, text_gen=text_gen):
 def set_model(current_model_name):
     global current_model
     current_model = next((m for m in models if m["name"] == current_model_name), None)
-    return gr.update(label=f"{current_model['name']}")
+    return gr.outputs.Label(current_model["name"])
 
 
 def send_it(inputs, model_choice):
@@ -64,43 +64,57 @@ def make_dropdown_searchable(dropdown_id):
         makeDropdownSearchable("{dropdown_id}");
     </script>
     """
-    return gr.HTML(script)
+    return gr.outputs.HTML(script)
 
 
 css = """"""
 
-with gr.Interface(blocks=True, css=css) as iface:
-    make_dropdown_searchable(iface.inputs[1].component_id)  # Make the second component searchable (model dropdown)
+with gr.Interface(text_it, inputs="text", outputs="text", css=css, capture_session=True) as iface:
+    make_dropdown_searchable(iface.inputs[0].component_id)  # Make the model dropdown searchable
 
     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)
+    model_name1 = gr.inputs.Dropdown(
+        label="Choose Model",
+        choices=[m["name"] for m in models],
+        type="index",
+        value=current_model["name"],
+        interactive=True,
+    )
+    see_prompts = gr.outputs.Button(label="Generate Prompts")
+    run = gr.outputs.Button(label="Generate Images", type="primary")
+    outputs = [
+        gr.outputs.Image(label=current_model["name"]),
+        gr.outputs.Image(label=current_model["name"]),
+        gr.outputs.Image(label=current_model["name"]),
+        gr.outputs.Image(label=current_model["name"]),
+        gr.outputs.Image(label=current_model["name"]),
+        gr.outputs.Image(label=current_model["name"]),
+        gr.outputs.Image(label=current_model["name"]),
+        gr.outputs.Image(label=current_model["name"]),
+    ]
+    magic_inputs = [
+        gr.inputs.Textbox(lines=4),
+        gr.inputs.Textbox(lines=4),
+        gr.inputs.Textbox(lines=4),
+        gr.inputs.Textbox(lines=4),
+        gr.inputs.Textbox(lines=4),
+        gr.inputs.Textbox(lines=4),
+        gr.inputs.Textbox(lines=4),
+        gr.inputs.Textbox(lines=4),
+    ]
+
+    iface.inputs = [input_text, model_name1, see_prompts, run] + magic_inputs
+    iface.outputs = outputs
+
+    def generate_prompts():
+        prompts = text_it(input_text.value)
+        for i, magic_input in enumerate(magic_inputs):
+            magic_input.update(prompts[i])
+
+    def generate_images():
+        return [send_it(magic_inputs[i].value, model_name1.value) for i in range(len(magic_inputs))]
+
+    see_prompts.set_action(generate_prompts)
     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)