Upload app.py with huggingface_hub

app.py

@@ -0,0 +1,203 @@
+import os
+import io
+import random
+import gradio as gr
+from huggingface_hub import InferenceClient
+from PIL import Image, ImageDraw, ImageFont
+
+# --- Configuration -----------------------------------------------------------
+HF_TOKEN = os.environ.get("HF_TOKEN", None)
+MODEL_ID = "black-forest-labs/FLUX.1-schnell"
+
+client = InferenceClient(model=MODEL_ID, token=HF_TOKEN)
+
+BG_COLOR = "#0A0F1A"
+ACCENT = "#1FE0E6"
+
+# Preset templates: name -> (base_image_prompt, default_top, default_bottom)
+TEMPLATES = {
+    "This is Fine AI": (
+        "A cartoon character sitting in a burning AI server room surrounded by flames, smiling and saying 'this is fine', meme style, comic illustration",
+        "MY GPU IS ON FIRE",
+        "THIS IS FINE",
+    ),
+    "Distracted Boyfriend AI": (
+        "A meme photo of a boyfriend walking with his girlfriend but looking back at a passing robot labeled 'NEW AI MODEL', girlfriend labeled 'MY CURRENT MODEL', meme style",
+        "MY CURRENT MODEL",
+        "NEW HYPED AI MODEL",
+    ),
+    "Drake Hotline AI": (
+        "Drake hotline bling meme format, Drake disapproving panel on top and Drake approving panel on bottom, left panel says 'Writing code myself', right panel says 'Asking ChatGPT to do it', meme template",
+        "WRITING CODE MYSELF",
+        "ASKING AI TO DO IT",
+    ),
+}
+
+# Random AI topics for the Random AI Topic button
+AI_TOPICS = [
+    "AGI is coming next year",
+    "My model hallucinated a whole paper",
+    "Prompt engineering is real engineering",
+    "I fine-tuned on my DMs",
+    "The model refused my prompt",
+    "I ran out of GPU credits",
+    "My agent looped forever",
+    "Context window too small for my TODO list",
+    "Open source beat closed source",
+    "Inference costs more than my rent",
+    "My dataset was just vibes",
+    "RLHF made my bot too polite",
+    "I forgot to clip the gradients",
+    "The model learned to lie",
+    "Quantization ruined my quality",
+    "My LoRA saved my marriage",
+    "Distillation taught the student to cheat",
+    "The tokenizer hates Arabic",
+    "I spent $0 on training and won the benchmark",
+    "Agents are just while loops with vibes",
+]
+
+# Try to find a usable font
+FONT_CANDIDATES = [
+    "/usr/share/fonts/truetype/dejavu/DejaVuSans-Bold.ttf",
+    "/usr/share/fonts/truetype/liberation/LiberationSans-Bold.ttf",
+    "C:\\Windows\\Fonts\\arialbd.ttf",
+    "arial.ttf",
+]
+
+
+def load_font(size):
+    for path in FONT_CANDIDATES:
+        if os.path.exists(path):
+            return ImageFont.truetype(path, size)
+    # fallback default
+    return ImageFont.load_default()
+
+
+def random_topic():
+    return random.choice(AI_TOPICS)
+
+
+# --- Core function -----------------------------------------------------------
+def generate_meme(top_caption, bottom_caption, template_name):
+    """Generate a meme image: generate base via FLUX, then overlay text."""
+    if template_name and template_name in TEMPLATES:
+        base_prompt, def_top, def_bot = TEMPLATES[template_name]
+        top = top_caption.strip() if top_caption.strip() else def_top
+        bot = bottom_caption.strip() if bottom_caption.strip() else def_bot
+    else:
+        base_prompt = "A humorous meme style illustration related to artificial intelligence, comic style"
+        top = top_caption.strip() if top_caption.strip() else "TOP TEXT"
+        bot = bottom_caption.strip() if bottom_caption.strip() else "BOTTOM TEXT"
+
+    w, h = 1024, 1024
+
+    try:
+        image = client.text_to_image(base_prompt, width=w, height=h)
+        if not isinstance(image, Image.Image):
+            image = Image.open(io.BytesIO(image)) if hasattr(image, "read") else Image.open(image)
+    except Exception as e:
+        image = Image.new("RGB", (w, h), BG_COLOR)
+        d = ImageDraw.Draw(image)
+        d.text((20, h // 2), f"Error: {e}", fill=ACCENT)
+
+    # Overlay meme text (classic Impact style)
+    image = overlay_meme_text(image, top.upper(), bot.upper())
+    return image, f"✅ Meme generated! Top: '{top}' | Bottom: '{bot}'"
+
+
+def overlay_meme_text(img, top_text, bottom_text):
+    """Draw classic Impact-font style meme text with black outline."""
+    img = img.convert("RGB")
+    draw = ImageDraw.Draw(img)
+    W, H = img.size
+
+    # Font sizes proportional to image
+    font_size = max(24, int(W / 12))
+    font = load_font(font_size)
+
+    def draw_text_with_outline(text, y, anchor="ma"):
+        # Outline (black)
+        for dx in (-2, -1, 1, 2):
+            for dy in (-2, -1, 1, 2):
+                draw.text((W // 2 + dx, y + dy), text, fill="black", font=font, anchor=anchor)
+        # Fill (white)
+        draw.text((W // 2, y), text, fill="white", font=font, anchor=anchor)
+
+    # Top text near top
+    draw_text_with_outline(top_text, int(H * 0.08), anchor="ma")
+    # Bottom text near bottom
+    draw_text_with_outline(bottom_text, int(H * 0.92), anchor="ma")
+
+    return img
+
+
+def load_template(template_name):
+    """Load default captions from a template."""
+    if template_name and template_name in TEMPLATES:
+        _, def_top, def_bot = TEMPLATES[template_name]
+        return def_top, def_bot
+    return "", ""
+
+
+# --- UI -----------------------------------------------------------------------
+CUSTOM_CSS = f"""
+.gradio-container {{background-color: {BG_COLOR} !important;}}
+h1, h2, h3 {{color: {ACCENT} !important;}}
+"""
+
+with gr.Blocks(
+    title="Dispatch AI Meme Lab",
+    theme=gr.themes.Base(
+        primary_hue="cyan",
+        neutral_hue="slate",
+        font=("Inter", "system-ui", "sans-serif"),
+    ),
+    css=CUSTOM_CSS,
+) as demo:
+    gr.Markdown(
+        f"""
+        <div style="text-align:center;">
+        <h1>Dispatch AI Meme Lab</h1>
+        <p style="color:{ACCENT};">Generate AI/tech memes with FLUX.1-schnell + PIL text overlay · Dispatch AI (FZE)</p>
+        </div>
+        """
+    )
+
+    with gr.Row():
+        with gr.Column(scale=1):
+            template_select = gr.Dropdown(
+                list(TEMPLATES.keys()),
+                label="Meme Template",
+                value=None,
+                info="Pick a template to auto-fill captions and base image",
+            )
+            load_template_btn = gr.Button("Load Template Captions", variant="secondary")
+            top_caption = gr.Textbox(label="Top Caption", placeholder="TOP TEXT", lines=1)
+            bottom_caption = gr.Textbox(label="Bottom Caption", placeholder="BOTTOM TEXT", lines=1)
+            random_topic_btn = gr.Button("🎲 Random AI Topic", variant="secondary")
+            topic_box = gr.Textbox(label="Random AI Topic", interactive=True, lines=1)
+            generate_btn = gr.Button("🤣 Generate Meme", variant="primary")
+        with gr.Column(scale=2):
+            output_image = gr.Image(label="Generated Meme", type="pil", show_download_button=True)
+            status_box = gr.Textbox(label="Status", interactive=False)
+
+    # Events
+    load_template_btn.click(load_template, inputs=template_select, outputs=[top_caption, bottom_caption])
+    random_topic_btn.click(random_topic, outputs=topic_box)
+    generate_btn.click(
+        generate_meme,
+        inputs=[top_caption, bottom_caption, template_select],
+        outputs=[output_image, status_box],
+    )
+
+    gr.Markdown(
+        """
+        <div style="text-align:center; opacity:0.6; padding-top:20px;">
+        <small>Dispatch AI (FZE) · UAE · Model: FLUX.1-schnell · Text overlay via Pillow</small>
+        </div>
+        """
+    )
+
+if __name__ == "__main__":
+    demo.launch()