Update app.py

app.py

@@ -28,6 +28,13 @@ from diffusers import (
     EulerDiscreteScheduler,
 )
 
+from dotenv import load_dotenv
+
+# Load environment variables from .env file
+load_dotenv()
+
+USERNAME = os.getenv("USERNAME")
+PASSWORD = os.getenv("PASSWORD")
 
 
 qrcode_generator = qrcode.QRCode(
@@ -68,7 +75,7 @@ def load_models_on_launch():
     loaded_controlnet = ControlNetModel.from_pretrained(
         controlnet_path,
         torch_dtype=torch.float16
-    ).to("cuda")
+    ).to("mps")
 
     diffusion_path = snapshot_download(DIFFUSION_MODELS["GhostMix"])
     loaded_pipe = StableDiffusionControlNetImg2ImgPipeline.from_pretrained(
@@ -76,7 +83,7 @@ def load_models_on_launch():
         controlnet=loaded_controlnet,
         torch_dtype=torch.float16,
         safety_checker=None,
-    ).to("cuda")
+    ).to("mps")
     print("Models loaded successfully!")
 
 # Modify the load_models function to use global variables
@@ -193,9 +200,9 @@ def inference(
     qr_code_content: str,
     prompt: str,
     negative_prompt: str,
-    guidance_scale: float = 15.0,  
-    controlnet_conditioning_scale: float = 1.5,  
-    strength: float = 0.6,  
+    guidance_scale: float = 15.0,
+    controlnet_conditioning_scale: float = 1.5,
+    strength: float = 0.6,
     seed: int = -1,
     init_image: Image.Image | None = None,
     qrcode_image: Image.Image | None = None,
@@ -208,13 +215,14 @@ def inference(
     controlnet_model: str = "QR Code Monster",
     diffusion_model: str = "GhostMix",
     reference_image_strength: float = 0.6,
+    progress: gr.Progress = gr.Progress()
 ):
     try:
-        progress = gr.Progress()
+        progress(0, desc="Initializing...")
         # Load models based on user selection
-        progress(0, desc="Downloading models...")
+        progress(0.1, desc="Loading models...")
         pipe = load_models(controlnet_model, diffusion_model)
-        progress(0.5, desc="Models downloaded, preparing for inference...")
+        progress(0.2, desc="Models loaded, preparing for inference...")
         
         if prompt is None or prompt == "":
             raise gr.Error("Prompt is required")
@@ -226,7 +234,7 @@ def inference(
         if count_tokens(prompt) > MAX_TOKENS:
             raise gr.Error(f"Prompt exceeds the maximum allowed tokens of {MAX_TOKENS}")
 
-        if count_tokens(negative_prompt) > MAX_TOKENS:
+        if negative_prompt and count_tokens(negative_prompt) > MAX_TOKENS:
             raise gr.Error(f"Negative prompt exceeds the maximum allowed tokens of {MAX_TOKENS}")
 
         pipe.scheduler = SAMPLER_MAP[sampler](pipe.scheduler.config)
@@ -235,7 +243,7 @@ def inference(
             seed = torch.seed()  # Generate a truly random seed
         generator = torch.manual_seed(seed)
 
-        if qr_code_content != "" or qrcode_image.size == (1, 1):
+        if qr_code_content != "" or (qrcode_image is not None and qrcode_image.size == (1, 1)):
             print("Generating QR Code from content")
             qr = qrcode.QRCode(
                 version=1,
@@ -248,9 +256,11 @@ def inference(
 
             qrcode_image = qr.make_image(fill_color=qr_color, back_color=bg_color)
             qrcode_image = resize_for_condition_image(qrcode_image, 1024)
-        else:
+        elif qrcode_image is not None:
             print("Using QR Code Image")
             qrcode_image = resize_for_condition_image(qrcode_image, 1024)
+        else:
+            raise gr.Error("No valid QR code content or image provided")
 
         # Determine which image to use as init_image and control_image
         if use_qr_code_as_init_image:
@@ -274,10 +284,10 @@ def inference(
         if invert_init_image and init_image is not None:
             init_image = invert_image(init_image)
 
-        final_image = None
+        progress(0.5, desc="Generating image...")
         out = pipe(
-            prompt=prompt,  # Use the full prompt
-            negative_prompt=negative_prompt,  # Use the full negative prompt
+            prompt=prompt,
+            negative_prompt=negative_prompt,
             image=init_image,
             control_image=control_image,
             width=1024,
@@ -287,19 +297,18 @@ def inference(
             generator=generator,
             strength=float(strength),
             num_inference_steps=50,
+            callback=lambda step, timestep, latents: progress(0.5 + step / 100, desc=f"Step {step}/50")
         )
-        final_image = out.images[0] if final_image is None else final_image
+        final_image = out.images[0]
 
         if invert_final_image:
             final_image = invert_image(final_image)
         
-        return final_image, seed
-    except gr.Error as e:
-        print(f"Gradio error in inference: {str(e)}")
-        return Image.new('RGB', (1024, 1024), color='white'), -1
+        progress(1.0, desc="Done!")
+        return final_image, seed, gr.update(visible=False)  # Add the third output here
     except Exception as e:
         print(f"Unexpected error in inference: {str(e)}")
-        return Image.new('RGB', (1024, 1024), color='white'), -1
+        return Image.new('RGB', (1024, 1024), color='white'), -1, gr.update(visible=False)
 
 
 
@@ -333,132 +342,301 @@ def invert_init_image_display(image):
     return inverted
 
 css = """
-h1 {
+h1, h2, h3, h4, h5, h6, p, li, ul, ol, a {
     text-align: center;
-    display:block;
+    display: block;
+}
+ul, ol {
+    margin-left: auto;
+    margin-right: auto;
+    display: table;
 }
 """
 
-with gr.Blocks(theme='Hev832/Applio', css=css) as blocks:
-    gr.Markdown(
-        """
-        ![Yamamoto Logo](https://cdn-uploads.huggingface.co/production/uploads/64740cf7485a7c8e1bd51ac9/_VyYxp5qE_nRZ_LJqBxmL.webp)
-        # 🎨 Yamamoto QR Code Art Generator
-        ## Transform Your QR Codes into Brand Masterpieces
-        Welcome to Yamamoto's innovative QR Code Art Generator! This cutting-edge tool empowers our creative team to craft 
-        visually stunning, on-brand QR codes that perfectly blend functionality with artistic expression.
-        ## 🚀 How It Works:
-        1. **Enter Your QR Code Content**: Start by inputting the URL or text for your QR code.
-        2. **Craft Your Prompt**: Describe the artistic style or theme you envision for your QR code.
-        3. **Fine-tune with Advanced Settings**: Adjust parameters to perfect your creation (see tips below).
-        4. **Generate and Iterate**: Click 'Run' to create your art, then refine as needed.
-        ## 🌟 Tips for Spectacular Results:
-        - **Artistic Freedom**: Set between 0.8 and 0.95 for a balance of creativity and scannability.
-        - **QR Code Visibility**: Aim for 0.6 to 2.0 to ensure your code is both artistic and functional.
-        - **Prompt Crafting**: Use vivid, specific descriptions that align with your brand identity.
-        - **Experimentation**: Don't hesitate to try different settings and prompts to find your perfect style!
-        ## 🎭 Prompt Ideas to Spark Your Creativity:
-        - "A serene Japanese garden with cherry blossoms and a koi pond"
-        - "A futuristic cityscape with neon lights and flying cars"
-        - "An abstract painting with swirling colors and geometric shapes"
-        - "A vintage-style travel poster featuring iconic landmarks"
-        Remember, the magic lies in the details of your prompt and the fine-tuning of your settings. 
-        Happy creating!
-        """
-    )
-    with gr.Row():
+# Add login function
+def login(username, password):
+    if username == USERNAME and password == PASSWORD:
+        return gr.update(visible=True), gr.update(visible=False), gr.update(visible=False)
+    else:
+        return gr.update(visible=False), gr.update(visible=True), gr.update(visible=True)
+
+# Add login elements to the Gradio interface
+with gr.Blocks(theme='Hev832/Applio', css=css, fill_width=True, fill_height=True) as blocks:
+    generated_images = gr.State([])
+
+    with gr.Tab("Welcome"):
+        gr.Markdown(
+            """
+            ![Yamamoto Logo](https://cdn-uploads.huggingface.co/production/uploads/64740cf7485q8e1bd51ac9/_VyYxp5qE_nRZ_LJqBxmL.webp)
+            # 🎨 Yamamoto QR Code Art Generator
+            ## Transform Your QR Codes into Brand Masterpieces
+            Welcome to Yamamoto's innovative QR Code Art Generator! This cutting-edge tool empowers our creative team to craft 
+            visually stunning, on-brand QR codes that perfectly blend functionality with artistic expression.
+            ## 🚀 How It Works:
+            1. **Enter Your QR Code Content**: Start by inputting the URL or text for your QR code.
+            2. **Craft Your Prompt**: Describe the artistic style or theme you envision for your QR code.
+            3. **Fine-tune with Advanced Settings**: Adjust parameters to perfect your creation (see tips below).
+            4. **Generate and Iterate**: Click 'Run' to create your art, then refine as needed.
+            ---
+            ## Login below using the Yamamoto internal ussername and password to access the full app.
+            ### Once logging in, a new tab will appear allowing you to access the QR Code Art Generator.
+            """
+        )
+        
+        # Add login form
         with gr.Column():
-            qr_code_content = gr.Textbox(
-                label="QR Code Content",
-                placeholder="Enter URL or text for your QR code",
-                info="This is what your QR code will link to or display when scanned.",
-                value="https://theunderground.digital/",
-            )
+                username = gr.Textbox(label="Username", placeholder="Enter your username", value="UDG")
+                password = gr.Textbox(label="Password", type="password", placeholder="Enter your password", value="UDG!")
+        with gr.Column():
+                login_button = gr.Button("Login")
+                login_error = gr.Markdown("Invalid username or password. Please try again.", visible=False)
+
+
+    with gr.Tab("QR Code and Prompt Input", visible=False) as app_container:
+        with gr.Row():
+            with gr.Column():
+                qr_code_content = gr.Textbox(
+                    label="QR Code Content",
+                    placeholder="Enter URL or text for your QR code",
+                    info="This is what your QR code will link to or display when scanned.",
+                    value="https://theunderground.digital/",
+                    lines=1,
+                )
 
-            prompt = gr.Textbox(
-                label="Artistic Prompt",
-                placeholder="Describe the style or theme for your QR code art (For best results, keep the prompt to 75 characters or less as seen below)",
-                value="A high-res, photo-realistic minimalist rendering of Mount Fuji as a sharp, semi-realistic silhouette on the horizon. The mountain conveys strength and motion with clean, crisp lines and natural flow. Features detailed snow textures, subtle ridge highlights, and a powerful yet serene atmosphere. Emphasizes strength with clarity and precision in texture and light. (Sharp outlines:1.5), (Photo-realistic:1.4), (Detailed textures:1.3), (Minimalist:1.3), (Semi-realistic:1.3), (Monochrome contrast:1.2), (Crisp detail:1.2), (Evoking strength:1.2). Inspired by traditional Japanese woodblock prints, nature photography, and minimalist design principles.",
-                info="Describe the style or theme for your QR code art (For best results, keep the prompt to 75 characters or less as seen in the example)",
-            )
-            negative_prompt = gr.Textbox(
-                label="Elements to Avoid",
-                placeholder="Describe what you don't want in the image",
-                value="ugly, disfigured, low quality, blurry, nsfw, bad_pictures, (bad_prompt_version2:0.8), EasyNegative, 3d, cartoon, anime, sketches, (worst quality:2), (low quality:2), (normal quality:2), lowres, normal quality, ((monochrome)), ((grayscale)), poorly drawn, distorted, overexposed, flat shading, bad proportions, deformed, pixelated, messy details, lack of contrast, unrealistic textures, bad anatomy, rough edges, low resolution, text artifacts.",
-                info="List elements or styles you want to avoid in your QR code art.",
-            )
+                prompt = gr.Textbox(
+                    label="Artistic Prompt",
+                    placeholder="Describe the style or theme for your QR code art (For best results, keep the prompt to 75 characters or less as seen below)",
+                    value="A high-res, photo-realistic minimalist rendering of Mount Fuji as a sharp, semi-realistic silhouette on the horizon. The mountain conveys strength and motion with clean, crisp lines and natural flow. Features detailed snow textures, subtle ridge highlights, and a powerful yet serene atmosphere. Emphasizes strength with clarity and precision in texture and light. (Sharp outlines:1.5), (Photo-realistic:1.4), (Detailed textures:1.3), (Minimalist:1.3), (Semi-realistic:1.3), (Monochrome contrast:1.2), (Crisp detail:1.2), (Evoking strength:1.2). Inspired by traditional Japanese woodblock prints, nature photography, and minimalist design principles.",
+                    info="Describe the style or theme for your QR code art (For best results, keep the prompt to 75 characters or less as seen in the example)",
+                    lines=8,
+                )
+                negative_prompt = gr.Textbox(
+                    label="Elements to Avoid",
+                    placeholder="Describe what you don't want in the image",
+                    value="ugly, disfigured, low quality, blurry, nsfw, bad_pictures, (bad_prompt_version2:0.8), EasyNegative, 3d, cartoon, anime, sketches, (worst quality:2), (low quality:2), (normal quality:2), lowres, normal quality, ((monochrome)), ((grayscale)), poorly drawn, distorted, overexposed, flat shading, bad proportions, deformed, pixelated, messy details, lack of contrast, unrealistic textures, bad anatomy, rough edges, low resolution, text artifacts.",
+                    info="List elements or styles you want to avoid in your QR code art.",
+                    lines=4,
+                )
 
-            run_btn = gr.Button("🎨 Create Your QR Art", variant="primary")  
+                run_btn = gr.Button("🎨 Create Your QR Art", variant="primary")  
+
+                with gr.Accordion(label="Needs Some Prompting Help?", open=False, visible=True):
+                    gr.Markdown(
+                        """
+                        ## 🌟 Tips for Spectacular Results:
+                        - Use concise details in your prompt to help the AI understand your vision.
+                        - Use negative prompts to avoid unwanted elements in your image.
+                        - Experiment with different ControlNet models and diffusion models to find the best combination for your prompt.
+
+                        ## 🎭 Prompt Ideas to Spark Your Creativity:
+                        - "A serene Japanese garden with cherry blossoms and a koi pond"
+                        - "A futuristic cityscape with neon lights and flying cars"
+                        - "An abstract painting with swirling colors and geometric shapes"
+                        - "A vintage-style travel poster featuring iconic landmarks"
+
+                        Remember, the magic lies in the details of your prompt and the fine-tuning of your settings. 
+                        Happy creating!
+                        """
+                    )
+
+                with gr.Accordion("Set Custom QR Code Colors", open=False):
+                    bg_color = gr.ColorPicker(
+                        label="Background Color", 
+                        value="#FFFFFF",
+                        info="Choose the background color for the QR code"
+                    )
+                    qr_color = gr.ColorPicker(
+                        label="QR Code Color", 
+                        value="#000000",
+                        info="Choose the color for the QR code pattern"
+                    )
+                    invert_final_image = gr.Checkbox(
+                        label="Invert Final Image", 
+                        value=False,
+                        info="Check this to invert the colors of the final image",
+                        visible=False,
+                    )
+                with gr.Accordion("AI Model Selection", open=False):
+                    controlnet_model_dropdown = gr.Dropdown(
+                        choices=list(CONTROLNET_MODELS.keys()),
+                        value="QR Code Monster",
+                        label="ControlNet Model",
+                        info="Select the ControlNet model for QR code generation"
+                    )
+                    diffusion_model_dropdown = gr.Dropdown(
+                        choices=list(DIFFUSION_MODELS.keys()),
+                        value="GhostMix",
+                        label="Diffusion Model",
+                        info="Select the main diffusion model for image generation"
+                    )
+                
             
-            with gr.Accordion(label="Use Your Own Image as a Reference", open=True, visible=True) as init_image_acc:
-                init_image = gr.Image(label="Reference Image", type="pil")
-                with gr.Row():
-                    use_qr_code_as_init_image = gr.Checkbox(
-                        label="Uncheck to use your own image for generation", 
-                        value=True, 
-                        interactive=True,
-                        info="Allows you to use your own image for generation, otherwise a generic QR Code is created automatically as the base image"
+                with gr.Accordion(label="QR Code Image (Optional)", open=False, visible=False):
+                    qr_code_image = gr.Image(
+                        label="QR Code Image (Optional). Leave blank to automatically generate QR code",
+                        type="pil",
                     )
-                    invert_init_image_button = gr.Button("Invert Init Image")
             
-            with gr.Accordion(label="QR Code Image (Optional)", open=False, visible=False):
-                qr_code_image = gr.Image(
-                    label="QR Code Image (Optional). Leave blank to automatically generate QR code",
-                    type="pil",
-                )
+            with gr.Column():
+                result_image = gr.Image(label="Your Artistic QR Code", show_download_button=True, show_fullscreen_button=True)
+                result_image_loading = gr.Markdown("Generating image...", visible=False)
+                scan_button = gr.Button("Verify QR Code Works", visible=False)
+                scan_result = gr.Textbox(label="Validation Result of QR Code", interactive=False, visible=False)
+                used_seed = gr.Number(label="Seed Used", interactive=False)
+
+                with gr.Accordion(label="Use Your Own Image as a Reference", open=True, visible=True) as init_image_acc:
+                    init_image = gr.Image(label="Reference Image", type="pil")
+                    with gr.Row():
+                        use_qr_code_as_init_image = gr.Checkbox(
+                            label="Uncheck to use your own image for generation", 
+                            value=True, 
+                            interactive=True,
+                            info="Allows you to use your own image for generation, otherwise a generic QR Code is created automatically as the base image"
+                        )
+                        invert_init_image_button = gr.Button("Invert Init Image", size="sm", visible=False)
 
-                
-            with gr.Accordion("Set Custom QR Code Colors", open=False):
-                bg_color = gr.ColorPicker(
-                    label="Background Color", 
-                    value="#FFFFFF",
-                    info="Choose the background color for the QR code"
+    with gr.Tab("Advanced Settings"):
+        with gr.Accordion("Advanced Art Controls", open=True):
+            with gr.Row():
+                controlnet_conditioning_scale = gr.Slider(
+                    minimum=0.0,
+                    maximum=5.0,
+                    step=0.01,
+                    value=2,
+                    label="QR Code Visibility in Image",
                 )
-                qr_color = gr.ColorPicker(
-                    label="QR Code Color", 
-                    value="#000000",
-                    info="Choose the color for the QR code pattern"
+                with gr.Accordion("How Much QR Code Visibility is in Final Image (Click For Explanation)", open=False):
+                    gr.Markdown(
+                        """
+                        **QR Code Visibility** determines how much the QR code itself stands out in the final design. Think of it like balancing between how "artistic" the image looks and how "functional" the QR code is.
+                
+                        - **Low settings (0.0-1)**: If you choose a lower value, the QR code will blend more into the art, and it might be hard to scan with a phone. This setting is great if you want the image to look amazing, but you might lose some of the scannability. Try this if you care more about art and less about the QR code being easily recognized.
+                        
+                        - **Medium settings (1-3)**: This is the sweet spot where the QR code remains clearly visible while still blending in with the art. You can still scan it easily with a phone, but it looks more creative. For most users, setting it around **1.1** is a great start to balance both art and function.
+                        
+                        - **High settings (3-5.0)**: If you need to make sure that the QR code is super easy to scan, even if it means the image looks less like art and more like a regular QR code, then choose a higher value. This is ideal when functionality is the main goal, and the artistic side can take a backseat.
+                        
+                        Start with **1.3** if you're unsure, and adjust up or down depending on whether you want the QR code to be more artistic or more functional.
+                        """
+                    )
+            
+            with gr.Row():
+                strength = gr.Slider(
+                    minimum=0.0, 
+                    maximum=5, 
+                    step=0.10, 
+                    value=2, 
+                    label="Artistic Freedom for the AI When Generating",
                 )
-                invert_final_image = gr.Checkbox(
-                    label="Invert Final Image", 
-                    value=False,
-                    info="Check this to invert the colors of the final image",
-                    visible=False,
+                with gr.Accordion("How Much Artistic Freedom the AI has When Generating Image (Click For Explanation)", open=False):
+                    gr.Markdown(
+                        """
+                        **Artistic Freedom** controls how much the AI is allowed to change the QR code's look to match your description. It's like telling the AI how creative it can get with your QR code:
+                
+                        - **Low settings (0.10-2)**: If you set this low, the AI will make small changes and your QR code will look more like a regular, plain QR code. This is useful if you want something that is still creative but not too wild, keeping it simple and easy to scan.
+                
+                        - **Medium settings (2-3)**: Here, the AI will add more artistic touches but keep the QR code recognizable. You get the best of both worlds—your QR code will have some creative flair, but it will still be easy to scan. For most cases, setting it to **0.6** is a great way to keep the code functional and artistic.
+                
+                        - **High settings (3-5)**: If you set this high, the AI will go all-out creative. The QR code will look amazing, but it might be difficult to scan because the art can start to take over the code. This setting is perfect if you're aiming for a highly creative piece of art and don't mind if it's a bit harder to scan. Start at **0.9** to explore creative but functional designs.
+                        """
+                    )
+            
+            with gr.Row():
+                guidance_scale = gr.Slider(
+                    minimum=0.0,
+                    maximum=100.0,
+                    step=0.25,
+                    value=7.5,
+                    label="How Closely the AI Follows the Prompt",
                 )
-            with gr.Accordion("AI Model Selection", open=False):
-                controlnet_model_dropdown = gr.Dropdown(
-                    choices=list(CONTROLNET_MODELS.keys()),
-                    value="QR Code Monster",
-                    label="ControlNet Model",
-                    info="Select the ControlNet model for QR code generation"
+                with gr.Accordion("How Closely the AI Follows the Prompt (Click For Explanation)", open=False):
+                    gr.Markdown(
+                        """
+                        **Follow the Prompt** tells the AI how closely it should follow your description when creating the QR code art. Think of it like giving the AI instructions on how strict or flexible it can be with your design ideas:
+                
+                        - **Low settings (0-5)**: If you choose a lower value, the AI has more freedom to get creative on its own and may not stick too closely to your description. This is great if you want to see how the AI interprets your ideas in unexpected ways.
+                
+                        - **Medium settings (5-15)**: This is a good balance where the AI will mostly follow your prompt but will also add some of its own creative touches. If you want to see some surprises but still want the design to look like what you described, start at around **7.5**.
+                
+                        - **High settings (15+)**: If you choose a higher value, the AI will stick very closely to what you wrote in the description. This is good if you have a very specific idea and don't want the AI to change much. Just keep in mind that this might limit the AI's creativity.
+                        
+                        Start at **7.5** for a balanced approach where the AI follows your ideas but still adds some artistic flair.
+                        """
+                    )
+            
+            with gr.Row():
+                sampler = gr.Dropdown(
+                    choices=list(SAMPLER_MAP.keys()), 
+                    value="DPM++ Karras SDE", 
+                    label="Art Style Used to Create the Image",
                 )
-                diffusion_model_dropdown = gr.Dropdown(
-                    choices=list(DIFFUSION_MODELS.keys()),
-                    value="GhostMix",
-                    label="Diffusion Model",
-                    info="Select the main diffusion model for image generation"
+                with gr.Accordion("Details on Art Style the AI Uses to Create the Image (Click For Explanation)", open=False):
+                    gr.Markdown(
+                        """
+                        **Art Style** changes how the AI creates the image, using different methods (or "samplers"). Each method has a different effect on how detailed or artistic the final QR code looks:
+                
+                        - **DPM++ Karras SDE**: This is a great all-around option for creating high-quality, detailed images. It's a good place to start if you want a balance of sharpness and creativity.
+                
+                        - **Euler**: This method creates very sharp, detailed images, making the QR code look crisp and clear. Choose this if you want a precise, well-defined design.
+                
+                        - **DDIM**: This method is better if you want the QR code to have a more artistic, abstract style. It's great for when you want the QR code to look like a piece of modern art.
+                        
+                        Feel free to experiment with different samplers to see what works best for the look you're going for!
+                        """
+                    )
+            
+            with gr.Row():
+                seed = gr.Slider(
+                    minimum=-1,
+                    maximum=9999999999,
+                    step=1,
+                    value=-1,
+                    label="Creative Seed for the Image Generation",                 
                 )
+                with gr.Accordion("How Creative Seed Works for Generating New and Unique Images (Click For Explanation)", open=False):
+                    gr.Markdown(
+                        """
+                        **Creative Seed** controls whether the AI creates a completely new design each time or sticks to a specific design. Think of it like a recipe: with the same seed number, you get the same "recipe" for your QR code every time.
+                
+                        - **-1**: This setting makes the AI create something completely new every time you run it. Use this if you want to explore different design ideas with each attempt.
+                
+                        - **Any other number**: If you set a specific number, the AI will always create the same image based on that number. This is useful if you find a design you like and want to recreate it exactly.
                 
+                        Try **-1** if you want to explore and generate different designs. If you find something you really love, write down the seed number and use it again to recreate the same design.
+                        """
+                    )
+
+            with gr.Row():
+                reference_image_strength = gr.Slider(
+                    minimum=0.0,
+                    maximum=5.0,
+                    step=0.05,
+                    value=0.6,
+                    label="Reference Image Influence",
+                    info="Controls how much the reference image influences the final result (0 = ignore, 5 = copy exactly)",
+                    visible=False  # We'll make this visible when a reference image is uploaded
+                )
+
+    with gr.Tab("Image Editing"):
         with gr.Column():
-            result_image = gr.Image(label="Your Artistic QR Code", show_download_button=True, show_fullscreen_button=True)
+            image_selector = gr.Dropdown(label="Select Image to Edit", choices=[], interactive=True)
+            image_to_edit = gr.Image(label="Your Artistic QR Code", show_download_button=True, show_fullscreen_button=True)
 
-            
-            with gr.Row(visible=False):
+            with gr.Row():
+                edited_image = gr.Image(label="Edited QR Code", show_download_button=True, show_fullscreen_button=True)
                 scan_button = gr.Button("Verify QR Code Works")
                 scan_result = gr.Textbox(label="Validation Result of QR Code", interactive=False)
 
-            
             with gr.Row():
                 brightness = gr.Slider(minimum=0.1, maximum=2.0, step=0.1, value=1.0, label="Brightness")
                 contrast = gr.Slider(minimum=0.1, maximum=2.0, step=0.1, value=1.0, label="Contrast")
                 saturation = gr.Slider(minimum=0.1, maximum=2.0, step=0.1, value=1.0, label="Saturation")
             with gr.Row():
-                invert_button = gr.Button("Invert Image")
+                invert_button = gr.Button("Invert Image", size="sm")
                 
             used_seed = gr.Number(label="Seed Used", interactive=False)
             
             gr.Markdown(
-            
                 """
                 ### 🔍 Analyzing Your Creation
                 - Is the QR code scannable? Check with your phone camera to see if it can scan it.
@@ -469,221 +647,119 @@ with gr.Blocks(theme='Hev832/Applio', css=css) as blocks:
                 """
             )
 
-            with gr.Accordion("Advanced Art Controls", open=True):
-                with gr.Row():
-                    controlnet_conditioning_scale = gr.Slider(
-                        minimum=0.0,
-                        maximum=5.0,
-                        step=0.01,
-                        value=2,
-                        label="QR Code Visibility in Image",
-                    )
-                    with gr.Accordion("How Much QR Code Visibility is in Final Image (Click For Explanation)", open=False):
-                        gr.Markdown(
-                            """
-                            **QR Code Visibility** determines how much the QR code itself stands out in the final design. Think of it like balancing between how "artistic" the image looks and how "functional" the QR code is.
-                    
-                            - **Low settings (0.0-1)**: If you choose a lower value, the QR code will blend more into the art, and it might be hard to scan with a phone. This setting is great if you want the image to look amazing, but you might lose some of the scannability. Try this if you care more about art and less about the QR code being easily recognized.
-                            
-                            - **Medium settings (1-3)**: This is the sweet spot where the QR code remains clearly visible while still blending in with the art. You can still scan it easily with a phone, but it looks more creative. For most users, setting it around **1.1** is a great start to balance both art and function.
-                    
-                            - **High settings (3-5.0)**: If you need to make sure that the QR code is super easy to scan, even if it means the image looks less like art and more like a regular QR code, then choose a higher value. This is ideal when functionality is the main goal, and the artistic side can take a backseat.
-                            
-                            Start with **1.3** if you're unsure, and adjust up or down depending on whether you want the QR code to be more artistic or more functional.
-                            """
-                        )
-                
-                with gr.Row():
-                    strength = gr.Slider(
-                        minimum=0.0, 
-                        maximum=5, 
-                        step=0.10, 
-                        value=2, 
-                        label="Artistic Freedom for the AI When Generating",
-                    )
-                    with gr.Accordion("How Much Artistic Freedom the AI has When Generating Image (Click For Explanation)", open=False):
-                        gr.Markdown(
-                            """
-                            **Artistic Freedom** controls how much the AI is allowed to change the QR code's look to match your description. It's like telling the AI how creative it can get with your QR code:
-                    
-                            - **Low settings (0.10-2)**: If you set this low, the AI will make small changes and your QR code will look more like a regular, plain QR code. This is useful if you want something that is still creative but not too wild, keeping it simple and easy to scan.
-                    
-                            - **Medium settings (2-3)**: Here, the AI will add more artistic touches but keep the QR code recognizable. You get the best of both worlds—your QR code will have some creative flair, but it will still be easy to scan. For most cases, setting it to **0.6** is a great way to keep the code functional and artistic.
-                    
-                            - **High settings (3-5)**: If you set this high, the AI will go all-out creative. The QR code will look amazing, but it might be difficult to scan because the art can start to take over the code. This setting is perfect if you're aiming for a highly creative piece of art and don't mind if it's a bit harder to scan. Start at **0.9** to explore creative but functional designs.
-                            """
-                        )
-                
-                with gr.Row():
-                    guidance_scale = gr.Slider(
-                        minimum=0.0,
-                        maximum=100.0,
-                        step=0.25,
-                        value=7.5,
-                        label="How Closely the AI Follows the Prompt",
- 
-                    )
-                    with gr.Accordion("How Closely the AI Follows the Prompt (Click For Explanation)", open=False):
-                        gr.Markdown(
-                            """
-                            **Follow the Prompt** tells the AI how closely it should follow your description when creating the QR code art. Think of it like giving the AI instructions on how strict or flexible it can be with your design ideas:
-                    
-                            - **Low settings (0-5)**: If you choose a lower value, the AI has more freedom to get creative on its own and may not stick too closely to your description. This is great if you want to see how the AI interprets your ideas in unexpected ways.
-                    
-                            - **Medium settings (5-15)**: This is a good balance where the AI will mostly follow your prompt but will also add some of its own creative touches. If you want to see some surprises but still want the design to look like what you described, start at around **7.5**.
-                    
-                            - **High settings (15+)**: If you choose a higher value, the AI will stick very closely to what you wrote in the description. This is good if you have a very specific idea and don't want the AI to change much. Just keep in mind that this might limit the AI's creativity.
-                            
-                            Start at **7.5** for a balanced approach where the AI follows your ideas but still adds some artistic flair.
-                            """
-                    )
-                
-                with gr.Row():
-                    sampler = gr.Dropdown(
-                        choices=list(SAMPLER_MAP.keys()), 
-                        value="DPM++ Karras SDE", 
-                        label="Art Style Used to Create the Image",
-
-                    )
-                    with gr.Accordion("Details on Art Style the AI Uses to Create the Image (Click For Explanation)", open=False):
-                        gr.Markdown(
-                            """
-                            **Art Style** changes how the AI creates the image, using different methods (or "samplers"). Each method has a different effect on how detailed or artistic the final QR code looks:
-                    
-                            - **DPM++ Karras SDE**: This is a great all-around option for creating high-quality, detailed images. It's a good place to start if you want a balance of sharpness and creativity.
-                    
-                            - **Euler**: This method creates very sharp, detailed images, making the QR code look crisp and clear. Choose this if you want a precise, well-defined design.
-                    
-                            - **DDIM**: This method is better if you want the QR code to have a more artistic, abstract style. It's great for when you want the QR code to look like a piece of modern art.
-                            
-                            Feel free to experiment with different samplers to see what works best for the look you're going for!
-                            """
-                        )
+            def update_image_list(image, seed):
+                generated_images.value.append((image, seed))
+                return gr.Dropdown(choices=[f"Image {i+1} (Seed: {s})" for i, (_, s) in enumerate(generated_images.value)])
+
+            def load_image_for_editing(choice):
+                if not choice:
+                    return None
+                index = int(choice.split()[1]) - 1
+                return generated_images.value[index][0]
+
+            def edit_image(image, brightness, contrast, saturation):
+                if image is None:
+                    return None
+                img = Image.fromarray(image) if isinstance(image, np.ndarray) else image
+                img = ImageEnhance.Brightness(img).enhance(brightness)
+                img = ImageEnhance.Contrast(img).enhance(contrast)
+                img = ImageEnhance.Color(img).enhance(saturation)
+                return np.array(img)
+
+            def invert_image(image):
+                if image is None:
+                    return None
+                return ImageOps.invert(Image.fromarray(image) if isinstance(image, np.ndarray) else image)
+
+            def scan_and_display(image):
+                if image is None:
+                    return "No image to scan"
                 
-                with gr.Row():
-                    seed = gr.Slider(
-                        minimum=-1,
-                        maximum=9999999999,
-                        step=1,
-                        value=-1,
-                        label="Creative Seed for the Image Generation",                 
-                    )
-                    with gr.Accordion("How Creative Seed Works for Generating New and Unique Images (Click For Explanation)", open=False):
-                        gr.Markdown(
-                            """
-                            **Creative Seed** controls whether the AI creates a completely new design each time or sticks to a specific design. Think of it like a recipe: with the same seed number, you get the same "recipe" for your QR code every time.
-                    
-                            - **-1**: This setting makes the AI create something completely new every time you run it. Use this if you want to explore different design ideas with each attempt.
-                    
-                            - **Any other number**: If you set a specific number, the AI will always create the same image based on that number. This is useful if you find a design you like and want to recreate it exactly.
-                    
-                            Try **-1** if you want to explore and generate different designs. If you find something you really love, write down the seed number and use it again to recreate the same design.
-                            """
-                        )
-
-                with gr.Row():
-                    reference_image_strength = gr.Slider(
-                        minimum=0.0,
-                        maximum=5.0,
-                        step=0.05,
-                        value=0.6,
-                        label="Reference Image Influence",
-                        info="Controls how much the reference image influences the final result (0 = ignore, 5 = copy exactly)",
-                        visible=False  # We'll make this visible when a reference image is uploaded
-                    )
-
-    def scan_and_display(image):
-        if image is None:
-            return "No image to scan"
-        
-        scanned_text = scan_qr_code(image)
-        if scanned_text:
-            return f"Scanned successfully: {scanned_text}"
-        else:
-            return "Failed to scan QR code. Try adjusting the settings for better visibility."
-
-    def invert_displayed_image(image):
-        if image is None:
-            return None
-        return invert_image(image)
-
-    scan_button.click(
-        scan_and_display,
-        inputs=[result_image],
-        outputs=[scan_result]
-    )
-
-    invert_button.click(
-        invert_displayed_image,
-        inputs=[result_image],
-        outputs=[result_image]
-    )
+                scanned_text = scan_qr_code(image)
+                if scanned_text:
+                    return f"Scanned successfully: {scanned_text}"
+                else:
+                    return "Failed to scan QR code. Try adjusting the settings for better visibility."
+
+            run_btn.click(
+                inference,
+                inputs=[
+                    qr_code_content,
+                    prompt,
+                    negative_prompt,
+                    guidance_scale,
+                    controlnet_conditioning_scale,
+                    strength,
+                    seed,
+                    init_image,
+                    qr_code_image,
+                    use_qr_code_as_init_image,
+                    sampler,
+                    bg_color,
+                    qr_color,
+                    invert_final_image,
+                    controlnet_model_dropdown,
+                    diffusion_model_dropdown,
+                    reference_image_strength,
+                ],
+                outputs=[result_image, used_seed, result_image_loading],
+                show_progress=True
+            ).then(
+                update_image_list,
+                inputs=[result_image, used_seed],
+                outputs=[image_selector]
+            )
 
-    invert_init_image_button.click(
-        invert_init_image_display,
-        inputs=[init_image],
-        outputs=[init_image]
-    )
+            image_selector.change(
+                load_image_for_editing,
+                inputs=[image_selector],
+                outputs=[image_to_edit]
+            )
 
-    brightness.change(
-        adjust_image,
-        inputs=[result_image, brightness, contrast, saturation],
-        outputs=[result_image]
-    )
-    contrast.change(
-        adjust_image,
-        inputs=[result_image, brightness, contrast, saturation],
-        outputs=[result_image]
-    )
-    saturation.change(
-        adjust_image,
-        inputs=[result_image, brightness, contrast, saturation],
-        outputs=[result_image]
-    )
+            brightness.change(
+                edit_image,
+                inputs=[image_to_edit, brightness, contrast, saturation],
+                outputs=[edited_image]
+            )
+            contrast.change(
+                edit_image,
+                inputs=[image_to_edit, brightness, contrast, saturation],
+                outputs=[edited_image]
+            )
+            saturation.change(
+                edit_image,
+                inputs=[image_to_edit, brightness, contrast, saturation],
+                outputs=[edited_image]
+            )
 
-    # Add logic to show/hide the reference_image_strength slider
-    def update_reference_image_strength_visibility(init_image, use_qr_code_as_init_image):
-        return gr.update(visible=init_image is not None and not use_qr_code_as_init_image)
+            invert_button.click(
+                invert_image,
+                inputs=[image_to_edit],
+                outputs=[edited_image]
+            )
 
-    init_image.change(
-        update_reference_image_strength_visibility,
-        inputs=[init_image, use_qr_code_as_init_image],
-        outputs=[reference_image_strength]
-    )
+            scan_button.click(
+                scan_and_display,
+                inputs=[edited_image],
+                outputs=[scan_result]
+            )
 
-    use_qr_code_as_init_image.change(
-        update_reference_image_strength_visibility,
-        inputs=[init_image, use_qr_code_as_init_image],
-        outputs=[reference_image_strength]
+    # Define login button click behavior
+    login_button.click(
+        login,
+        inputs=[username, password],
+        outputs=[app_container, login_error, login_button]
     )
 
-    run_btn.click(
-        inference,
-        inputs=[
-            qr_code_content,
-            prompt,
-            negative_prompt,
-            guidance_scale,
-            controlnet_conditioning_scale,
-            strength,
-            seed,
-            init_image,
-            qr_code_image,
-            use_qr_code_as_init_image,
-            sampler,
-            bg_color,
-            qr_color,
-            invert_final_image,
-            controlnet_model_dropdown,
-            diffusion_model_dropdown,
-            reference_image_strength,
-        ],
-        outputs=[result_image, used_seed],
-        concurrency_limit=20
+    # Define password textbox submit behavior
+    password.submit(
+        login,
+        inputs=[username, password],
+        outputs=[app_container, login_error, login_button]
     )
 
 # Load models on launch
 load_models_on_launch()
 
 blocks.queue(max_size=20)
-blocks.launch(share=True, show_api=True)
+blocks.launch(share=False, show_api=True)