Update app.py

app.py

@@ -15,14 +15,10 @@ import re
 from gradio_client import Client
 from fake_useragent import UserAgent
 import random
-from theme import theme
-from fastapi import FastAPI
 
-app = FastAPI()
-@app.get("/")
 
-def query(prompt, negative_prompt, task, steps=35, sampler="DPM++ 2M Karras", cfg_scale=7, seed=-1, width=896, height=1152):
-    result = {prompt:"prompt", negative_prompt:"negative_prompt", task:"task", steps:"steps", sampler:"sampler", cfg_scale:print(result)}
+def flip_text(prompt, negative_prompt, task, steps, sampler, cfg_scale, seed):
+    result = {"prompt": prompt,"negative_prompt": negative_prompt,"task": task,"steps": steps,"sampler": sampler,"cfg_scale": cfg_scale,"seed": seed}
     print(result)
     try:
         language = detect(prompt)
@@ -179,39 +175,36 @@ def mirror(image_output, scale_by, method, gfpgan, codeformer):
     img = Image.open(BytesIO(response2.content))
     return img
 
-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 to style the app
 css = """
-.gradio-container {background-color: MediumAquaMarine}
-
-footer {
-    visibility: hidden;
+#generate {
+    width: 100%;
+    background: #e253dd !important;
+    border: none;
+    border-radius: 50px;
+    outline: none !important;
+    color: white;
+}
+#generate:hover {
+    background: #de6bda !important;
+    outline: none !important;
+    color: #fff;
+    }
+#image_output {
+height: 100% !important;
 }
 """
 
-with gr.Blocks(css=css, theme=theme, fill_width= False) as app:
+with gr.Blocks(css=css) as demo:
 
-    with gr.Tab("Basic Settings"):
+    with gr.Tab("Базовые настройки"):
         with gr.Row():
-            prompt = gr.Textbox(placeholder="Enter the image description...", show_label=True, label='Description of the image:', lines=3)
+            prompt = gr.Textbox(placeholder="Введите описание изображения...", show_label=True, label='Описание изображения:', lines=3)
         with gr.Row():
-            task = gr.Radio(interactive=True, value="Stable Diffusion XL 1.0", show_label=True, label="Model of neural network:", choices=['Stable Diffusion XL 1.0', 'Crystal Clear XL', 
+            task = gr.Radio(interactive=True, value="Stable Diffusion XL 1.0", show_label=True, label="Модель нейросети:", choices=['Stable Diffusion XL 1.0', 'Crystal Clear XL', 
                                                                                                               'Juggernaut XL', 'DreamShaper XL',
                                                                                                               'SDXL Niji', 'Cinemax SDXL', 'NightVision XL',
                                                                                                               'Playground v2', 'Artigen v3'])
-            gr.Examples(
-                examples = examples,    
-                inputs = [prompt], 
-            )
-            
-
-            
-    with gr.Tab("Extended settings"):
+    with gr.Tab("Расширенные настройки"):
         with gr.Row():
             negative_prompt = gr.Textbox(placeholder="Negative Prompt", show_label=True, label='Negative Prompt:', lines=3, value="[deformed | disfigured], poorly drawn, [bad : wrong] anatomy, [extra | missing | floating | disconnected] limb, (mutated hands and fingers), blurry")
         with gr.Row():
@@ -224,31 +217,26 @@ with gr.Blocks(css=css, theme=theme, fill_width= False) as app:
         with gr.Row():
             seed = gr.Number(show_label=True, label="Seed:", minimum=-1, maximum=1000000, value=-1, step=1)
     
-    with gr.Tab("Upscaling Settings"):
+    with gr.Tab("Настройки апскейлинга"):
         with gr.Column():
             with gr.Row():
-                scale_by = gr.Number(show_label=True, label="How many times to increase:", minimum=1, maximum=2, value=2, step=1)
+                scale_by = gr.Number(show_label=True, label="Во сколько раз увеличить:", minimum=1, maximum=2, value=2, step=1)
             with gr.Row():
-                method = gr.Dropdown(show_label=True, value="ESRGAN_4x", label="Algorithm to enlarge", choices=["ScuNET GAN", "SwinIR 4x", "ESRGAN_4x", "R-ESRGAN 4x+", "R-ESRGAN 4x+ Anime6B"])
+                method = gr.Dropdown(show_label=True, value="ESRGAN_4x", label="Алгоритм увеличения", choices=["ScuNET GAN", "SwinIR 4x", "ESRGAN_4x", "R-ESRGAN 4x+", "R-ESRGAN 4x+ Anime6B"])
         with gr.Column():
             with gr.Row():
-                gfpgan = gr.Slider(show_label=True, label="Эффект GFPGAN (to improve the face)", minimum=0, maximum=1, value=0, step=0.1)
+                gfpgan = gr.Slider(show_label=True, label="Эффект GFPGAN (для улучшения лица)", minimum=0, maximum=1, value=0, step=0.1)
             with gr.Row():
-                codeformer = gr.Slider(show_label=True, label="Effect CodeFormer (to improve the face)", minimum=0, maximum=1, value=0, step=0.1)
+                codeformer = gr.Slider(show_label=True, label="Эффект CodeFormer (для улучшения лица)", minimum=0, maximum=1, value=0, step=0.1)
     
     with gr.Column():
-        text_button = gr.Button("Generate image", variant='primary', elem_id="generate")
+        text_button = gr.Button("Сгенерировать изображение", variant='primary', elem_id="generate")
     with gr.Column():
         image_output = gr.Image(show_download_button=True, interactive=False, label='Результат:', elem_id='image_output', type='filepath')
-        text_button.click(query, inputs=[prompt, negative_prompt, task, steps, sampler, cfg_scale, seed], outputs=image_output, concurrency_limit=48)
-        clear_prompt =gr.Button("Clear 🗑️",variant="primary", elem_id="clear_button")
-        clear_prompt.click(lambda: (None, None), None, [prompt, image_output], queue=False, show_api=False)
+        text_button.click(flip_text, inputs=[prompt, negative_prompt, task, steps, sampler, cfg_scale, seed], outputs=image_output, concurrency_limit=48)
         
-        img2img_b = gr.Button("Increase the image", variant='secondary')
-        image_i2i = gr.Image(show_label=True, label='Increased image:')
+        img2img_b = gr.Button("Увеличить изображение", variant='secondary')
+        image_i2i = gr.Image(show_label=True, label='Увеличенное изображение:')
         img2img_b.click(mirror, inputs=[image_output, scale_by, method, gfpgan, codeformer], outputs=image_i2i, concurrency_limit=48)
     
-app.queue(default_concurrency_limit=200, max_size=200)  # <-- Sets up a queue with default parameters
-if __name__ == "__main__":
-    app.launch()
-    
+demo.launch()