Update app.py

app.py

@@ -69,52 +69,61 @@ def make_dropdown_searchable(dropdown_id):
 
 css = """"""
 
-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 = 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)
-
-    iface.launch(inline=True, show=True)
+iface = gr.Interface(
+    fn=text_it,
+    inputs="text",
+    outputs="text",
+    css=css,
+    capture_session=True
+)
+
+make_dropdown_searchable(iface.inputs[0].component_id)  # Make the model dropdown searchable
+
+input_text = iface.inputs[0]
+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),
+]
+
+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)
+
+iface.launch(inline=True, show=True)