Update app.py

app.py

@@ -10,7 +10,7 @@ CANDIDATE_FONTS = [
 def get_font(size: int):
     for p in CANDIDATE_FONTS:
         if os.path.exists(p):
-            return ImageFont.truetype(p, size=size)
+            return ImageFont.truetype(p, size=int(size))
     return ImageFont.load_default()
 
 # ---------------- Style Presets ----------------
@@ -25,8 +25,13 @@ PRESETS = {
     "Synthwave Grid": "purple/indigo gradient sky, mountain silhouette, glowing sun, grid floor",
 }
 
-# ---------------- Utility: gradient fallback (fast, offline) ----------------
+# -------- helpers --------
+def i(v):  # safe int
+    try: return int(round(float(v)))
+    except: return int(v)
+
 def gradient_from_prompt(prompt: str, w=768, h=768):
+    w, h = i(w), i(h)
     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))
@@ -43,7 +48,6 @@ def gradient_from_prompt(prompt: str, w=768, h=768):
             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))
@@ -84,17 +88,16 @@ def smart_split_text(prompt: str):
         return " ".join(words[:mid]).upper(), " ".join(words[mid:]).upper()
     return p.upper(), ""
 
-# ---------------- Optional: call public FLUX Space ----------------
+# -------- Optional public Space call --------
 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")
+    width, height = i(width), i(height)
     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]
@@ -102,17 +105,21 @@ def try_generate_with_flux(prompt: str, width: int, height: int):
         except Exception:
             raise e
 
-# ---------------- Main pipeline ----------------
+# -------- 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, use_prompt_for_text, top_text_manual, bottom_text_manual
 ):
-    base = (prompt or "").strip()
+    width, height = i(width), i(height)
+    top_nudge, bottom_nudge = i(top_nudge), i(bottom_nudge)
+    stroke_width = i(stroke_width)
+
+    base = (prompt or "").trim() if hasattr(str, "trim") else (prompt or "").strip()
     style_suffix = PRESETS.get(preset_name or "None", "")
     gen_prompt = (base + " " + style_suffix).strip()
 
-    # 1) Image
+    # 1) image
     if use_ai:
         try:
             img = try_generate_with_flux(gen_prompt, width, height)
@@ -125,57 +132,48 @@ def generate_and_meme(
     draw = ImageDraw.Draw(img)
     w, h = img.size
 
-    # 2) Meme text
+    # 2) 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))
+    # 3) draw
+    base_size = max(12, int((w * float(font_size)) / 100))
     font = get_font(base_size)
     stroke = int(max(0, stroke_width))
 
-    top_y = int(h * 0.03) + int(top_nudge)
+    top_y = int(h * 0.03) + top_nudge
     _, _ = 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)
+    bottom_y_start = int(h - total_bottom_h - h*0.03) - bottom_nudge
     draw_block(draw, bottom_text, w, bottom_y_start, font, text_color, outline_color, stroke, align=align)
 
     return img
 
-# ---------------- Theme + Retro CSS ----------------
-THEME = gr.themes.Soft(
-    primary_hue="indigo",
-    secondary_hue="violet",
-    neutral_hue="slate"
-)
+# -------- Theme + Retro CSS --------
+THEME = gr.themes.Soft(primary_hue="indigo", secondary_hue="violet", neutral_hue="slate")
 
 CUSTOM_CSS = """
 @import url('https://fonts.googleapis.com/css2?family=Press+Start+2P&display=swap');
-
 :root { --radius: 14px; }
 * { -webkit-tap-highlight-color: transparent; }
 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;
+.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;
-}
+  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>🕹️ MEME LAB — RETRO EDITION</h2>"
                 "<p>One prompt → generate image → auto meme text. Style presets for instant vibes.</p>")
@@ -185,7 +183,6 @@ with gr.Blocks(theme=THEME, css=CUSTOM_CSS) as demo:
             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)
-
             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")
@@ -198,11 +195,9 @@ with gr.Blocks(theme=THEME, css=CUSTOM_CSS) as demo:
             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 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)")
@@ -217,7 +212,6 @@ with gr.Blocks(theme=THEME, css=CUSTOM_CSS) as demo:
 
     generate.click(fn=generate_and_meme, inputs=inputs, outputs=out)
 
-    # 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)