Update app.py

app.py

@@ -23,10 +23,10 @@ def text_it(inputs, text_gen=text_gen):
     return text_gen(inputs)
 
 
-def set_model(current_model_name):
+def set_model(current_model_index):
     global current_model
-    current_model = next((m for m in models if m["name"] == current_model_name), None)
-    return gr.outputs.Label(current_model["name"])
+    current_model = models[current_model_index]
+    return gr.update(label=f"{current_model['name']}")
 
 
 def send_it(inputs, model_choice):
@@ -34,6 +34,34 @@ def send_it(inputs, model_choice):
     return proc(inputs)
 
 
+css = """
+<style>
+    .searchable-dropdown {
+        position: relative;
+    }
+
+    .searchable-dropdown input {
+        width: 100%;
+        padding: 0.5rem;
+        border-radius: 5px;
+        border: 1px solid #ccc;
+    }
+
+    .searchable-dropdown select {
+        width: 100%;
+        padding: 0.5rem;
+        border-radius: 5px;
+        border: 1px solid #ccc;
+        position: absolute;
+        top: 0;
+        left: 0;
+        opacity: 0;
+        pointer-events: none;
+    }
+</style>
+"""
+
+
 def make_dropdown_searchable(dropdown_id):
     script = f"""
     <script>
@@ -41,7 +69,7 @@ def make_dropdown_searchable(dropdown_id):
             const input = document.getElementById(dropdownId);
             input.addEventListener("input", function() {{
                 const value = this.value.toLowerCase();
-                const options = document.querySelectorAll(`#${dropdownId} option`);
+                const options = Array.from(document.querySelectorAll(`#${dropdownId} option`));
                 let found = false;
                 for (let i = 0; i < options.length; i++) {{
                     const option = options[i];
@@ -67,63 +95,27 @@ def make_dropdown_searchable(dropdown_id):
     return gr.outputs.HTML(script)
 
 
-css = """"""
-
-iface = gr.Interface(
+with 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])
-
+) as iface:
+    model_dropdown = gr.outputs.Dropdown(
+        label="Choose Model",
+        choices=[m["name"] for m in models],
+        type="index",
+        value=current_model["name"],
+        interactive=True,
+        component_id="model_dropdown"
+    )
 
-def generate_images():
-    return [send_it(magic_inputs[i].value, model_name1.value) for i in range(len(magic_inputs))]
+    make_dropdown_searchable("model_dropdown")
 
+    iface.inputs.insert(0, model_dropdown)
 
-see_prompts.set_action(generate_prompts)
-run.set_action(generate_images)
+    iface.layout["top"] = ["inputs", "outputs", "model_dropdown"]
+    iface.layout["heights"] = ["200px", "auto", "40px"]
 
-iface.launch(inline=True, show=True)
+    iface.launch()