Update app.py

app.py

@@ -8,9 +8,11 @@
 # furnished to do so, subject to the following conditions:
 #
 # ..
+
 import os
 import random
 import uuid
+from typing import Tuple
 
 import gradio as gr
 import numpy as np
@@ -20,8 +22,6 @@ import torch
 from diffusers import StableDiffusionXLPipeline, EulerAncestralDiscreteScheduler
 
 DESCRIPTION = """
-
-
 """
 
 def save_image(img):
@@ -54,7 +54,45 @@ if torch.cuda.is_available():
     pipe.load_lora_weights("prithivMLmods/Canopus-Realism-LoRA", weight_name="Canopus-Realism-LoRA.safetensors", adapter_name="rlms")
     pipe.set_adapters("rlms")
     pipe.to("cuda")
-    
+
+style_list = [
+    {
+        "name": "3840 x 2160",
+        "prompt": "hyper-realistic 8K image of {prompt}. ultra-detailed, lifelike, high-resolution, sharp, vibrant colors, photorealistic",
+        "negative_prompt": "cartoonish, low resolution, blurry, simplistic, abstract, deformed, ugly",
+    },
+    {
+        "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": "HD+",
+        "prompt": "hyper-realistic 2K image of {prompt}. ultra-detailed, lifelike, high-resolution, sharp, vibrant colors, photorealistic",
+        "negative_prompt": "cartoonish, low resolution, blurry, simplistic, abstract, deformed, ugly",
+    },
+    {
+        "name": "Style Zero",
+        "prompt": "{prompt}",
+        "negative_prompt": "",
+    },
+]
+
+styles = {k["name"]: (k["prompt"], k["negative_prompt"]) for k in style_list}
+
+DEFAULT_STYLE_NAME = "3840 x 2160"
+STYLE_NAMES = list(styles.keys())
+
+def apply_style(style_name: str, positive: str, negative: str = "") -> Tuple[str, str]:
+    if style_name in styles:
+        p, n = styles.get(style_name, styles[DEFAULT_STYLE_NAME])
+    else:
+        p, n = styles[DEFAULT_STYLE_NAME]
+
+    if not negative:
+        negative = ""
+    return p.replace("{prompt}", positive), n + negative
+
 @spaces.GPU(duration=60, enable_queue=True)
 def generate(
     prompt: str,
@@ -65,18 +103,19 @@ def generate(
     height: int = 1024,
     guidance_scale: float = 3,
     randomize_seed: bool = False,
+    style_name: str = DEFAULT_STYLE_NAME,
     progress=gr.Progress(track_tqdm=True),
 ):
-
-    
     seed = int(randomize_seed_fn(seed, randomize_seed))
 
+    positive_prompt, effective_negative_prompt = apply_style(style_name, prompt, negative_prompt)
+    
     if not use_negative_prompt:
-        negative_prompt = ""  # type: ignore
+        effective_negative_prompt = ""  # type: ignore
 
     images = pipe(
-        prompt=prompt,
-        negative_prompt=negative_prompt,
+        prompt=positive_prompt,
+        negative_prompt=effective_negative_prompt,
         width=width,
         height=height,
         guidance_scale=guidance_scale,
@@ -103,8 +142,7 @@ footer {
 }
 '''
 
-
-with gr.Blocks(css=css,theme="prithivMLmods/theme_brief") as demo:
+with gr.Blocks(css=css, theme="prithivMLmods/theme_brief") as demo:
 
     with gr.Group():
         with gr.Row():
@@ -117,6 +155,7 @@ with gr.Blocks(css=css,theme="prithivMLmods/theme_brief") as demo:
             )
             run_button = gr.Button("Run", scale=0)
         result = gr.Gallery(label="Result", columns=1, preview=True, show_label=False)
+    
     with gr.Accordion("Advanced options", open=False):
         use_negative_prompt = gr.Checkbox(label="Use negative prompt", value=True)
         negative_prompt = gr.Text(
@@ -136,6 +175,7 @@ with gr.Blocks(css=css,theme="prithivMLmods/theme_brief") as demo:
             visible=True
         )
         randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
+        
         with gr.Row(visible=True):
             width = gr.Slider(
                 label="Width",
@@ -151,6 +191,7 @@ with gr.Blocks(css=css,theme="prithivMLmods/theme_brief") as demo:
                 step=8,
                 value=1024,
             )
+        
         with gr.Row():
             guidance_scale = gr.Slider(
                 label="Guidance Scale",
@@ -160,6 +201,15 @@ with gr.Blocks(css=css,theme="prithivMLmods/theme_brief") as demo:
                 value=3.0,
             )
 
+        style_selection = gr.Radio(
+            show_label=True,
+            container=True,
+            interactive=True,
+            choices=STYLE_NAMES,
+            value=DEFAULT_STYLE_NAME,
+            label="Quality Style",
+        )
+
     gr.Examples(
         examples=examples,
         inputs=prompt,
@@ -174,7 +224,6 @@ with gr.Blocks(css=css,theme="prithivMLmods/theme_brief") as demo:
         outputs=negative_prompt,
         api_name=False,
     )
-    
 
     gr.on(
         triggers=[
@@ -192,10 +241,12 @@ with gr.Blocks(css=css,theme="prithivMLmods/theme_brief") as demo:
             height,
             guidance_scale,
             randomize_seed,
+            style_selection,
         ],
         outputs=[result, seed],
         api_name="run",
     )
-    
+
 if __name__ == "__main__":
-    demo.queue(max_size=40).launch()
+    demo.queue(max_size=40).launch()
+