Update app.py

app.py

@@ -14,6 +14,53 @@ DESCRIPTION = "# T2I-Adapter-SDXL Sketch"
 if not torch.cuda.is_available():
     DESCRIPTION += "\n<p>Running on CPU 🥶 This demo does not work on CPU.</p>"
 
+styles = [
+    {
+        "name": "cinematic-default",
+        "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": "sai-3d-model",
+        "prompt": "professional 3d model {prompt} . octane render, highly detailed, volumetric, dramatic lighting",
+        "negative_prompt": "ugly, deformed, noisy, low poly, blurry, painting"
+    },
+    {
+        "name": "sai-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": "sai-digital art",
+        "prompt": "concept art {prompt} . digital artwork, illustrative, painterly, matte painting, highly detailed",
+        "negative_prompt": "photo, photorealistic, realism, ugly"
+    },
+    {
+        "name": "sai-photographic",
+        "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": "sai-pixel art",
+        "prompt": "pixel-art {prompt} . low-res, blocky, pixel art style, 8-bit graphics",
+        "negative_prompt": "sloppy, messy, blurry, noisy, highly detailed, ultra textured, photo, realistic"
+    },
+    {
+        "name": "sai-fantasy art",
+        "prompt": "ethereal fantasy concept art of  {prompt} . magnificent, celestial, ethereal, painterly, epic, majestic, magical, fantasy art, cover art, dreamy",
+        "negative_prompt": "photographic, realistic, realism, 35mm film, dslr, cropped, frame, text, deformed, glitch, noise, noisy, off-center, deformed, cross-eyed, closed eyes, bad anatomy, ugly, disfigured, sloppy, duplicate, mutated, black and white"
+    },
+]
+
+styles = {k['name']: (k['prompt'], k['negative_prompt']) for k in styles}
+default_style = styles['sai-photographic']
+style_names = list(styles.keys())
+
+
+def apply_style(style, positive, negative=""):
+    p, n = styles.get(style, default_style)
+    return p.replace('{prompt}', positive), n + negative
+
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 if torch.cuda.is_available():
     model_id = "stabilityai/stable-diffusion-xl-base-1.0"
@@ -84,6 +131,11 @@ with gr.Blocks() as demo:
             prompt = gr.Textbox(label="Prompt")
             run_button = gr.Button("Run")
             with gr.Accordion("Advanced options", open=False):
+                style = gr.Dropdown(
+                    choices=style_names,
+                    value=default_style,
+                    label="Style"
+                )
                 negative_prompt = gr.Textbox(
                     label="Negative prompt", value="extra digit, fewer digits, cropped, worst quality, low quality"
                 )
@@ -140,6 +192,25 @@ with gr.Blocks() as demo:
         outputs=result,
         api_name=False,
     )
+    prompt.submit(
+        fn=apply_style,
+        inputs=[style, prompt, negative_prompt],
+        outputs=[prompt, negative_prompt],
+        queue=False,
+        api_name=False,
+    ).then(
+        fn=run,
+        inputs=inputs,
+        outputs=result,
+        api_name=False,
+    )
+    style.change(
+        fn=apply_style,
+        inputs=[style, prompt, negative_prompt],
+        outputs=[prompt, negative_prompt],
+        queue=False,
+        api_name=False,
+    )
     run_button.click(
         fn=randomize_seed_fn,
         inputs=[seed, randomize_seed],