Update app.py

app.py

@@ -1,33 +1,60 @@
 import gradio as gr
 from PIL import Image, ImageDraw, ImageFont
-import textwrap, os
+import textwrap, os, hashlib
 
-# -------- Fonts --------
+# ---------------- Fonts ----------------
 CANDIDATE_FONTS = [
     "/usr/share/fonts/truetype/dejavu/DejaVuSans-Bold.ttf",
     "/usr/share/fonts/truetype/dejavu/DejaVuSans.ttf",
 ]
-def get_font(size: int) -> ImageFont.FreeTypeFont:
+def get_font(size: int):
     for p in CANDIDATE_FONTS:
         if os.path.exists(p):
             return ImageFont.truetype(p, size=size)
     return ImageFont.load_default()
 
-# -------- Text drawing helpers --------
-def draw_text_block(draw, text, img_w, y, font, fill, stroke_fill, stroke_width, align="center"):
-    """Draw wrapped multi-line text with stroke at given Y; returns last Y used."""
+# ---------------- Style Presets ----------------
+PRESETS = {
+    "None": "",
+    "Retro Comic": "bold comic outline, grain, 35mm scan, high contrast",
+    "Vaporwave": "vaporwave, neon pink and cyan, miami sunset, grid floor, retro synth",
+    "Game Boy": "4-color green palette, pixelated, 160x144 vibe, dithering",
+    "Newspaper Halftone": "b&w halftone dots, newspaper print texture, vintage headline vibe",
+    "Cyberpunk Neon": "rainy night, neon kanji signs, purple and blue rim light, blade runner",
+    "90s Web": "early web aesthetic, gradients, bevel buttons, clipart stars, lens flare",
+    "Synthwave Grid": "purple/indigo gradient sky, mountain silhouette, glowing sun, grid floor",
+}
+
+# ---------------- Utility: gradient fallback (fast, offline) ----------------
+def gradient_from_prompt(prompt: str, w=768, h=768):
+    hsh = hashlib.sha256((prompt or "meme").encode()).hexdigest()
+    c1 = tuple(int(hsh[i:i+2], 16) for i in (0, 2, 4))
+    c2 = tuple(int(hsh[i:i+2], 16) for i in (6, 8, 10))
+    c1 = tuple(min(255, int(v*1.2)) for v in c1)
+    c2 = tuple(min(255, int(v*1.1)) for v in c2)
+    img = Image.new("RGB", (w, h), c1)
+    px = img.load()
+    for y in range(h):
+        t = y / (h-1)
+        r = int(c1[0]*(1-t) + c2[0]*t)
+        g = int(c1[1]*(1-t) + c2[1]*t)
+        b = int(c1[2]*(1-t) + c2[2]*t)
+        for x in range(w):
+            px[x, y] = (r, g, b)
+    return img
+
+# ---------------- Text helpers ----------------
+def wrap_lines(draw, text, img_w, font, stroke):
     lines = []
     max_chars = max(12, min(30, img_w // 30))
     for paragraph in (text or "").split("\n"):
         wrapped = textwrap.wrap(paragraph, width=max_chars)
         lines.extend(wrapped if wrapped else [""])
+    heights = [draw.textbbox((0, 0), ln, font=font, stroke_width=stroke)[3] for ln in lines]
+    return lines, heights
 
-    # measure each line height
-    line_heights = []
-    for line in lines:
-        bbox = draw.textbbox((0, 0), line, font=font, stroke_width=stroke_width)
-        line_heights.append(bbox[3] - bbox[1])
-
+def draw_block(draw, text, img_w, y, font, fill, stroke_fill, stroke_width, align="center"):
+    lines, heights = wrap_lines(draw, text, img_w, font, stroke_width)
     curr_y = y
     for i, line in enumerate(lines):
         w = draw.textlength(line, font=font, stroke_width=stroke_width)
@@ -35,105 +62,165 @@ def draw_text_block(draw, text, img_w, y, font, fill, stroke_fill, stroke_width,
             x = (img_w - w) / 2
         elif align == "left":
             x = int(img_w * 0.05)
-        else:  # right
+        else:
             x = img_w - int(img_w * 0.05) - w
         draw.text((x, curr_y), line, font=font, fill=fill,
                   stroke_width=stroke_width, stroke_fill=stroke_fill)
-        curr_y += line_heights[i] + int(font.size * 0.25)
-    return curr_y
-
-# -------- Core meme function --------
-def make_meme(
-    image, top_text, bottom_text,
+        curr_y += heights[i] + int(font.size * 0.25)
+    total_h = sum(heights) + (len(heights)-1) * int(font.size*0.25)
+    return curr_y, total_h
+
+def smart_split_text(prompt: str):
+    p = (prompt or "").strip()
+    if not p:
+        return "TOP TEXT", "BOTTOM TEXT"
+    for sep in ["|", " - ", " — ", ":", ";"]:
+        if sep in p:
+            a, b = p.split(sep, 1)
+            return a.strip().upper(), b.strip().upper()
+    words = p.split()
+    if len(words) > 6:
+        mid = len(words) // 2
+        return " ".join(words[:mid]).upper(), " ".join(words[mid:]).upper()
+    return p.upper(), ""
+
+# ---------------- Optional: call public FLUX Space ----------------
+def try_generate_with_flux(prompt: str, width: int, height: int):
+    from gradio_client import Client
+    # Using a popular public Space; if its API changes or rate-limits, we'll fallback.
+    client = Client("black-forest-labs/FLUX.1-schnell")
+    try:
+        result = client.predict(prompt, width, height, api_name="/predict")
+        if isinstance(result, list): result = result[0]
+        return Image.open(result)
+    except Exception as e:
+        # last-chance alternate endpoint some Spaces expose
+        try:
+            result = client.predict(prompt, api_name="/run")
+            if isinstance(result, list): result = result[0]
+            return Image.open(result)
+        except Exception:
+            raise e
+
+# ---------------- Main pipeline ----------------
+def generate_and_meme(
+    prompt, preset_name, use_ai, width, height,
     font_size, stroke_width, text_color, outline_color,
-    align, top_nudge, bottom_nudge
+    align, top_nudge, bottom_nudge, use_prompt_for_text, top_text_manual, bottom_text_manual
 ):
-    if image is None:
-        return None
-
-    img = image.convert("RGB")
+    base = (prompt or "").strip()
+    style_suffix = PRESETS.get(preset_name or "None", "")
+    gen_prompt = (base + " " + style_suffix).strip()
+
+    # 1) Image
+    if use_ai:
+        try:
+            img = try_generate_with_flux(gen_prompt, width, height)
+        except Exception:
+            img = gradient_from_prompt(gen_prompt, w=width, h=height)
+    else:
+        img = gradient_from_prompt(gen_prompt, w=width, h=height)
+
+    img = img.convert("RGB")
     draw = ImageDraw.Draw(img)
     w, h = img.size
 
-    # font scales with width (feels right on mobile photos)
+    # 2) Meme text
+    if use_prompt_for_text:
+        top_text, bottom_text = smart_split_text(base)
+    else:
+        top_text, bottom_text = (top_text_manual or "").upper(), (bottom_text_manual or "").upper()
+
+    # 3) Draw
     base_size = max(12, int((w * font_size) / 100))
     font = get_font(base_size)
     stroke = int(max(0, stroke_width))
 
-    # Top block
     top_y = int(h * 0.03) + int(top_nudge)
-    draw_text_block(
-        draw=draw, text=(top_text or "").strip(), img_w=w, y=top_y, font=font,
-        fill=text_color, stroke_fill=outline_color, stroke_width=stroke, align=align
-    )
-
-    # Bottom block: measure height first to anchor from bottom
-    lines = textwrap.wrap(bottom_text or "", width=max(20, w // 30))
-    line_heights = [draw.textbbox((0, 0), l, font=font, stroke_width=stroke)[3] for l in lines]
-    total_bottom_h = sum(line_heights) + (len(lines) - 1) * int(font.size * 0.25)
-    bottom_y_start = int(h - total_bottom_h - h * 0.03) - int(bottom_nudge)
-
-    draw_text_block(
-        draw=draw, text=(bottom_text or "").strip(), img_w=w, y=bottom_y_start, font=font,
-        fill=text_color, stroke_fill=outline_color, stroke_width=stroke, align=align
-    )
+    _, _ = draw_block(draw, top_text, w, top_y, font, text_color, outline_color, stroke, align=align)
+
+    lines, heights = wrap_lines(draw, bottom_text, w, font, stroke)
+    total_bottom_h = sum(heights) + (len(heights)-1) * int(font.size*0.25)
+    bottom_y_start = int(h - total_bottom_h - h*0.03) - int(bottom_nudge)
+    draw_block(draw, bottom_text, w, bottom_y_start, font, text_color, outline_color, stroke, align=align)
+
     return img
 
-# -------- Theme (only props supported on older Gradio) --------
+# ---------------- Theme + Retro CSS ----------------
 THEME = gr.themes.Soft(
     primary_hue="indigo",
     secondary_hue="violet",
     neutral_hue="slate"
-).set(
-    button_border_width="0px",
-    button_large_padding="14px",
-    button_large_radius="14px",
-    body_background_fill_dark="linear-gradient(180deg, #0b0f17 0%, #0a0920 100%)",
-    block_shadow="none",
 )
 
 CUSTOM_CSS = """
+@import url('https://fonts.googleapis.com/css2?family=Press+Start+2P&display=swap');
+
 :root { --radius: 14px; }
 * { -webkit-tap-highlight-color: transparent; }
-.gradio-container { padding: 10px; max-width: 820px; margin: 0 auto; }
-h2, p { text-align:center }
+body { background: radial-gradient(1200px 600px at 50% -10%, #0d1220 10%, #05060b 70%); }
+.gradio-container { max-width: 900px; margin: 0 auto; padding: 12px; }
+h2, p { text-align: center; color: #cde3ff; text-shadow: 0 0 10px rgba(80,120,255,.25); }
+h2 { font-family: 'Press Start 2P', system-ui, sans-serif; letter-spacing: 1px; font-size: 18px; }
+.crt {
+  position: relative; border: 2px solid #2a3350; border-radius: 12px; overflow: hidden;
+  box-shadow: 0 0 0 1px #0f1427 inset, 0 0 40px rgba(60,80,255,.25);
+}
+.crt::before {
+  content: ""; position: absolute; inset: 0; pointer-events: none;
+  background: repeating-linear-gradient(180deg, rgba(255,255,255,0.05), rgba(255,255,255,0.05) 1px, transparent 1px, transparent 3px);
+  mix-blend-mode: overlay; opacity: .25;
+}
+label { color: #a9b7ff !important; }
+.gr-button { font-weight: 800; border-radius: 12px; }
 """
 
-# -------- App --------
+# ---------------- App ----------------
 with gr.Blocks(theme=THEME, css=CUSTOM_CSS) as demo:
-    gr.Markdown("<h2>📱 Mobile Meme Maker</h2><p>Upload → Type → Download. Built on iPhone, for iPhone.</p>")
+    gr.Markdown("<h2>🕹️ MEME LAB — RETRO EDITION</h2>"
+                "<p>One prompt → generate image → auto meme text. Style presets for instant vibes.</p>")
 
     with gr.Row():
-        with gr.Column():
-            in_img = gr.Image(type="pil", label="Upload photo", height=320)
+        with gr.Column(scale=1, elem_classes=["crt"]):
+            prompt = gr.Textbox(label="Your idea (one prompt)", value="cat typing on a laptop at midnight")
+            preset = gr.Dropdown(choices=list(PRESETS.keys()), value="Retro Comic", label="Style preset")
+            use_ai = gr.Checkbox(label="Use AI generator (public FLUX Space)", value=False)
 
-            top_text = gr.Textbox(label="Top text", value="WHEN THE WIFI JUST WORKS")
-            bottom_text = gr.Textbox(label="Bottom text", value="CHEF'S KISS")
+            with gr.Row():
+                width = gr.Slider(384, 1024, value=768, step=64, label="Width")
+                height = gr.Slider(384, 1024, value=768, step=64, label="Height")
 
-            # No inline=… (older Gradio)
-            align = gr.Radio(choices=["left", "center", "right"], value="center", label="Alignment")
+            gr.Markdown("### Meme Text")
+            use_prompt_for_text = gr.Checkbox(label="Auto from prompt", value=True)
+            top_text_manual = gr.Textbox(label="Top text (if not auto)", value="", interactive=True)
+            bottom_text_manual = gr.Textbox(label="Bottom text (if not auto)", value="", interactive=True)
 
-            font_size = gr.Slider(8, 22, value=12, step=1, label="Font size (% of width)")
-            stroke_width = gr.Slider(0, 14, value=4, step=1, label="Outline thickness")
+            align = gr.Radio(choices=["left", "center", "right"], value="center", label="Text alignment")
+            font_size = gr.Slider(8, 24, value=12, step=1, label="Font size (% of width)")
+            stroke_width = gr.Slider(0, 16, value=4, step=1, label="Outline thickness")
 
             with gr.Row():
-                text_color = gr.ColorPicker(value="#FFFFFF", label="Text")
-                outline_color = gr.ColorPicker(value="#000000", label="Outline")
+                text_color = gr.ColorPicker(value="#FFFFFF", label="Text color")
+                outline_color = gr.ColorPicker(value="#000000", label="Outline color")
 
             with gr.Row():
                 top_nudge = gr.Slider(-300, 300, value=0, step=1, label="Top nudge (px)")
                 bottom_nudge = gr.Slider(-300, 300, value=0, step=1, label="Bottom nudge (px)")
 
-        with gr.Column():
-            out = gr.Image(type="pil", label="Preview / Download", height=440, show_download_button=True)
-            generate = gr.Button("✨ Generate", size="lg")
+        with gr.Column(scale=1, elem_classes=["crt"]):
+            out = gr.Image(type="pil", label="Preview / Download", height=540, show_download_button=True)
+            generate = gr.Button("✨ Generate Image + Meme", variant="primary")
+
+    inputs = [prompt, preset, use_ai, width, height,
+              font_size, stroke_width, text_color, outline_color,
+              align, top_nudge, bottom_nudge, use_prompt_for_text, top_text_manual, bottom_text_manual]
 
-    inputs = [in_img, top_text, bottom_text, font_size, stroke_width, text_color, outline_color, align, top_nudge, bottom_nudge]
-    generate.click(fn=make_meme, inputs=inputs, outputs=out)
+    generate.click(fn=generate_and_meme, inputs=inputs, outputs=out)
 
-    # live preview as controls change
-    for comp in inputs:
-        comp.change(fn=make_meme, inputs=inputs, outputs=out, show_progress=False)
+    # Quick-refresh when only style/text settings change
+    for comp in [preset, use_prompt_for_text, top_text_manual, bottom_text_manual,
+                 font_size, stroke_width, text_color, outline_color, align, top_nudge, bottom_nudge]:
+        comp.change(fn=generate_and_meme, inputs=inputs, outputs=out, show_progress=False)
 
 if __name__ == "__main__":
     demo.launch()