css

app.py

@@ -1,24 +1,18 @@
 import gradio as gr
 import numpy as np
 import random
+import spaces
 import torch
 from diffusers import DiffusionPipeline
-from typing import Dict, Tuple
+
+# Load the model
+dtype = torch.bfloat16
+device = "cuda" if torch.cuda.is_available() else "cpu"
+pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=dtype).to(device)
 
 # Constants
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 2048
-DEFAULT_MODEL = "black-forest-labs/FLUX.1-schnell"
-DEFAULT_STYLE_NAME = "D&D Map"
-
-# Load the default model
-def load_model(model_name: str = DEFAULT_MODEL):
-    dtype = torch.bfloat16
-    device = "cuda" if torch.cuda.is_available() else "cpu"
-    return DiffusionPipeline.from_pretrained(model_name, torch_dtype=dtype).to(device)
-
-# Initialize the model
-pipe = load_model()
 
 # Style list for prompt customization
 style_list = [
@@ -206,62 +200,54 @@ style_list = [
     {"name": "(No style)", "prompt": "{prompt}", "negative_prompt": ""},
 ]
 
-# Create a styles dictionary for quick lookup
-STYLES: Dict[str, Tuple[str, str]] = {
-    style["name"]: (style["prompt"], style["negative_prompt"]) for style in STYLE_LIST
-}
+styles = {k["name"]: (k["prompt"], k["negative_prompt"]) for k in style_list}
+STYLE_NAMES = list(styles.keys())
+DEFAULT_STYLE_NAME = "D&D Map"
 
 # Function to apply selected style
-def apply_style(style_name: str, prompt: str, negative_prompt: str = "") -> Tuple[str, str]:
-    positive, negative = STYLES.get(style_name, STYLES[DEFAULT_STYLE_NAME])
-    return positive.replace("{prompt}", prompt), negative + (negative_prompt or "")
-
-# Inference function for generating images
-def generate_image(
-    prompt: str,
-    style: str,
-    seed: int = 42,
-    randomize_seed: bool = False,
-    width: int = 1024,
-    height: int = 1024,
-    num_inference_steps: int = 4,
-    progress: gr.Progress = gr.Progress(track_tqdm=True)
-) -> Tuple[str, int]:
-    try:
-        # Randomize seed if requested
-        if randomize_seed:
-            seed = random.randint(0, MAX_SEED)
-
-        # Apply style to prompt
-        styled_prompt, negative_prompt = apply_style(style, prompt)
-
-        # Generate image
-        generator = torch.Generator().manual_seed(seed)
-        with torch.no_grad():
-            image = pipe(
-                prompt=styled_prompt,
-                width=width,
-                height=height,
-                num_inference_steps=num_inference_steps,
-                generator=generator,
-                guidance_scale=7.5,
-                negative_prompt=negative_prompt,
-            ).images[0]
-        
-        return image, seed
-    except Exception as e:
-        return f"Error during image generation: {str(e)}", seed
+def apply_style(style_name: str, positive: str, negative: str = ""):
+    p, n = styles.get(style_name, styles[DEFAULT_STYLE_NAME])
+    return p.replace("{prompt}", positive), n + (negative or "")
+
+# Inference function
+@spaces.GPU()
+def infer(
+    prompt,
+    style,
+    seed=42,
+    randomize_seed=False,
+    width=1024,
+    height=1024,
+    num_inference_steps=4,
+    progress=gr.Progress(track_tqdm=True),
+):
+    if randomize_seed:
+        seed = random.randint(0, MAX_SEED)
+    
+    # Apply style to prompt
+    styled_prompt, negative_prompt = apply_style(style, prompt)
+    generator = torch.Generator().manual_seed(seed)
+    image = pipe(
+        prompt=styled_prompt,
+        width=width,
+        height=height,
+        num_inference_steps=num_inference_steps,
+        generator=generator,
+        guidance_scale=0.0,
+        negative_prompt=negative_prompt,
+    ).images[0]
+    return image, seed
 
-# Example prompts for inspiration
-EXAMPLES = [
+# Example prompts
+examples = [
     ["A 10x8 battle map located in Ravenloft", "D&D Map"],
-    ["A hero wielding a glowing sword atop a cliff", "Heroic Portrait"],
-    ["A sprawling dungeon with traps and treasures", "Dungeon Map"],
-    ["A fierce dragon battling adventurers in a cavern", "Epic Battle"],
+    ["A heroic adventurer wielding a flaming sword standing on a cliff", "D&D Art"],
+    ["A mystical library with ancient scrolls and glowing runes", "Dark Fantasy"],
+    ["A ferocious dragon breathing fire in a dark cavern", "Epic Battle"],
 ]
 
-# Custom CSS for a thematic UI
-CUSTOM_CSS = """
+css = """
+/* General body styling */
 body {
     background-color: #1b1b1b;
     font-family: 'Cinzel', serif;
@@ -270,9 +256,11 @@ body {
     margin: 0;
     padding: 0;
 }
+
+/* Container */
 #col-container {
     margin: 0 auto;
-    max-width: 600px;
+    max-width: 550px;
     padding: 20px;
     border: 6px solid #8b4513;
     background: radial-gradient(circle, #302d2b 0%, #3b3634 60%, #1b1b1b 100%);
@@ -280,31 +268,124 @@ body {
     box-shadow: 0 0 30px rgba(0, 0, 0, 0.9), 0 0 15px rgba(139, 69, 19, 0.7);
     transition: transform 0.3s ease-in-out, box-shadow 0.3s ease-in-out;
 }
+
 #col-container:hover {
     transform: scale(1.03);
     box-shadow: 0 0 40px rgba(255, 223, 0, 0.8), 0 0 20px rgba(255, 140, 0, 0.8);
 }
-h1, h2, h3 {
+
+/* Fancy title and text effects */
+#col-container h1, #col-container h2, #col-container h3 {
     color: #ffebcc;
     text-shadow: 0 0 10px #8b4513, 0 0 20px #ffd700;
 }
+
+#col-container p {
+    color: #dcdcdc;
+    text-shadow: 0 0 6px rgba(139, 69, 19, 0.8);
+}
+
+/* Markdown links */
+#col-container a {
+    color: #e6ac00;
+    text-decoration: none;
+    font-weight: bold;
+    transition: color 0.3s ease-in-out, text-shadow 0.3s ease-in-out;
+}
+
+#col-container a:hover {
+    color: #fff700;
+    text-shadow: 0 0 10px #fff700, 0 0 20px #ffebcc;
+}
+
+/* Buttons with D&D glowing effect */
+.gr-button {
+    background-color: #8b4513;
+    color: #f5f5f5;
+    font-weight: bold;
+    padding: 10px 20px;
+    border: 2px solid #b8860b;
+    border-radius: 8px;
+    cursor: pointer;
+    transition: background-color 0.4s ease-in-out, box-shadow 0.4s ease-in-out, transform 0.2s ease-in-out;
+}
+
+.gr-button:hover {
+    background-color: #b8860b;
+    transform: scale(1.05);
+    box-shadow: 0 0 12px rgba(255, 223, 0, 0.9), 0 0 24px rgba(255, 140, 0, 0.7);
+}
+
+/* Input and Dropdown styling */
+input, select, .gr-input, .gr-slider {
+    background-color: #302d2b;
+    color: #b59e54;
+    border: 2px solid #8b4513;
+    border-radius: 5px;
+    padding: 8px 12px;
+    font-family: 'Cinzel', serif;
+    transition: border-color 0.3s ease, box-shadow 0.3s ease;
+}
+
+input:focus, select:focus, .gr-input:focus, .gr-slider:focus {
+    outline: none;
+    border-color: #ffd700;
+    box-shadow: 0 0 8px #ffd700;
+}
+
+/* Slider styling */
+.gr-slider > div {
+    background-color: #8b4513 !important;
+}
+
+/* Checkbox customization */
+.gr-checkbox-label {
+    font-weight: bold;
+    color: #f5f5f5;
+    text-shadow: 0 0 5px rgba(139, 69, 19, 0.7);
+}
+
+/* Accordion effects */
+.gr-accordion {
+    background: linear-gradient(145deg, #302d2b, #3b3634);
+    color: #f5f5f5;
+    border: 2px solid #8b4513;
+    border-radius: 10px;
+    padding: 8px;
+    transition: all 0.3s ease;
+}
+
+.gr-accordion:hover {
+    box-shadow: 0 0 12px rgba(255, 223, 0, 0.7), 0 0 20px rgba(255, 140, 0, 0.5);
+}
+
+/* Image glow effect */
+.gr-image {
+    border: 4px solid #8b4513;
+    border-radius: 10px;
+    box-shadow: 0 0 10px #8b4513, 0 0 20px #ffd700;
+}
+
+/* Add glow on hover for images */
+.gr-image:hover {
+    box-shadow: 0 0 20px rgba(255, 140, 0, 0.7), 0 0 40px rgba(255, 223, 0, 0.9);
+}
 """
 
-# Gradio interface
-with gr.Blocks(css=CUSTOM_CSS) as demo:
+# Interface
+with gr.Blocks(css=css) as demo:
     with gr.Column(elem_id="col-container"):
         # Title and Description
         gr.Markdown(
             """
-            # 🛡️ D&D Art & Map Generator ⚔️
-            **Create stunning artwork and maps for your Dungeons & Dragons campaigns!**
-            - Generate battle maps, character portraits, and epic scenes.
-            - Tailored for Dungeon Masters and adventurers alike.
-            - [Visit Our Website](https://chatdnd.net) | [Support Us](https://buymeacoffee.com/watchoutformike)
+            # 🛡️ ChatDnD.net D&D Map Generator ⚔️
+            **Unleash Your Imagination!** Create heroes, maps, quests, and epic scenes to bring your campaigns to life. 
+            Tailored for adventurers seeking inspiration or Dungeon Masters constructing their next grand story. <br>
+            [Visit Our Website](https://chatdnd.net) | [Support Us](https://buymeacoffee.com/watchoutformike)
             """
         )
 
-        # Input fields
+        # Prompt input and style selector
         with gr.Row():
             prompt = gr.Textbox(
                 label="🎲 Describe Your Vision:",
@@ -313,7 +394,7 @@ with gr.Blocks(css=CUSTOM_CSS) as demo:
             )
             style = gr.Dropdown(
                 label="🎨 Select a Style",
-                choices=list(STYLES.keys()),
+                choices=STYLE_NAMES,
                 value=DEFAULT_STYLE_NAME,
             )
         
@@ -321,7 +402,7 @@ with gr.Blocks(css=CUSTOM_CSS) as demo:
         with gr.Row():
             run_button = gr.Button("Generate Image")
         result = gr.Image(label="🖼️ Your Legendary Vision")
-
+        
         # Advanced settings
         with gr.Accordion("⚙️ Advanced Settings", open=False):
             seed = gr.Slider(
@@ -357,23 +438,19 @@ with gr.Blocks(css=CUSTOM_CSS) as demo:
                 value=4,
             )
 
-        # Examples
+        # Examples with styles
         gr.Examples(
-            examples=EXAMPLES,
+            examples=examples,
             inputs=[prompt, style],
             outputs=[result],
-            fn=generate_image,
+            fn=infer,
             cache_examples="lazy",
         )
 
-    # Attach event handlers
-    run_button.click(
-        fn=generate_image,
-        inputs=[prompt, style, seed, randomize_seed, width, height, num_inference_steps],
-        outputs=[result, seed],
-    )
-    prompt.submit(
-        fn=generate_image,
+    # Interactivity
+    gr.on(
+        triggers=[run_button.click, prompt.submit],
+        fn=infer,
         inputs=[prompt, style, seed, randomize_seed, width, height, num_inference_steps],
         outputs=[result, seed],
     )