Update app.py

app.py

@@ -10,13 +10,13 @@ import json
 
 # Project by Nymbo
 
-API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl"
+API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl-base-1.0"
 API_TOKEN = os.getenv("HF_READ_TOKEN")
 headers = {"Authorization": f"Bearer {API_TOKEN}"}
 timeout = 100
 
 def query(prompt, is_negative=False, steps=30, cfg_scale=7, sampler="DPM++ 2M Karras", seed=-1, strength=0.7):
-    if prompt == "" or prompt is None:
+    if prompt == "" or prompt == None:
         return None
 
     key = random.randint(0, 999)
@@ -25,10 +25,10 @@ def query(prompt, is_negative=False, steps=30, cfg_scale=7, sampler="DPM++ 2M Ka
     headers = {"Authorization": f"Bearer {API_TOKEN}"}
     
     prompt = GoogleTranslator(source='ru', target='en').translate(prompt)
-    print(f'\033[1mГенерация {key} перевод:\033[0m {prompt}')
+    print(f'\033[1mGeneration {key} translation:\033[0m {prompt}')
 
     prompt = f"{prompt} | ultra detail, ultra elaboration, ultra quality, perfect."
-    print(f'\033[1mГенерация {key}:\033[0m {prompt}')
+    print(f'\033[1mGeneration {key}:\033[0m {prompt}')
     
     payload = {
         "inputs": prompt,
@@ -41,20 +41,19 @@ def query(prompt, is_negative=False, steps=30, cfg_scale=7, sampler="DPM++ 2M Ka
 
     response = requests.post(API_URL, headers=headers, json=payload, timeout=timeout)
     if response.status_code != 200:
-        print(f"Ошибка: Не удалось получить изображение. Статус ответа: {response.status_code}")
-        print(f"Содержимое ответа: {response.text}")
+        print(f"Error: Failed to get image. Response status: {response.status_code}")
+        print(f"Response content: {response.text}")
         if response.status_code == 503:
             raise gr.Error(f"{response.status_code} : The model is being loaded")
         raise gr.Error(f"{response.status_code}")
-        return None
     
     try:
         image_bytes = response.content
         image = Image.open(io.BytesIO(image_bytes))
-        print(f'\033[1mГенерация {key} завершена!\033[0m ({prompt})')
+        print(f'\033[1mGeneration {key} completed!\033[0m ({prompt})')
         return image
     except Exception as e:
-        print(f"Ошибка при попытке открыть изображение: {e}")
+        print(f"Error when trying to open the image: {e}")
         return None
 
 css = """
@@ -62,27 +61,77 @@ css = """
 footer {visibility: hidden !important;}
 """
 
-with gr.Blocks(theme='Nymbo/Nymbo_Theme') as dalle:
+with gr.Blocks(theme='Nymbo/Alyx_Theme') as dalle:
     with gr.Tab("Basic Settings"):
         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=3, elem_id="prompt-text-input")
+                with gr.Row():
+                    with gr.Accordion("Advanced Settings", open=False):
+                        negative_prompt = gr.Textbox(label="Negative Prompt", placeholder="What should not be in the image", value="[deformed | disfigured], poorly drawn, [bad : wrong] anatomy, [extra | missing | floating | disconnected] limb, (mutated hands and fingers), blurry, text, fuzziness", lines=3, elem_id="negative-prompt-text-input")
+                        steps = gr.Slider(label="Sampling steps", value=35, minimum=1, maximum=100, step=1)
+                        cfg = gr.Slider(label="CFG Scale", value=7, minimum=1, maximum=20, step=1)
+                        method = gr.Radio(label="Sampling method", value="DPM++ 2M Karras", choices=["DPM++ 2M Karras", "DPM++ SDE Karras", "Euler", "Euler a", "Heun", "DDIM"])
+                        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.Tab("Advanced Settings"):
-        with gr.Accordion("Advanced Settings", open=True):
-            with gr.Row():
-                negative_prompt = gr.Textbox(label="Negative Prompt", placeholder="What should not be in the image", value="[deformed | disfigured], poorly drawn, [bad : wrong] anatomy, [extra | missing | floating | disconnected] limb, (mutated hands and fingers), blurry, text, fuzziness", lines=3, elem_id="negative-prompt-text-input")
-            with gr.Row():
-                steps = gr.Slider(label="Sampling steps", value=35, minimum=1, maximum=100, step=1)
-            with gr.Row():
-                cfg = gr.Slider(label="CFG Scale", value=7, minimum=1, maximum=20, step=1)
-            with gr.Row():
-                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():
-                strength = gr.Slider(label="Strength", value=0.7, minimum=0, maximum=1, step=0.001)
+    with gr.Tab("Image Editor"):
+        def sleep(im):
+            time.sleep(5)
+            return [im["background"], im["layers"][0], im["layers"][1], im["composite"]]
+
+        def predict(im):
+            return im["composite"]
+
+        with gr.Blocks() as demo:
             with gr.Row():
-                seed = gr.Slider(label="Seed", value=-1, minimum=-1, maximum=1000000000, step=1)
+                im = gr.ImageEditor(
+                    type="numpy",
+                    crop_size="1:1",
+                )
+
+    with gr.Tab("Information"):
+        with gr.Row():
+            gr.Textbox(label="Sample prompt", value="{prompt} | ultra detail, ultra elaboration, ultra quality, perfect.")
+
+        with gr.Accordion("Advanced Settings Overview", open=False):
+            gr.Markdown(
+                """ # `Alyxsissy.com`
+            ## 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.
+
+            ## 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!
+            """
+            )
+
+        with gr.Accordion("Error Codes and What They Mean", open=False):
+            gr.Markdown(
+                """ # `Alyxsissy.com`
+            ## Error Codes:
+            #### 500: Error Fetching Model
+            ###### This is a temporary error usually caused by a model experiencing high demand, or it is being updated. Try again in a few minutes.
+
+            #### 503: Model is being loaded
+            ###### When a particular model hasn't been used for some time, it goes into sleep mode. Error 503 means that the model is being loaded and will be ready within a minute.
+            """
+            )
 
     with gr.Row():
         text_button = gr.Button("Run", variant='primary', elem_id="gen-button")