Update app.py

app.py

@@ -1,14 +1,14 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
 
-# List of available models (custom models included)
-model_options = {
-    "Stable Diffusion 2": "stabilityai/stable-diffusion-2",
-    "Stable Diffusion 1.5": "runwayml/stable-diffusion-v1-5",
-    "DALL-E Mini": "dalle-mini/dalle-mini",
-    "FLUX 1.0 (black-forest-labs)": "black-forest-labs/FLUX.1-dev",
-    "Pony Diffusion V6": "John6666/pony-diffusion-v6-xl-sdxl-spo"
-}
+
+
+def update_model(selected_model, use_textbox, custom_model):
+    if use_textbox:
+        return custom_model  # Return the custom model from textbox
+    else:
+        return selected_model  # Return the selected model from dropdown
+
 
 # Initialize Hugging Face Inference Client
 def get_client(model_name):
@@ -22,8 +22,19 @@ def generate_image(prompt, model_name):
 
 # Gradio interface
 with gr.Blocks() as demo:
+    # List of available models (custom models included)
+    model_options = {
+        "Stable Diffusion 2": "stabilityai/stable-diffusion-2",
+        "Stable Diffusion 1.5": "runwayml/stable-diffusion-v1-5",
+        "DALL-E Mini": "dalle-mini/dalle-mini",
+        "FLUX 1.0 (black-forest-labs)": "black-forest-labs/FLUX.1-dev",
+        "Pony Diffusion V6": "John6666/pony-diffusion-v6-xl-sdxl-spo"
+    }
+
     gr.Markdown("# Text to Image Generator using Hugging Face Inference Client")
-    
+    with gr.Row():
+        use_textbox = gr.Checkbox(label="Use Custom Model", value=False)
+        
     with gr.Row():
         # Input for text prompt
         prompt_input = gr.Textbox(label="Enter your prompt", placeholder="Describe the image you want...")
@@ -38,10 +49,39 @@ with gr.Blocks() as demo:
                 choices=list(model_options.keys()),  # Display model names
                 value="FLUX 1.0 (black-forest-labs)",  # Default model
             )
-                       
+        custom_model_textbox = gr.Textbox(label="Enter Custom Model", placeholder="Type your model name here")
+
+        output = gr.Textbox(label="Selected Model", interactive=False)
+
+
+
+        with gr.Row():
+            submit_button = gr.Button("Submit Option")
         with gr.Column():
             # Image output
             image_output = gr.Image(label="Generated Image")
+
+
+   # Event to update the selected model based on the checkbox
+    use_textbox.change(
+        lambda checked: model_dropdown.update(visible=not checked, value=None),
+        inputs=[use_textbox],
+        outputs=[model_dropdown]
+    )
+    
+    use_textbox.change(
+        lambda checked: custom_model_textbox.update(visible=checked, value=""),
+        inputs=[use_textbox],
+        outputs=[custom_model_textbox]
+    )
+    
+    # Button to confirm selection
+    
+    submit_button.click(
+        update_model,
+        inputs=[model_dropdown, use_textbox, custom_model_textbox],
+        outputs=[output]
+    ) 
     
     # Link the button click to the function
     generate_button.click(generate_image, inputs=[prompt_input, model_dropdown], outputs=image_output)