Update app.py

app.py

@@ -25,12 +25,14 @@ from PIL import Image
 # Clients
 # ---------------------------------------------------------------------------
 
-GROQ_API_KEY   = os.environ.get("GROQ_API_KEY", "")
-GOOGLE_API_KEY = os.environ.get("GOOGLE_API_KEY", "")
+GROQ_API_KEY = os.environ.get("GROQ_API_KEY", "")
 
 CODE_MODEL   = "llama-3.1-8b-instant"      # Groq — game code
 PROMPT_MODEL = "llama-3.3-70b-versatile"   # Groq — creative prompts
 
+# Pollinations.AI — completely free, no API key, no signup needed
+POLLINATIONS_URL = "https://image.pollinations.ai/prompt/{prompt}?width={w}&height={h}&model=flux&nologo=true&seed={seed}"
+
 
 def get_groq_client():
     if not GROQ_API_KEY:
@@ -44,16 +46,6 @@ def get_groq_client():
     )
 
 
-def get_gemini_image_client():
-    """Gemini image generation — 500 free images/day, no credit card needed."""
-    if not GOOGLE_API_KEY:
-        return None
-    return OpenAI(
-        base_url="https://generativelanguage.googleapis.com/v1beta/openai/",
-        api_key=GOOGLE_API_KEY,
-    )
-
-
 # ---------------------------------------------------------------------------
 # Z-Image-Engineer system prompt (from BennyDaBall/Qwen3-4b-Z-Image-Engineer-V4)
 # ---------------------------------------------------------------------------
@@ -253,36 +245,33 @@ def _colored_placeholder(name: str) -> str:
 
 
 def generate_sprites(image_prompts: dict) -> tuple:
-    """Generate images via Gemini imagen API (500 free/day, no credit card).
+    """Generate images via Pollinations.AI (free, no API key, no signup).
     Returns (sprite_map, errors_list)"""
-    client = get_gemini_image_client()
+    import requests
+    from urllib.parse import quote
+
     sprite_map = {}
     errors = []
 
-    if not client:
-        errors.append("GOOGLE_API_KEY not set - using placeholder sprites. Add GOOGLE_API_KEY as a Space secret.")
-        for sprite_name in image_prompts:
-            sprite_map[sprite_name] = _colored_placeholder(sprite_name.replace(".png", ""))
-        return sprite_map, errors
-
-    for sprite_name, prompt in image_prompts.items():
+    for i, (sprite_name, prompt) in enumerate(image_prompts.items()):
         try:
-            response = client.images.generate(
-                model="imagen-3.0-generate-002",
-                prompt=prompt,
-                n=1,
-                response_format="b64_json",
+            is_bg  = "background" in sprite_name
+            w, h   = (800, 450) 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
             )
-            b64 = response.data[0].b64_json
-            pil_img = Image.open(io.BytesIO(base64.b64decode(b64)))
-            is_bg   = "background" in sprite_name
+            response = requests.get(url, timeout=60)
+            response.raise_for_status()
+            pil_img = Image.open(io.BytesIO(response.content))
             size    = None if is_bg else (64, 64)
             sprite_map[sprite_name] = _pil_to_data_uri(pil_img, size=size)
-            print(f"[Gemini Imagen] OK: {sprite_name}")
+            print(f"[Pollinations] OK: {sprite_name}")
         except Exception as exc:
             error_msg = str(exc)
             errors.append(f"{sprite_name}: {error_msg}")
-            print(f"[Gemini Imagen] FAILED {sprite_name}: {error_msg}")
+            print(f"[Pollinations] FAILED {sprite_name}: {error_msg}")
             sprite_map[sprite_name] = _colored_placeholder(sprite_name.replace(".png", ""))
 
     return sprite_map, errors
@@ -364,12 +353,12 @@ def generate_game_code(game_type: str, theme: str, temperature: float, max_new_t
         if sprite_errors:
             error_detail = " | ".join(sprite_errors)
             status = (
-                f"Code generated. Images: {n_real} by Gemini Imagen, {n_fallback} fallback. "
+                f"Code generated. Images: {n_real} by Pollinations/FLUX, {n_fallback} fallback. "
                 f"FLUX errors: {error_detail}"
             )
         else:
             status = (
-                f"Done! {n_real} sprite(s) generated by Gemini Imagen. "
+                f"Done! {n_real} sprite(s) generated by Pollinations/FLUX. "
                 "Click Launch Game to play."
             )
 
@@ -547,7 +536,7 @@ def build_ui():
         gr.Markdown(
             "---\n"
             "**Pipeline:** Theme → [Groq Llama] game code + [Groq 70B as Z-Image-Engineer] "
-            "cinematic prompts → [Gemini Imagen-3] sprites → embedded in game. "
+            "cinematic prompts → [Pollinations.AI/FLUX] sprites → embedded in game. "
             "Edit the HTML and click **Launch Game** to hot-reload."
         )