Update app.py

app.py

@@ -19,6 +19,8 @@ import traceback
 import time
 import gradio as gr
 from openai import OpenAI
+import requests
+from urllib.parse import quote
 from huggingface_hub import InferenceClient
 from PIL import Image
 
@@ -32,6 +34,7 @@ HF_TOKEN     = os.environ.get("HF_TOKEN", "")
 CODE_MODEL   = "llama-3.1-8b-instant"          # Groq — game code
 PROMPT_MODEL = "llama-3.3-70b-versatile"       # Groq — image prompt enhancement
 IMAGE_MODEL  = "black-forest-labs/FLUX.1-dev"  # fal-ai — image generation
+POLLINATIONS_URL = "https://image.pollinations.ai/prompt/{prompt}?width={w}&height={h}&model=flux&nologo=true&seed={seed}"
 
 # Canvas dimensions — iframe is fixed to these so game fits perfectly
 CANVAS_W = 800
@@ -258,33 +261,71 @@ def _colored_placeholder(name: str) -> str:
     return "data:image/svg+xml;base64," + base64.b64encode(svg.encode()).decode()
 
 
+def _generate_via_hf(prompt: str) -> Image.Image:
+    """Try FLUX.1-dev via HF auto-provider selection.
+    HF automatically picks the fastest available provider — if one is
+    quota-depleted it routes to the next available one automatically."""
+    client = InferenceClient(api_key=HF_TOKEN)
+    return client.text_to_image(prompt, model=IMAGE_MODEL)
+
+
+def _generate_via_pollinations(prompt: str, sprite_name: str, is_bg: bool) -> Image.Image:
+    """Fallback: Pollinations.AI, free, no key needed."""
+    w, h = (CANVAS_W, CANVAS_H) if is_bg else (64, 64)
+    seed = abs(hash(sprite_name)) % 99999
+    url  = POLLINATIONS_URL.format(prompt=quote(prompt), w=w, h=h, seed=seed)
+    for attempt in range(3):
+        try:
+            resp = requests.get(url, timeout=120)
+            resp.raise_for_status()
+            return Image.open(io.BytesIO(resp.content))
+        except Exception as exc:
+            if attempt < 2:
+                time.sleep(20)
+            else:
+                raise exc
+
+
 def generate_sprites(image_prompts: dict) -> tuple:
     sprite_map = {}
     errors     = []
 
-    if not HF_TOKEN:
-        errors.append("HF_TOKEN not set — using placeholder sprites. Add HF_TOKEN as a Space secret.")
-        for sprite_name in image_prompts:
-            sprite_map[sprite_name] = _colored_placeholder(sprite_name.replace(".png", ""))
-        return sprite_map, errors
-
-    client = InferenceClient(provider="fal-ai", api_key=HF_TOKEN)
-
-    for sprite_name, prompt in image_prompts.items():
+    for i, (sprite_name, prompt) in enumerate(image_prompts.items()):
         try:
-            is_bg   = "background" in sprite_name
-            pil_img = client.text_to_image(prompt, model=IMAGE_MODEL)
-            # Remove background for character/object sprites (not background scene)
+            is_bg = "background" in sprite_name
+
+            # Try FLUX.1-dev via fal-ai if HF_TOKEN is set
+            if HF_TOKEN:
+                try:
+                    pil_img = _generate_via_hf(prompt)
+                    provider = "FLUX.1-dev/auto"
+                except Exception as fal_exc:
+                    fal_err = str(fal_exc)
+                    print(f"[fal-ai] Failed {sprite_name}: {fal_err} — falling back to Pollinations")
+                    errors.append(f"{sprite_name} (fal-ai failed, used Pollinations): {fal_err[:60]}")
+                    if i > 0:
+                        time.sleep(20)
+                    pil_img  = _generate_via_pollinations(prompt, sprite_name, is_bg)
+                    provider = "Pollinations"
+            else:
+                # No HF_TOKEN — go straight to Pollinations
+                if i > 0:
+                    time.sleep(20)
+                pil_img  = _generate_via_pollinations(prompt, sprite_name, is_bg)
+                provider = "Pollinations"
+
             if not is_bg:
                 pil_img = _remove_background(pil_img)
             size = None if is_bg else (64, 64)
             sprite_map[sprite_name] = _pil_to_data_uri(pil_img, size=size)
-            print(f"[FLUX.1-dev] OK: {sprite_name}")
+            print(f"[{provider}] OK: {sprite_name}")
+
         except Exception as exc:
             error_msg = str(exc)
             errors.append(f"{sprite_name}: {error_msg[:80]}")
-            print(f"[FLUX.1-dev] FAILED {sprite_name}: {error_msg}")
+            print(f"[Image] FAILED {sprite_name}: {error_msg}")
             sprite_map[sprite_name] = _colored_placeholder(sprite_name.replace(".png", ""))
+
     return sprite_map, errors
 
 # ---------------------------------------------------------------------------