Update app.py

app.py

@@ -4,10 +4,14 @@ import io
 import random
 import os
 import time
-import cv2
+import numpy as np
+import subprocess
+import torch
+from transformers import AutoProcessor, AutoModelForCausalLM
 from PIL import Image
 from deep_translator import GoogleTranslator
 import json
+from datetime import datetime
 from fastapi import FastAPI
 
 app = FastAPI()
@@ -52,8 +56,11 @@ timeout = 100
 
 def flip_image(x):
     return np.fliplr(x)
-    
-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):
+
+def clear():
+  return None    
+
+def query(lora_id, prompt, is_negative=False, steps=28, cfg_scale=3.5, sampler="DPM++ 2M Karras", seed=-1, strength=0.7, width=896, height=1152):
     if prompt == "" or prompt == None:
         return None
 
@@ -66,7 +73,7 @@ def query(lora_id, prompt, is_negative=False, steps=28,  cfg_scale=3.5, sampler=
 
     API_TOKEN = random.choice([os.getenv("HF_READ_TOKEN")])
     headers = {"Authorization": f"Bearer {API_TOKEN}"}
-    #NUM_IMAGES_PER_PROMPT=1
+    
     # prompt = GoogleTranslator(source='ru', target='en').translate(prompt)
     # print(f'\033[1mGeneration {key} translation:\033[0m {prompt}')
     prompt = GoogleTranslator(source='ru', target='en').translate(prompt)
@@ -109,27 +116,22 @@ def query(lora_id, prompt, is_negative=False, steps=28,  cfg_scale=3.5, sampler=
     except Exception as e:
         print(f"Error when trying to open the image: {e}")
         return None
-        
-def clear():
-    return None
-    
+
 examples = [
     "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: 930px;
     margin-left: auto;
     margin-right: auto;
-    
 }
+".gradio-container {background: url('file=abstract.jpg')}
    
 """
-
 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"):
@@ -139,63 +141,58 @@ with gr.Blocks(theme=theme, css=css, elem_id="app-container") as app:
                     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(label="Lora trigger words", open=False):
+                         with gr.Accordion("Lora trigger words", open=False):
                         		gr.Markdown("""
-                                    - **sdxl-realistic**: szn style
-                                    - **stylesdxl-cyberpunk**: szn 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
-                                    - **flux-lora-pro-headshot**: PROHEADSHOT
-                            		""",
-                                    label="Trigger words")                       
+                        		### 🎨 Lora trigger words:
+                                            - **sdxl-realistic**: szn style
+                                            - **stylesdxl-cyberpunk**: szn 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
+                        		""")                       
                             
                     with gr.Row():
-                        custom_lora = gr.Dropdown([" ", "jwu114/lora-sdxl-realistic", "issaccyj/lora-sdxl-cyberpunk", "hugovntr/flux-schnell-realism", "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", "dvyio/flux-lora-pro-headshot"], label="Custom LoRA", info="Please select from the list")
+                        custom_lora = gr.Dropdown([" ", "jwu114/lora-sdxl-realistic", "issaccyj/lora-sdxl-cyberpunk", "hugovntr/flux-schnell-realism", "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",)
                     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=" bad anatomy, bad hands, three hands, three legs, bad arms, missing legs, missing arms, poorly drawn face, bad face, fused face, cloned face, worst face, three crus, extra crus, fused crus, worst feet, three feet, fused feet, fused thigh, three thigh, fused thigh, extra thigh, worst thigh, missing fingers, extra fingers, ugly fingers, long fingers, horn, extra eyes, huge eyes, 2girl, amputation, disconnected limbs, cartoon, cg, 3d, unreal, animate", 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)
+                                width = gr.Slider(label="Width", value=896, minimum=64, maximum=1216, step=32)
+                                height = gr.Slider(label="Height", value=1152, minimum=64, maximum=1216, step=32)
                             steps = gr.Slider(label="Sampling steps", value=50, 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", "DEIS", "LMS", "DPM Adaptive", "DPM++ 2M", "DPM++ S", "DPM++ SDE", "DDPM", "DPM Fast", "Euler", "Euler CFG PP", "Euler a", "Euler+beta", "Heun", "Heun PP2", "DDIM", "PLMS", "UniPC", "UniPC BH2"])
                             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")
-                clr_button =gr.Button("Clear prompt",variant="primary", elem_id="clear_button")
+                clr_button =gr.Button("Clear Prompt",variant="primary", elem_id="clear_button")
                 clr_button.click(lambda: gr.Textbox(value=""), None, text_prompt)
-            with gr.Column(elem_id="image-container"):                
-
-                with gr.Row():
-                    image_output = gr.Image(type="pil", label="Image Output", format="png", elem_id="gallery")
-                    
-                with gr.Row(): 
-                    clear_btn = gr.Button(value="Clear Image",variant="primary", elem_id="clear_button")
-                    clear_btn.click(clear, inputs=[ ], outputs=[image_output])
-    
-    
+                
+            with gr.Row():
+                image_output = gr.Image(type="pil", label="Image Output", format="png", elem_id="gallery")
+            with gr.Row():
+                clear_btn = gr.Button(value="Clear Image", variant="primary", elem_id="clear_button")
+                clear_btn.click(clear, inputs=[], outputs=[image_output])
     
-                gr.Examples(
-                    examples = examples,
-                    inputs = [text_prompt],
-                )
-                                
+            gr.Examples(
+                examples = examples,
+                inputs = [text_prompt],
+            )
+            
     text_button.click(query, inputs=[custom_lora, text_prompt, negative_prompt, steps, cfg, method, seed, strength, width, height], outputs=[image_output, seed_output])
 
     with gr.Tab("Flip Image"):
@@ -205,6 +202,7 @@ with gr.Blocks(theme=theme, css=css, elem_id="app-container") as app:
         with gr.Row():    
             image_button = gr.Button("Run", variant='primary')
             image_button.click(flip_image, inputs=image_input, outputs=image_output)
-            
-if __name__ == "__main__":    
-    app.launch(show_api=False, share=False)
+
+    if __name__ == "__main__": 
+        app.launch(show_api=False, share=False)
+