Update app.py

app.py

@@ -11,89 +11,79 @@ import torch
 # =====================================================
 pipe = StableDiffusionPipeline.from_pretrained(
     "stabilityai/stable-diffusion-2-1-base",
-    torch_dtype=torch.float16  # use fp16 for GPU
+    torch_dtype=torch.float16  # FP16 for GPU
 ).to("cuda")  # change to "cpu" if GPU not available
 
 # =====================================================
 # Comic Styles & Descriptions
 # =====================================================
 comic_styles = {
-    "Superhero comics": "The archetypal American comic style, defined by bold, detailed, and dynamic figures, and vibrant colors. Emphasizes action, drama, and idealism.",
-    "Alternative comics": "Independent comics outside the mainstream superhero genre. Artwork can range from realistic to highly abstract.",
-    "Noir": "Uses strong contrasts of light and shadow (chiaroscuro), deep blacks, and gritty details to create a dark and moody atmosphere.",
-    "Cartooning/Toon style": "Simplified, exaggerated, often caricatured characters. Examples: Betty Boop, modern webcomics.",
-    "Retro comics/Silver Age": "Homage to comics from the 1950s-1970s. Simpler, energetic designs, vibrant colors.",
-    "Webcomics": "A medium rather than a style, ranging from simple hand-drawn to polished professional sequential art.",
-    # Japanese manga
-    "Shōnen": "For young male audience. Bold lines, dynamic action. Examples: Dragon Ball, Naruto.",
-    "Shōjo": "For young female readers. Fluid lines, highly detailed eyes, soft and decorative aesthetic.",
-    "Seinen": "For adult male readers (18–40). More realistic/gritty art, mature themes.",
-    "Chibi": "Super-deformed, exaggeratedly cute characters with tiny bodies and large heads.",
-    "Moe": "Cute and innocent-looking characters, appealing to dedicated fanbase.",
-    "Kodomo": "For young children. Simple, rounded shapes and bright, cheerful artwork.",
-    # Techniques / hybrid styles
-    "Line art": "Emphasizes strong, distinct lines without extensive shading or coloring.",
-    "Slice of life": "Depicts everyday events, can be cartoonish or realistic, focuses on relatability.",
+    "Superhero comics": "Bold, dynamic figures and vibrant colors. Emphasizes action, drama, and idealism.",
+    "Alternative comics": "Independent comics outside the mainstream genre, from realistic to abstract.",
+    "Noir": "High contrast light and shadow, deep blacks, gritty and moody atmosphere.",
+    "Cartooning/Toon style": "Simplified, exaggerated, caricatured characters, e.g., Betty Boop.",
+    "Retro comics/Silver Age": "1950s–1970s style, simpler energetic designs and vibrant colors.",
+    "Webcomics": "Digital medium ranging from simple hand-drawn to polished sequential art.",
+    "Shōnen": "For young male audience. Bold lines, dynamic action. Examples: Dragon Ball.",
+    "Shōjo": "For young female readers. Fluid lines, detailed eyes, decorative aesthetic.",
+    "Seinen": "For adult male readers. More realistic/gritty art, mature themes.",
+    "Chibi": "Super-deformed cute characters with tiny bodies and oversized heads.",
+    "Moe": "Cute, innocent-looking characters, appealing to fans.",
+    "Kodomo": "For young children. Simple, rounded shapes, bright artwork.",
+    "Line art": "Strong, distinct lines without extensive shading or coloring.",
+    "Slice of life": "Depicts everyday events, cartoonish or realistic, relatable.",
     "Celtic art style": "Decorative, stylized, with intricate knotwork and patterns.",
-    "Ink wash": "Uses diluted ink for varying shades, soft, textured, or painted feel.",
-    "Pointillism": "Small, distinct dots of color form images, creating interesting textures.",
-    "Watercolor painting style": "Soft, painterly look with translucent watercolors and delicate gradients.",
-    "Photorealism": "Attempts photographic accuracy in figures and scenes.",
-    "Grotesque": "Intentionally distorted/exaggerated to emphasize weird, unsettling, or macabre elements."
+    "Ink wash": "Diluted ink for soft, textured, or painted feel.",
+    "Pointillism": "Small dots of color form images, creating textures.",
+    "Watercolor painting style": "Soft, painterly look with translucent gradients.",
+    "Photorealism": "Attempts photographic accuracy.",
+    "Grotesque": "Distorted/exaggerated for weird, unsettling, or macabre effect."
 }
 
 # =====================================================
 # Comic Generation Functions
 # =====================================================
-def detect_characters(img: Image.Image):
-    """Fallback character positions."""
-    width, height = img.size
-    return [(width // 4, height // 4), (3 * width // 4, height // 4)]
-
 def add_speech_bubbles(img: Image.Image, dialogues):
-    """Draw speech bubbles at placeholder positions."""
+    """Add simple speech bubbles at top-left positions."""
     draw = ImageDraw.Draw(img)
     try:
         font = ImageFont.truetype("ComicNeue-Bold.ttf", 20)
     except:
         font = ImageFont.load_default()
-
-    positions = detect_characters(img)
-    for i, text in enumerate(dialogues[:len(positions)]):
-        x, y = positions[i]
-        bubble_w, bubble_h = 160, 70
-        rect = [x - bubble_w // 2, y - bubble_h // 2, x + bubble_w // 2, y + bubble_h // 2]
+    x, y = 20, 20
+    for text in dialogues:
+        bubble_w, bubble_h = 200, 60
+        rect = [x, y, x + bubble_w, y + bubble_h]
         draw.rectangle(rect, fill="white", outline="black", width=3)
-        draw.text((x - bubble_w // 2 + 10, y - bubble_h // 2 + 10), text, font=font, fill="black")
+        draw.text((x + 10, y + 10), text, font=font, fill="black")
+        y += bubble_h + 10
     return img
 
 def generate_panel(prompt):
-    """Generate a single panel with Stable Diffusion."""
-    image = pipe(prompt).images[0]
-    return image
+    """Generate a single panel using Stable Diffusion."""
+    return pipe(prompt).images[0]
 
 def generate_comic(title, story, pages, panels, style, dialogues, explicit):
-    """Generate multi-panel comics."""
-    generated_images = []
+    """Generate multi-panel comics and return images + ZIP."""
     dialogues_list = [d.strip() for d in dialogues.splitlines() if d.strip()]
+    generated_images = []
 
-    for p in range(int(pages)):
-        for n in range(int(panels)):
+    for _ in range(int(pages)):
+        for _ in range(int(panels)):
             prompt = f"{style} comic panel ({comic_styles[style]}), theme: {story}"
             if explicit:
-                prompt += " with mature themes, blood and gore, offensive language"
+                prompt += " with mature themes, blood and offensive language"
             else:
                 prompt += " safe for all ages"
 
-            panel_img = generate_panel(prompt)
+            img = generate_panel(prompt)
 
-            # Add dialogues if provided
             if dialogues_list:
-                panel_img = add_speech_bubbles(panel_img, dialogues_list)
+                img = add_speech_bubbles(img, dialogues_list)
 
-            generated_images.append(panel_img)
+            generated_images.append(img)
 
-    # Pack into ZIP
+    # Save images to ZIP
     buffer = BytesIO()
     with zipfile.ZipFile(buffer, "w") as zf:
         for i, img in enumerate(generated_images):
@@ -108,10 +98,8 @@ def generate_comic(title, story, pages, panels, style, dialogues, explicit):
 # Gradio UI
 # =====================================================
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.Markdown(
-        "<h1 style='text-align: center;'>🦸‍♂️ AI Comic Crafter</h1>"
-        "<p style='text-align: center;'>Generate multi-page, multi-panel comics with AI! 🎨</p>"
-    )
+    gr.Markdown("<h1 style='text-align:center'>🦸‍♂️ AI Comic Crafter</h1>"
+                "<p style='text-align:center'>Generate multi-page, multi-panel comics with AI! 🎨</p>")
 
     with gr.Row():
         with gr.Column(scale=1):
@@ -119,8 +107,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             story = gr.Textbox(label="Story Setup", placeholder="Describe characters, setting, and plot...")
             pages = gr.Slider(1, 5, value=2, step=1, label="Number of Pages")
             panels = gr.Slider(1, 6, value=3, step=1, label="Panels Per Page")
-
-            # Style dropdown and description
+            
             style_dropdown = gr.Dropdown(
                 choices=list(comic_styles.keys()),
                 label="Art Style / Theme",
@@ -138,20 +125,12 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                 outputs=style_description
             )
 
-            dialogues = gr.Textbox(
-                label="Custom Dialogues (Optional)",
-                placeholder="One dialogue per line...",
-                lines=5
-            )
+            dialogues = gr.Textbox(label="Custom Dialogues (Optional)", placeholder="One dialogue per line...", lines=5)
             explicit = gr.Checkbox(label="Allow Explicit Content (blood/gore, offensive language)", value=False)
             generate_btn = gr.Button("⚡ Generate Comic")
 
         with gr.Column(scale=1):
-            output_gallery = gr.Gallery(
-                label="Your Comic Preview",
-                columns=2,
-                height="auto"
-            )
+            output_gallery = gr.Gallery(label="Your Comic Preview", columns=2, height="auto")
             download_file = gr.File(label="Download Complete Comic (.zip)", type="binary")
 
     generate_btn.click(