Update app.py

app.py

@@ -12,7 +12,6 @@ import io
 from io import BytesIO
 import base64
 import re
-from typing import Tuple
 from gradio_client import Client
 from fake_useragent import UserAgent
 import random
@@ -22,8 +21,8 @@ from fastapi import FastAPI
 app = FastAPI()
 @app.get("/")
 
-def query(prompt, negative_prompt, task, style_name, 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:"cfg_scale", seed:"seed" , style_name:"style_name", width:"width" , height:"height" }
+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:"cfg_scale", width:"width" , height:"height" }
     print(result)
     try:
         language = detect(prompt)
@@ -182,55 +181,7 @@ def mirror(image_output, scale_by, method, gfpgan, codeformer):
     img = Image.open(BytesIO(response2.content))
     return img
 
-#####
 
-style_list = [
-
-    {
-        "name": "2560 x 1440",
-        "prompt": "hyper-realistic 4K image of {prompt}. ultra-detailed, lifelike, high-resolution, sharp, vibrant colors, photorealistic",
-        "negative_prompt": "cartoonish, low resolution, blurry, simplistic, abstract, deformed, ugly",
-    },
-
-    {
-        "name": "Photo",
-        "prompt": "cinematic photo {prompt}. 35mm photograph, film, bokeh, professional, 4k, highly detailed",
-        "negative_prompt": "drawing, painting, crayon, sketch, graphite, impressionist, noisy, blurry, soft, deformed, ugly",
-    },   
-
-    {
-        "name": "Cinematic",
-        "prompt": "cinematic still {prompt}. emotional, harmonious, vignette, highly detailed, high budget, bokeh, cinemascope, moody, epic, gorgeous, film grain, grainy",
-        "negative_prompt": "anime, cartoon, graphic, text, painting, crayon, graphite, abstract, glitch, deformed, mutated, ugly, disfigured",
-    },
-
-    {
-        "name": "Anime",
-        "prompt": "anime artwork {prompt}. anime style, key visual, vibrant, studio anime, highly detailed",
-        "negative_prompt": "photo, deformed, black and white, realism, disfigured, low contrast",
-    },
-    {
-        "name": "3D Model",
-        "prompt": "professional 3d model {prompt}. octane render, highly detailed, volumetric, dramatic lighting",
-        "negative_prompt": "ugly, deformed, noisy, low poly, blurry, painting",
-    },
-    {
-        "name": "(No style)",
-        "prompt": "{prompt}",
-        "negative_prompt": "",
-    },
-]
-
-styles = {k["name"]: (k["prompt"], k["negative_prompt"]) for k in style_list}
-STYLE_NAMES = list(styles.keys())
-DEFAULT_STYLE_NAME = "2560 x 1440"
-
-def apply_style(style_name: str, positive: str, negative: str = "") -> Tuple[str, str]:
-    p, n = styles.get(style_name, styles[DEFAULT_STYLE_NAME])
-    if not negative:
-        negative = ""
-    return p.replace("{prompt}", positive), n + negative
-    
 
 examples = [
     "a beautiful woman with blonde hair and blue eyes",
@@ -269,15 +220,6 @@ with gr.Blocks(css=css, theme=theme, fill_width= False) as app:
             with gr.Row():
                 sampler = gr.Dropdown(value="DPM++ S", show_label=True, label="Sampling Method:", choices=[
                     "DPM++ 2M Karras", "DPM++ 2S a Karras", "DPM2 a Karras", "DPM2 Karras", "DPM++ SDE Karras", "DEIS", "LMS", "DPM Adaptive", "DPM++ 2M", "DPM2 Ancestral", "DPM++ S", "DPM++ SDE", "DDPM", "DPM Fast", "dpmpp_2s_ancestral", "Euler", "Euler CFG PP", "Euler a", "Euler Ancestral", "Euler+beta", "Heun", "Heun PP2", "DDIM", "LMS Karras", "PLMS", "UniPC", "UniPC BH2"])
-            with gr.Row(visible=True):
-                style_selection = gr.Radio(
-                    show_label=True,
-                    container=True,
-                    interactive=True,
-                    choices=STYLE_NAMES,
-                    value=DEFAULT_STYLE_NAME,
-                    label="Image Style",
-                )
             with gr.Row():
                 steps = gr.Slider(show_label=True, label="Sampling Steps:", minimum=1, maximum=50, value=35, step=1)
             with gr.Row():
@@ -295,7 +237,7 @@ with gr.Blocks(css=css, theme=theme, fill_width= False) as app:
     with gr.Column(scale=0):
         image_output = gr.Image(show_download_button=True, interactive=False, label='Generated Image 🌄', show_share_button=False, show_fullscreen_button=True, format="png", elem_id="gallery")
         
-        text_button.click(query, inputs=[prompt, negative_prompt, style_selection, task, steps, sampler, cfg_scale, seed, width, height], outputs=image_output, concurrency_limit=48)
+        text_button.click(query, inputs=[prompt, negative_prompt, task, steps, sampler, cfg_scale, seed, width, height], 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)