Update app.py

app.py

@@ -51,21 +51,8 @@ API_TOKEN = os.getenv("HF_READ_TOKEN")
 headers = {"Authorization": f"Bearer {API_TOKEN}"}
 timeout = 100
 
-article_text = """
-<div style="text-align: center;">
-    <p>Enjoying the tool? Buy me a coffee and get exclusive prompt guides!</p>
-    <p><i>Instantly unlock helpful tips for creating better prompts!</i></p>
-    <div style="display: flex; justify-content: center;">
-        <a href="https://piczify.lemonsqueezy.com/buy/0f5206fa-68e8-42f6-9ca8-4f80c587c83e">
-            <img src="https://www.buymeacoffee.com/assets/img/custom_images/yellow_img.png" 
-                 alt="Buy Me a Coffee" 
-                 style="height: 40px; width: auto; box-shadow: 0 4px 8px rgba(0, 0, 0, 0.2); border-radius: 10px;">
-        </a>
-    </div>
-</div>
-"""
 
-def query(lora_id, prompt, steps=28, cfg_scale=3.5, randomize_seed=True, seed=-1, width=1024, height=1024):
+def query(lora_id, prompt, is_negative=False, steps=28, cfg_scale=3.5, sampler="DPM++ 2M Karras", seed=-1, strength=0.7, width=1024, height=1024):
     if prompt == "" or prompt == None:
         return None
 
@@ -86,14 +73,16 @@ def query(lora_id, prompt, steps=28, cfg_scale=3.5, randomize_seed=True, seed=-1
     # print(f'\033[1mGeneration {key}:\033[0m {prompt}')
 
     # If seed is -1, generate a random seed and use it
-    if randomize_seed:
-        seed = random.randint(1, 4294967296)
-    
-    payload = {
+        if seed == -1:
+        seed = random.randint(1, 1000000000)
+            
+        payload = {
         "inputs": prompt,
+        "is_negative": is_negative,
         "steps": steps,
         "cfg_scale": cfg_scale,
-        "seed": seed,
+        "seed": seed if seed != -1 else random.randint(1, 1000000000),
+        "strength": strength,
         "parameters": {
             "width": width,  # Pass the width to the API
             "height": height  # Pass the height to the API
@@ -119,48 +108,75 @@ def query(lora_id, prompt, steps=28, cfg_scale=3.5, randomize_seed=True, seed=-1
 
 
 examples = [
-    "a tiny astronaut hatching from an egg on the moon",
-    "a cat holding a sign that says hello world",
-    "an anime illustration of a wiener schnitzel",
+    "a beautiful woman with blonde hair and blue eyes",
+    "a beautiful woman with brown hair and grey eyes",
+    "a beautiful woman with black hair and brown eyes",
 ]
 
 css = """
 #app-container {
-    max-width: 600px;
+    max-width: 896px;
     margin-left: auto;
     margin-right: auto;
+    body{background-image:"FLUX.Dev-LORA-Serverless/abstract.jpg";}
 }
+   
 """
 
-with gr.Blocks(theme=theme, css=css) as app:
-    gr.HTML("<center><h1>FLUX.1-Dev with LoRA support</h1></center>")
-    with gr.Column(elem_id="app-container"):
-        with gr.Row():
-            with gr.Column(elem_id="prompt-container"):
-                with gr.Row():
-                    text_prompt = gr.Textbox(label="Prompt", placeholder="Enter a prompt here", lines=2, elem_id="prompt-text-input")
-                with gr.Row():
-                    custom_lora = gr.Textbox(label="Custom LoRA", info="LoRA Hugging Face path (optional)", placeholder="multimodalart/vintage-ads-flux")
-                with gr.Row():
-                    with gr.Accordion("Advanced Settings", open=False):
-                        with gr.Row():
-                            width = gr.Slider(label="Width", value=1024, minimum=64, maximum=1216, step=8)
-                            height = gr.Slider(label="Height", value=1024, minimum=64, maximum=1216, step=8)
-                        seed = gr.Slider(label="Seed", value=-1, minimum=-1, maximum=4294967296, step=1)
-                        randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
-                        with gr.Row():
+with gr.Blocks(theme=theme, css=css, elem_id="app-container") as app:
+    gr.HTML("<center><h6>🎨 FLUX.1-Dev with LoRA ++ 🇬🇧</h6></center>")
+    with gr.Tab("Text to Image"):
+        with gr.Column(elem_id="app-container"):
+            with gr.Row():
+                with gr.Column(elem_id="prompt-container"):
+                    with gr.Row():
+                        text_prompt = gr.Textbox(label="Prompt", placeholder="Enter a prompt here", lines=2, elem_id="prompt-text-input")
+                    with gr.Row():
+                         with gr.Accordion("Lora trigger words", open=False):
+                        		gr.Markdown("""
+                                    - **sdxl-realistic**: szn style
+                                    - **stylesdxl-cyberpunk**: szn style
+                                    - **maxfield-parrish-stylee**: Maxfield Parrish Style
+                                    - **surreal-harmony**: Surreal Harmony
+                                    - **extremely-detailed**: extremely detailed
+                                    - **dark-fantasy**: Dark Fantasy
+                                    - **analogredmond**: AnalogRedmAF
+                                    - **jules-bastien-lepage-style**: Jules Bastien Lepage Style
+                                    - **john-singer-sargent-style**: John Singer Sargent Style
+                                    - **alphonse-mucha-style**: Alphonse Mucha Style
+                                    - **ultra-realistic-illustration**: ultra realistic illustration
+                                    - **eye-catching**: eye-catching
+                                    - **john-constable-style**: John Constable Style
+                                    - **film-noir**: in the style of FLMNR
+                                    - **director-sofia-coppola-style**: Director Sofia Coppola Style
+                            		""",
+                                    label="Trigger words")                       
+                            
+                    with gr.Row():
+                        custom_lora = gr.Dropdown([" ", "jwu114/lora-sdxl-realistic", "issaccyj/lora-sdxl-cyberpunk", "KappaNeuro/maxfield-parrish-style", "fofr/sdxl-deep-down", "KappaNeuro/surreal-harmony", "ntc-ai/SDXL-LoRA-slider.extremely-detailed", "prithivMLmods/Canopus-LoRA-Flux-FaceRealism", "KappaNeuro/dark-fantasy", "artificialguybr/analogredmond", "KappaNeuro/jules-bastien-lepage-style", "KappaNeuro/john-singer-sargent-style", "KappaNeuro/alphonse-mucha-style", "ntc-ai/SDXL-LoRA-slider.ultra-realistic-illustration", "ntc-ai/SDXL-LoRA-slider.eye-catching", "KappaNeuro/john-constable-style", "dvyio/flux-lora-film-noir", "KappaNeuro/director-sofia-coppola-style"], label="Custom LoRA (Please select)",)
+                    with gr.Row():
+                        with gr.Accordion("⚙️ Advanced Settings", open=False, elem_id="settings-container"):
+                            negative_prompt = gr.Textbox(label="Negative Prompt", placeholder="What should not be in the image", value="((((out of frame))), 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():
+                                width = gr.Slider(label="Width", value=1024, minimum=64, maximum=1216, step=32)
+                                height = gr.Slider(label="Height", value=1024, minimum=64, maximum=1216, step=32)
                             steps = gr.Slider(label="Sampling steps", value=28, minimum=1, maximum=100, step=1)
                             cfg = gr.Slider(label="CFG Scale", value=3.5, minimum=1, maximum=20, step=0.5)
-                        # method = gr.Radio(label="Sampling method", value="DPM++ 2M Karras", choices=["DPM++ 2M Karras", "DPM++ SDE Karras", "Euler", "Euler a", "Heun", "DDIM"])
-
-        with gr.Row():
-            text_button = gr.Button("Run", variant='primary', elem_id="gen-button")
-        with gr.Row():
-            image_output = gr.Image(type="pil", label="Image Output", elem_id="gallery")
-        with gr.Row():
-            seed_output = gr.Textbox(label="Seed Used", show_copy_button = True, elem_id="seed-output")
-
-        gr.Markdown(article_text)
+                            method = gr.Radio(label="Sampling method", value="DPM++ 2M Karras", choices=["DPM++ 2M Karras", "DPM++ SDE Karras", "DPM Fast" "Euler", "Euler a", "Euler+beta", "Heun", "DDIM", "PLMS", "UniPC"])
+                            strength = gr.Slider(label="Strength", value=0.7, minimum=0, maximum=1, step=0.001)
+                            seed = gr.Slider(label="Seed", value=-1, minimum=-1, maximum=1000000000, step=1)
+                    with gr.Row():
+                        with gr.Accordion("🫘Seed", open=False):
+                            seed_output = gr.Textbox(label="Seed Used", show_copy_button = True, elem_id="seed-output")
+                            
+            with gr.Row():
+                text_button = gr.Button("Run", variant='primary', elem_id="gen-button")
+            with gr.Row():                
+                clr_button =gr.Button("Clear",variant="primary", elem_id="clear_button")
+                clr_button.click(lambda: gr.Textbox(value=""), None, text_prompt)
+                
+            with gr.Row():
+                image_output = gr.Image(type="pil", label="Image Output", format="png", elem_id="gallery")
         
         gr.Examples(
             examples = examples,
@@ -168,6 +184,6 @@ with gr.Blocks(theme=theme, css=css) as app:
         )
 
         
-        text_button.click(query, inputs=[custom_lora, text_prompt, steps, cfg, randomize_seed, seed, width, height], outputs=[image_output,seed_output, seed])
+        text_button.click(query, inputs=[custom_lora, text_prompt, negative_prompt, steps, cfg, method, seed, strength, width, height], outputs=[image_output, seed_output])
 
 app.launch(show_api=False, share=False)