Update app.py

app.py

@@ -14,7 +14,7 @@ model_options = {
     "PonyRealism": "John6666/pony-realism-v21main-sdxl"
 }
 
-# Define styles with prompts and negative prompts
+# Full style list for applying styles to the prompt
 style_list = [
     {
         "name": "(No style)",
@@ -33,7 +33,7 @@ style_list = [
     },
     {
         "name": "Anime",
-        "prompt": "anime artwork {prompt} . anime style, key visual, vibrant, studio anime, highly detailed",
+        "prompt": "anime artwork {prompt} . anime style, key visual, vibrant, studio anime,  highly detailed",
         "negative_prompt": "photo, deformed, black and white, realism, disfigured, low contrast",
     },
     {
@@ -53,7 +53,7 @@ style_list = [
     },
     {
         "name": "Fantasy art",
-        "prompt": "ethereal fantasy concept art of {prompt} . magnificent, celestial, ethereal, painterly, epic, majestic, magical, fantasy art, cover art, dreamy",
+        "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",
     },
     {
@@ -68,10 +68,12 @@ style_list = [
     },
 ]
 
+# Styles dictionary to map style names to prompts and negative prompts
 styles = {k["name"]: (k["prompt"], k["negative_prompt"]) for k in style_list}
 STYLE_NAMES = list(styles.keys())
 DEFAULT_STYLE_NAME = "(No style)"
 
+# Function to apply the selected style
 def apply_style(style_name: str, positive: str, negative: str = "") -> tuple[str, str]:
     p, n = styles.get(style_name, styles[DEFAULT_STYLE_NAME])
     return p.replace("{prompt}", positive), n + negative
@@ -128,12 +130,12 @@ def generate_image(images, model_choice, style_name, prompt, negative_prompt, fa
         faceid_all_embeds.append(faceid_embed)
 
     average_embedding = torch.mean(torch.stack(faceid_all_embeds, dim=0), dim=0)
-
-    # Apply style to the prompt and negative prompt
-    prompt, negative_prompt = apply_style(style_name, prompt, negative_prompt)
-
+    
+    # Apply the selected style
+    styled_prompt, styled_negative_prompt = apply_style(style_name, prompt, negative_prompt)
+    
     image = ip_model.generate(
-        prompt=prompt, negative_prompt=negative_prompt, faceid_embeds=average_embedding,
+        prompt=styled_prompt, negative_prompt=styled_negative_prompt, faceid_embeds=average_embedding,
         scale=likeness_strength, width=width, height=height, guidance_scale=guidance_scale, num_inference_steps=num_inference_steps
     )
 
@@ -155,7 +157,7 @@ with gr.Blocks(css=css) as demo:
     with gr.Row():
         with gr.Column():
             model_dropdown = gr.Dropdown(label="Select Model", choices=list(model_options.keys()), value="PonyRealism")
-            style_dropdown = gr.Dropdown(label="Select Style", choices=STYLE_NAMES, value=DEFAULT_STYLE_NAME)
+            style_dropdown = gr.Dropdown(label="Style", choices=STYLE_NAMES, value=DEFAULT_STYLE_NAME)
             files = gr.Files(label="Drag 1 or more photos of your face", file_types=["image"])
             uploaded_files = gr.Gallery(label="Your images", visible=False, columns=5, rows=1, height=250)
             with gr.Column(visible=False) as clear_button: