Update app.py

app.py

@@ -6,15 +6,16 @@ import gradio as gr
 # ==============================
 # Secrets / Config
 # ==============================
-HF_TOKEN = os.getenv("HF_TOKEN")  # Space secret
-DEFAULT_MODEL_ID = "stabilityai/sd-turbo"  # fast + public
-
-# Use ONLY model IDs that are real + public
-MODEL_CHOICES = {
-    "sd-1.5 (runwayml)": "runwayml/stable-diffusion-v1-5",
-    "sd-2.1 (stabilityai)": "stabilityai/stable-diffusion-2-1",
-}
-
+HF_TOKEN = os.getenv("HF_TOKEN")  # optional, via Repo Secrets
+# Try these Inference API models in order; skip silently on 404/403/5xx
+INFERENCE_CANDIDATES = [
+    "stabilityai/stable-diffusion-2-1",
+    "runwayml/stable-diffusion-v1-5",
+    "stabilityai/sd-turbo",  # may 404 for some accounts
+]
+# Public Space as last resort (no token). If they change UI, we still fail gracefully.
+PUBLIC_SPACE_ID = "black-forest-labs/FLUX.1-schnell"
+PUBLIC_SPACE_APIS = ["/predict", "/run"]
 
 # ==============================
 # Fonts
@@ -80,9 +81,21 @@ def draw_block(draw, text, img_w, y, font, fill, stroke_fill, stroke_width, alig
         draw.text((x, curr_y), line, font=font, fill=fill,
                   stroke_width=stroke_width, stroke_fill=stroke_fill)
         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
+    return curr_y, sum(heights) + (len(heights)-1) * int(font.size*0.25)
 
+# ==============================
+# Styles / text split
+# ==============================
+PRESETS = {
+    "None": "",
+    "Retro Comic": "bold comic outline, grain, high contrast, 35mm scan",
+    "Vaporwave": "vaporwave, neon pink and cyan, miami sunset, synth grid",
+    "Game Boy": "pixel art, 4-color green palette, dithering",
+    "Newspaper Halftone": "b&w halftone dots, newsprint texture",
+    "Cyberpunk Neon": "neon city at night, purple blue rim light, rain",
+    "90s Web": "bevel buttons, gradients, clipart stars, lens flare",
+    "Synthwave Grid": "purple/indigo sky, glowing sun, mountains, grid floor",
+}
 def smart_split_text(prompt: str):
     p = (prompt or "").strip()
     if not p:
@@ -98,9 +111,9 @@ def smart_split_text(prompt: str):
     return p.upper(), ""
 
 # ==============================
-# HF Inference API
+# Generators (multi-fallback)
 # ==============================
-def generate_via_hf_inference(model_id: str, prompt: str, width: int, height: int) -> Image.Image:
+def call_inference_api(model_id: str, prompt: str, width: int, height: int) -> Image.Image:
     if not HF_TOKEN:
         raise RuntimeError("HF token not set")
     url = f"https://api-inference.huggingface.co/models/{model_id}"
@@ -108,19 +121,53 @@ def generate_via_hf_inference(model_id: str, prompt: str, width: int, height: in
     payload = {
         "inputs": prompt,
         "options": {"wait_for_model": True},
-        # These params are accepted by sd-turbo and ignored by some others (fine)
         "parameters": {"width": int(width), "height": int(height)}
     }
     r = requests.post(url, headers=headers, json=payload, timeout=180)
     if r.status_code != 200:
-        raise RuntimeError(f"HF API {r.status_code}: {r.text[:160]}")
+        raise RuntimeError(f"{r.status_code}:{r.text[:160]}")
     return Image.open(BytesIO(r.content)).convert("RGB")
 
+def call_public_space(prompt: str, width: int, height: int) -> Image.Image:
+    # no token required
+    from gradio_client import Client
+    client = Client(PUBLIC_SPACE_ID)
+    # try multiple API names to survive UI tweaks
+    last_err = None
+    for api in PUBLIC_SPACE_APIS:
+        try:
+            res = client.predict(prompt, width, height, api_name=api)
+            if isinstance(res, list): res = res[0]
+            return Image.open(res).convert("RGB")
+        except Exception as e:
+            last_err = e
+            continue
+    raise RuntimeError(f"public-space-fail:{last_err}")
+
+def generate_image_auto(prompt: str, width: int, height: int):
+    # 1) try HF Inference API candidates
+    tried = []
+    if HF_TOKEN:
+        for mid in INFERENCE_CANDIDATES:
+            try:
+                img = call_inference_api(mid, prompt, width, height)
+                return img, f"✅ Inference API: **{mid}** (token present)"
+            except Exception as e:
+                tried.append(f"{mid}→{str(e)[:80]}")
+    # 2) try public Space (no token)
+    try:
+        img = call_public_space(prompt, width, height)
+        return img, f"✅ Public Space: **{PUBLIC_SPACE_ID}**"
+    except Exception as e:
+        tried.append(f"{PUBLIC_SPACE_ID}→{str(e)[:80]}")
+    # 3) fallback gradient
+    return gradient_from_prompt(prompt, w=width, h=height), f"⚠️ Fallback gradient | tried: {', '.join(tried)}"
+
 # ==============================
 # Core pipeline (returns image + status)
 # ==============================
 def generate_and_meme(
-    prompt, preset_name, use_ai, model_label, width, height,
+    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
 ):
@@ -132,27 +179,18 @@ def generate_and_meme(
     style_suffix = PRESETS.get(preset_name or "None", "")
     gen_prompt = (base + " " + style_suffix).strip()
 
-    # 1) Image
     if use_ai:
-        model_id = MODEL_CHOICES.get(model_label, DEFAULT_MODEL_ID)
-        try:
-            img = generate_via_hf_inference(model_id, gen_prompt, width, height)
-            status = f"✅ AI image via **{model_id}** | token={'present' if bool(HF_TOKEN) else 'missing'}"
-        except Exception as e:
-            img = gradient_from_prompt(gen_prompt, w=width, h=height)
-            status = f"⚠️ Fallback to gradient. Model=**{model_id}** | token={'present' if bool(HF_TOKEN) else 'missing'} | Reason: {str(e)[:180]}"
+        img, status = generate_image_auto(gen_prompt, width, height)
     else:
-        img = gradient_from_prompt(gen_prompt, w=width, h=height)
-        status = "ℹ️ AI generator is OFF"
+        img, status = gradient_from_prompt(gen_prompt, w=width, h=height), "ℹ️ AI generator is OFF"
 
-    # 2) Text
+    # text
     if use_prompt_for_text:
         top_text, bottom_text = smart_split_text(base)
     else:
         top_text = (top_text_manual or "").upper()
         bottom_text = (bottom_text_manual or "").upper()
 
-    # 3) Draw
     img = img.convert("RGB")
     draw = ImageDraw.Draw(img)
     w_img, h_img = img.size
@@ -171,25 +209,10 @@ def generate_and_meme(
 
     return img, status
 
-# ==============================
-# Style presets
-# ==============================
-PRESETS = {
-    "None": "",
-    "Retro Comic": "bold comic outline, grain, high contrast, 35mm scan",
-    "Vaporwave": "vaporwave, neon pink and cyan, miami sunset, synth grid",
-    "Game Boy": "pixel art, 4-color green palette, dithering",
-    "Newspaper Halftone": "b&w halftone dots, newsprint texture",
-    "Cyberpunk Neon": "neon city at night, purple blue rim light, rain",
-    "90s Web": "bevel buttons, gradients, clipart stars, lens flare",
-    "Synthwave Grid": "purple/indigo sky, glowing sun, mountains, grid floor",
-}
-
 # ==============================
 # Retro theme + 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; }
@@ -218,13 +241,7 @@ with gr.Blocks(theme=THEME, css=CUSTOM_CSS) as demo:
         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 (HF Inference API, needs secret)", value=False)
-
-            model_label = gr.Dropdown(
-                choices=list(MODEL_CHOICES.keys()),
-                value="sd-turbo (fast)",
-                label="Model (only used when AI is ON)"
-            )
+            use_ai = gr.Checkbox(label="Use AI image (auto-fallbacks, no key required)", value=True)
 
             with gr.Row():
                 width = gr.Slider(384, 1024, value=768, step=64, label="Width")
@@ -249,16 +266,16 @@ with gr.Blocks(theme=THEME, css=CUSTOM_CSS) as demo:
 
         with gr.Column(scale=1, elem_classes=["crt"]):
             out = gr.Image(type="pil", label="Preview / Download", height=540, show_download_button=True)
-            status = gr.Markdown("…")  # shows why it fell back / which model was used
+            status = gr.Markdown("…")
             generate = gr.Button("✨ Generate Image + Meme", variant="primary")
 
-    inputs = [prompt, preset, use_ai, model_label, width, height,
+    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]
 
     generate.click(fn=generate_and_meme, inputs=inputs, outputs=[out, status])
 
-    for comp in [preset, model_label, use_prompt_for_text, top_text_manual, bottom_text_manual,
+    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, status], show_progress=False)