css

app.py

@@ -1,8 +1,42 @@
 import gradio as gr
+from typing import Tuple
 
 # Load the model
 model = gr.load("models/strangerzonehf/Flux-Midjourney-Mix2-LoRA")
 
+# List of styles
+style_list = [
+    {"name": "D&D Art", "prompt": "dungeons & dragons style artwork {prompt}. d&d style, key visual, vibrant, studio anime, highly detailed", 
+     "negative_prompt": "photo, deformed, black and white, realism, disfigured, low contrast"},
+    {"name": "2560 x 1440", "prompt": "hyper-realistic 4K image of {prompt}. ultra-detailed, lifelike, high-resolution, sharp, vibrant colors, photorealistic, fantastical environments, epic scenery, rich textures", 
+     "negative_prompt": "cartoonish, low resolution, blurry, simplistic, abstract, sci-fi, modern cityscapes, deformed, ugly"},
+    {"name": "Character Portrait", "prompt": "dungeons & dragons character portrait of {prompt}. intricate details, expressive faces, heroic poses, rich textures, elaborate costumes, iconic weapons, fantasy aesthetic, studio-quality rendering", 
+     "negative_prompt": "generic, overly stylized, blurry, simplistic, futuristic, deformed"},
+    {"name": "Epic Battle", "prompt": "dynamic dungeons & dragons artwork of {prompt}. epic battle scene, legendary heroes, fierce monsters, intense action, dramatic lighting, high-detail environment, magical effects, vibrant colors", 
+     "negative_prompt": "peaceful, mundane, low energy, modern, sci-fi, simplistic, cartoonish, low contrast"},
+    {"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 = "(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
+
+def generate_image(style: str, positive_prompt: str):
+    styled_prompt, styled_negative = apply_style(style, positive_prompt)
+    result = model(
+        inputs={
+            "prompt": styled_prompt,
+            "negative_prompt": styled_negative,
+        }
+    )
+    return result
+
 # Enhanced D&D-themed CSS with previously used styles and new additions
 custom_css = """
 body {
@@ -123,10 +157,13 @@ button:hover {
 }
 """
 
-# Define Gradio interface with enhanced description and external links
+# Define Gradio interface
 iface = gr.Interface(
-    fn=model,
-    inputs=gr.Textbox(lines=3, label="🎲 Enter Your Quest:", placeholder="Describe your scene, hero, or epic landscape..."),
+    fn=generate_image,
+    inputs=[
+        gr.Dropdown(label="🎨 Choose Style", choices=STYLE_NAMES, value=DEFAULT_STYLE_NAME),
+        gr.Textbox(label="🎲 Enter Your Quest:", placeholder="Describe your scene, hero, or epic landscape..."),
+    ],
     outputs=gr.Image(type="pil", label="🖼️ Your Legendary Vision"),
     title="🛡️ Dungeons & Dragons Image Generator ⚔️",
     description="**Unleash Your Imagination!** Create heroes, maps, quests, and epic scenes to bring your campaigns to life. "