Update app.py

app.py

@@ -7,16 +7,14 @@ import time
 from PIL import Image
 import json
 
-# Project by Nymbo
-
-# Base API URL for Hugging Face inference
+# Base API URL for Hugging Face inference 
 API_URL = "https://api-inference.huggingface.co/models/black-forest-labs/FLUX.1-dev"
-# Retrieve the API token from environment variables
 API_TOKEN = os.getenv("HF_READ_TOKEN")
 headers = {"Authorization": f"Bearer {API_TOKEN}"}
-# Timeout for requests
 timeout = 100
 
+
+
 def query(prompt, model, custom_lora, is_negative=False, steps=35, cfg_scale=7, sampler="DPM++ 2M Karras", seed=-1, strength=0.7, width=1024, height=1024):
     # Debug log to indicate function start
     print("Starting query function...")
@@ -307,231 +305,153 @@ footer {
 }
 """
 
-print("Initializing Gradio interface...")  # Debug log
+print("Initializing Gradio interface...")
 
 # Define the Gradio interface
 with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css) as dalle:
-    # Tab for basic settings
-    with gr.Tab("Basic Settings"):
-        with gr.Row():
-            with gr.Column(elem_id="prompt-container"):
-                with gr.Row():
-                    # Textbox for user to input the prompt
-                    text_prompt = gr.Textbox(label="Prompt", placeholder="Enter a prompt here", lines=3, elem_id="prompt-text-input")
-                with gr.Row():
-                    # Textbox for custom LoRA input
-                    custom_lora = gr.Textbox(label="Custom LoRA", info="LoRA Hugging Face path (optional)", placeholder="multimodalart/vintage-ads-flux")
-                with gr.Row():
-                    # Accordion for selecting the model
-                    with gr.Accordion("Featured Models", open=True):
-                        # Textbox for searching models
-                        model_search = gr.Textbox(label="Filter Models", placeholder="Search for a featured model...", lines=1, elem_id="model-search-input")
-                        models_list = (
-                            "3D Sketchfab",
-                            "90s Anime Art",
-                            "AbsoluteReality 1.8.1",
-                            "Analog",
-                            "Analog Madness Realistic v7",
-                            "Analog Redmond",
-                            "Archfey Anime",
-                            "Ascii Art",
-                            "Brain Melt Acid Art",
-                            "Boreal",
-                            "Caricature",
-                            "Collage Flux",
-                            "Character Design",
-                            "Coloring Book Generator",
-                            "Cyborg Style XL",
-                            "Disney",
-                            "DreamPhotoGASM",
-                            "Duchaiten Real3D NSFW XL",
-                            "EpiCPhotoGasm",
-                            "Fashion Hut Modeling LoRA",
-                            "Filmgrain",
-                            "FLUX.1 [Dev]",
-                            "FLUX.1 [Schnell]",
-                            "Flux Realism LoRA",
-                            "Flux Super Realism LoRA",
-                            "Flux Uncensored",
-                            "Flux Uncensored V2",
-                            "Flux Game Assets V2",
-                            "Flux Ghibsky Illustration",
-                            "Flux Animex V2",
-                            "Flux Animeo V1",
-                            "Flux AestheticAnime",
-                            "Flux SyntheticAnime",
-                            "Flux Stickers",
-                            "Flux Koda",
-                            "Flux Tarot v1",
-                            "Flux Tarot Cards",
-                            "Flux Midjourney Anime",
-                            "Flux Logo Design",
-                            "Flux Product Ad Backdrop",
-                            "Flux Outfit Generator",
-                            "Frosting Lane Flux",
-                            "Half Illustration",
-                            "How2Draw",
-                            "Huggieverse",
-                            "Leonardo AI Style Illustration",
-                            "Little Tinies",
-                            "Lofi Cuties",
-                            "Lustly Flux Uncensored v1",
-                            "Maple Syrup",
-                            "Midjourney",
-                            "Movie Board",
-                            "NSFW Master Flux",
-                            "NSFW XL",
-                            "OpenDalle v1.1",
-                            "Perfect Lewd Fantasy",
-                            "Pixel Art Redmond",
-                            "Pixel Art XL",
-                            "Pixel Art Sprites",
-                            "Product Design",
-                            "Propaganda Poster",
-                            "Purple Dreamy",
-                            "Phantasma Anime",
-                            "PS1 Style Flux",
-                            "Redmond SDXL",
-                            "Retro Comic Flux",
-                            "Softserve Anime",
-                            "SoftPasty Flux",
-                            "Soviet Diffusion XL",
-                            "Sketched Out Manga",
-                            "Selfie Photography",
-                            "Stable Diffusion 2-1",
-                            "Stable Diffusion XL",
-                            "Stable Diffusion 3 Medium",
-                            "Stable Diffusion 3.5 Large",
-                            "Stable Diffusion 3.5 Large Turbo",
-                            "YiffyMix",
-                        )
-
-                        # Radio buttons to select the desired model
-                        model = gr.Radio(label="Select a model below", value="FLUX.1 [Schnell]", choices=models_list, interactive=True, elem_id="model-radio")
-
-                        # Filtering models based on search input
-                        def filter_models(search_term):
-                            filtered_models = [m for m in models_list if search_term.lower() in m.lower()]
-                            return gr.update(choices=filtered_models)
-
-                        # Update model list when search box is used
-                        model_search.change(filter_models, inputs=model_search, outputs=model)
-
-    # Tab for advanced settings
-    with gr.Tab("Advanced Settings"):
-        with gr.Row():
-            # Textbox for specifying elements to exclude from the image
-            negative_prompt = gr.Textbox(label="Negative Prompt", placeholder="What should not be in the image", value="(deformed, distorted, disfigured), poorly drawn, bad anatomy, wrong anatomy, extra limb, missing limb, floating limbs, (mutated hands and fingers), disconnected limbs, mutation, mutated, ugly, disgusting, blurry, amputation, misspellings, typos", lines=3, elem_id="negative-prompt-text-input")
-        with gr.Row():
-            # Slider for selecting the image width
-            width = gr.Slider(label="Width", value=1024, minimum=64, maximum=1216, step=32)
-            # Slider for selecting the image height
-            height = gr.Slider(label="Height", value=1024, minimum=64, maximum=1216, step=32)
-        with gr.Row():
-            # Slider for setting the number of sampling steps
-            steps = gr.Slider(label="Sampling steps", value=35, minimum=1, maximum=100, step=1)
-        with gr.Row():
-            # Slider for adjusting the CFG scale (guidance scale)
-            cfg = gr.Slider(label="CFG Scale", value=7, minimum=1, maximum=20, step=1)
-        with gr.Row():
-            # Slider for adjusting the transformation strength
-            strength = gr.Slider(label="Strength", value=0.7, minimum=0, maximum=1, step=0.001)
-        with gr.Row():
-            # Slider for setting the seed for reproducibility
-            seed = gr.Slider(label="Seed", value=-1, minimum=-1, maximum=1000000000, step=1)
-        with gr.Row():
-            # Radio buttons for selecting the sampling method
-            method = gr.Radio(label="Sampling method", value="DPM++ 2M Karras", choices=["DPM++ 2M Karras", "DPM++ SDE Karras", "Euler", "Euler a", "Heun", "DDIM"])
+    gr.Markdown("# AI Image Generator")
+    
+    with gr.Row():
+        with gr.Column(scale=2):
+            # Main prompt input
+            text_prompt = gr.Textbox(
+                label="Prompt",
+                placeholder="Describe what you want to create...",
+                lines=3
+            )
+            
+            # Negative prompt
+            negative_prompt = gr.Textbox(
+                label="Negative Prompt",
+                placeholder="What should not be in the image",
+                value="(deformed, distorted, disfigured), poorly drawn, bad anatomy, wrong anatomy, extra limb, missing limb, floating limbs, (mutated hands and fingers), disconnected limbs, mutation, mutated, ugly, disgusting, blurry, amputation",
+                lines=2
+            )
 
-    # Tab for image editing options
-    with gr.Tab("Image Editor"):
-        # Function to simulate a delay for processing
-        def sleep(im):
-            print("Sleeping for 5 seconds...")  # Debug log
-            time.sleep(5)
-            return [im["background"], im["layers"][0], im["layers"][1], im["composite"]]
+            # Custom LoRA input
+            custom_lora = gr.Textbox(
+                label="Custom LoRA Path (Optional)",
+                placeholder="e.g., multimodalart/vintage-ads-flux",
+                lines=1
+            )
 
-        # Function to return the composite image
-        def predict(im):
-            print("Predicting composite image...")  # Debug log
-            return im["composite"]
+        with gr.Column(scale=1):
+            # Image dimensions
+            with gr.Group():
+                gr.Markdown("### Image Settings")
+                width = gr.Slider(label="Width", value=1024, minimum=512, maximum=1216, step=64)
+                height = gr.Slider(label="Height", value=1024, minimum=512, maximum=1216, step=64)
+                
+            # Generation parameters
+            with gr.Group():
+                gr.Markdown("### Generation Parameters")
+                steps = gr.Slider(label="Steps", value=35, minimum=1, maximum=100, step=1)
+                cfg = gr.Slider(label="CFG Scale", value=7, minimum=1, maximum=20, step=0.5)
+                strength = gr.Slider(label="Strength", value=0.7, minimum=0, maximum=1, step=0.1)
+                seed = gr.Slider(label="Seed (-1 for random)", value=-1, minimum=-1, maximum=1000000000, step=1)
 
-        with gr.Blocks() as demo:
-            with gr.Row():
-                # Image editor component for user adjustments
-                im = gr.ImageEditor(
-                    type="numpy",
-                    crop_size="1:1",  # Set crop size to a square aspect ratio
+# Model selection
+            with gr.Group():
+                gr.Markdown("### Model Selection")
+                model_search = gr.Textbox(
+                    label="Search Models",
+                    placeholder="Type to filter models...",
+                    lines=1
                 )
+                
+                # Updated model list (reordered by popularity/recency)
+                models_list = [
+                    "Stable Diffusion 3.5 Large",
+                    "Stable Diffusion 3.5 Large Turbo",
+                    "Stable Diffusion XL",
+                    "FLUX.1 [Schnell]",
+                    "FLUX.1 [Dev]",
+                    "Midjourney",
+                    "DreamPhotoGASM",
+                    "Disney",
+                    "Leonardo AI Style Illustration",
+                    "AbsoluteReality 1.8.1",
+                    "Analog Redmond",
+                    "Stable Diffusion 3 Medium",
+                    "Flux Super Realism LoRA",
+                    "Flux Realism LoRA",
+                    "Selfie Photography",
+                    "Character Design",
+                    "Pixel Art XL",
+                    "3D Sketchfab",
+                    "Anime Collection", # Group of anime-related models
+                    "Flux Animex V2",
+                    "Flux Animeo V1",
+                    "Flux AestheticAnime",
+                    "90s Anime Art",
+                    "Softserve Anime",
+                    "Artistic Styles", # Group of artistic style models
+                    "Brain Melt Acid Art",
+                    "Retro Comic Flux",
+                    "Purple Dreamy",
+                    "SoftPasty Flux",
+                    "Specialized", # Group of specialized models
+                    "Flux Logo Design",
+                    "Product Design",
+                    "Propaganda Poster",
+                    "Movie Board",
+                    "Collage Flux",
+                    # Additional models...
+                ]
 
-    # Tab to provide information to the user
-    with gr.Tab("Information"):
-        with gr.Row():
-            # Display a sample prompt for guidance
-            gr.Textbox(label="Sample prompt", value="{prompt} | ultra detail, ultra elaboration, ultra quality, perfect.")
+                model = gr.Radio(
+                    label="Select Model",
+                    choices=models_list,
+                    value="Stable Diffusion 3.5 Large",
+                    interactive=True
+                )
 
-        # Accordion displaying featured models
-        with gr.Accordion("Featured Models (WiP)", open=False):
-            gr.HTML(
-                """
-            <p><a href="https://huggingface.co/models?inference=warm&pipeline_tag=text-to-image&sort=trending">See all available models</a></p>
-            <table style="width:100%; text-align:center; margin:auto;">
-                <tr>
-                    <th>Model Name</th>
-                    <th>Typography</th>
-                    <th>Notes</th>
-                </tr>
-                <tr>
-                    <td>FLUX.1 Dev</td>
-                    <td>✅</td>
-                    <td></td>
-                </tr>
-                <tr>
-                    <td>FLUX.1 Schnell</td>
-                    <td>✅</td>
-                    <td></td>
-                </tr>
-                <tr>
-                    <td>Stable Diffusion 3.5 Large</td>
-                    <td>✅</td>
-                    <td></td>
-                </tr>
-            </table>
-            """
-            )
+                def filter_models(search_term):
+                    filtered_models = [m for m in models_list if search_term.lower() in m.lower()]
+                    return gr.update(choices=filtered_models)
 
-        # Accordion providing an overview of advanced settings
-        with gr.Accordion("Advanced Settings Overview", open=False):
-            gr.Markdown(
-            """
-            ## Negative Prompt
-            ###### This box is for telling the AI what you don't want in your images. Think of it as a way to avoid certain elements. For instance, if you don't want blurry images or extra limbs showing up, this is where you'd mention it.
-            ## Width & Height
-            ###### These sliders allow you to specify the resolution of your image. Default value is 1024x1024, and maximum output is 1216x1216.
-            ## Sampling Steps
-            ###### Think of this like the number of brushstrokes in a painting. A higher number can give you a more detailed picture, but it also takes a bit longer. Generally, a middle-ground number like 35 is a good balance between quality and speed.
-            ## CFG Scale
-            ###### CFG stands for "Control Free Guidance." The scale adjusts how closely the AI follows your prompt. A lower number makes the AI more creative and free-flowing, while a higher number makes it stick closely to what you asked for. If you want the AI to take fewer artistic liberties, slide this towards a higher number. Just think "Control Freak Gauge".
-            ## Sampling Method
-            ###### This is the technique the AI uses to create your image. Each option is a different approach, like choosing between pencils, markers, or paint. You don't need to worry too much about this; the default setting is usually the best choice for most users.
-            ## Strength
-            ###### This setting is a bit like the 'intensity' knob. It determines how much the AI modifies the base image it starts with. If you're looking to make subtle changes, keep this low. For more drastic transformations, turn it up.
-            ## Seed
-            ###### You can think of the seed as a 'recipe' for creating an image. If you find a seed that gives you a result you love, you can use it again to create a similar image. If you leave it at -1, the AI will generate a new seed every time.
-            ### Remember, these settings are all about giving you control over the image generation process. Feel free to experiment and see what each one does. And if you're ever in doubt, the default settings are a great place to start. Happy creating!
-            """
-            )
+                model_search.change(filter_models, inputs=model_search, outputs=model)
 
-    # Row containing the 'Run' button to trigger the image generation
+    # Generate button and output
     with gr.Row():
-        text_button = gr.Button("Run", variant='primary', elem_id="gen-button")
-    # Row for displaying the generated image output
+        generate_btn = gr.Button("Generate Image", variant="primary", size="lg")
+        
     with gr.Row():
-        image_output = gr.Image(type="pil", label="Image Output", elem_id="gallery")
+        image_output = gr.Image(
+            type="pil",
+            label="Generated Image",
+            show_label=True
+        )
+
+    # Set up the generation event
+    generate_btn.click(
+        fn=query,
+        inputs=[
+            text_prompt,
+            model,
+            custom_lora,
+            negative_prompt,
+            steps,
+            cfg,
+            method,
+            seed,
+            strength,
+            width,
+            height
+        ],
+        outputs=image_output
+    )
 
-    # Set up button click event to call the query function
-    text_button.click(query, inputs=[text_prompt, model, custom_lora, negative_prompt, steps, cfg, method, seed, strength, width, height], outputs=image_output)
+    # Add some helpful tips
+    with gr.Accordion("Tips", open=False):
+        gr.Markdown("""
+        ### Quick Tips:
+        - Use detailed descriptions for better results
+        - Adjust CFG Scale to control how closely the AI follows your prompt
+        - Higher steps generally mean better quality but slower generation
+        - Save seeds of images you like to recreate similar results
+        - Use negative prompts to avoid unwanted elements
+        """)
 
-print("Launching Gradio interface...")  # Debug log
-# Launch the Gradio interface without showing the API or sharing externally
-dalle.launch(show_api=False, share=False)
+print("Launching Gradio interface...")
+dalle.launch(show_api=False, share=False)