Update app.py

app.py

@@ -86,10 +86,17 @@ styles = {k["name"]: (k["prompt"], k["negative_prompt"]) for k in style_list}
 STYLE_NAMES = list(styles.keys())
 DEFAULT_STYLE_NAME = "(No style)"
 
+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
+
 @spaces.GPU #[uncomment to use ZeroGPU]
 def infer(
     prompt,
     negative_prompt,
+    style,
     seed,
     randomize_seed,
     width,
@@ -100,7 +107,7 @@ def infer(
 ):
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
-
+    prompt, negative_prompt = apply_style(style, prompt, negative_prompt)
     generator = torch.Generator().manual_seed(seed)
 
     image = pipe(
@@ -215,6 +222,7 @@ with gr.Blocks(css=css) as demo:
         inputs=[
             prompt,
             negative_prompt,
+            style_selection,
             seed,
             randomize_seed,
             width,