Update app.py

app.py

@@ -19,19 +19,17 @@ import traceback
 
 import gradio as gr
 from openai import OpenAI
-from huggingface_hub import InferenceClient
 from PIL import Image
 
 # ---------------------------------------------------------------------------
 # Clients
 # ---------------------------------------------------------------------------
 
-GROQ_API_KEY = os.environ.get("GROQ_API_KEY", "")
-HF_TOKEN     = os.environ.get("HF_TOKEN", "")
+GROQ_API_KEY   = os.environ.get("GROQ_API_KEY", "")
+GOOGLE_API_KEY = os.environ.get("GOOGLE_API_KEY", "")
 
-CODE_MODEL  = "llama-3.1-8b-instant"   # Groq — game code
-PROMPT_MODEL = "llama-3.3-70b-versatile" # Groq — better model for creative prompts
-IMAGE_MODEL = "black-forest-labs/FLUX.1-schnell"  # hf-inference
+CODE_MODEL   = "llama-3.1-8b-instant"      # Groq — game code
+PROMPT_MODEL = "llama-3.3-70b-versatile"   # Groq — creative prompts
 
 
 def get_groq_client():
@@ -46,10 +44,14 @@ def get_groq_client():
     )
 
 
-def get_image_client():
-    if not HF_TOKEN:
+def get_gemini_image_client():
+    """Gemini image generation — 500 free images/day, no credit card needed."""
+    if not GOOGLE_API_KEY:
         return None
-    return InferenceClient(provider="hf-inference", api_key=HF_TOKEN)
+    return OpenAI(
+        base_url="https://generativelanguage.googleapis.com/v1beta/openai/",
+        api_key=GOOGLE_API_KEY,
+    )
 
 
 # ---------------------------------------------------------------------------
@@ -251,29 +253,36 @@ def _colored_placeholder(name: str) -> str:
 
 
 def generate_sprites(image_prompts: dict) -> tuple:
-    """Generate actual images from enhanced prompts via FLUX.1-schnell.
+    """Generate images via Gemini imagen API (500 free/day, no credit card).
     Returns (sprite_map, errors_list)"""
-    image_client = get_image_client()
+    client = get_gemini_image_client()
     sprite_map = {}
     errors = []
 
-    if not image_client:
-        errors.append("HF_TOKEN not set - using placeholder sprites. Add HF_TOKEN as a Space secret.")
+    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():
         try:
-            is_bg = "background" in sprite_name
-            pil_img = image_client.text_to_image(prompt, model=IMAGE_MODEL)
+            response = client.images.generate(
+                model="imagen-3.0-generate-002",
+                prompt=prompt,
+                n=1,
+                response_format="b64_json",
+            )
+            b64 = response.data[0].b64_json
+            pil_img = Image.open(io.BytesIO(base64.b64decode(b64)))
+            is_bg   = "background" in sprite_name
             size    = None if is_bg else (64, 64)
             sprite_map[sprite_name] = _pil_to_data_uri(pil_img, size=size)
-            print(f"[FLUX] OK: {sprite_name}")
+            print(f"[Gemini Imagen] OK: {sprite_name}")
         except Exception as exc:
             error_msg = str(exc)
             errors.append(f"{sprite_name}: {error_msg}")
-            print(f"[FLUX] FAILED {sprite_name}: {error_msg}")
+            print(f"[Gemini Imagen] FAILED {sprite_name}: {error_msg}")
             sprite_map[sprite_name] = _colored_placeholder(sprite_name.replace(".png", ""))
 
     return sprite_map, errors
@@ -355,12 +364,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} FLUX, {n_fallback} fallback. "
+                f"Code generated. Images: {n_real} by Gemini Imagen, {n_fallback} fallback. "
                 f"FLUX errors: {error_detail}"
             )
         else:
             status = (
-                f"Done! {n_real} sprite(s) generated by FLUX. "
+                f"Done! {n_real} sprite(s) generated by Gemini Imagen. "
                 "Click Launch Game to play."
             )
 
@@ -538,7 +547,7 @@ def build_ui():
         gr.Markdown(
             "---\n"
             "**Pipeline:** Theme → [Groq Llama] game code + [Groq 70B as Z-Image-Engineer] "
-            "cinematic prompts → [FLUX.1-schnell/nscale] sprites → embedded in game. "
+            "cinematic prompts → [Gemini Imagen-3] sprites → embedded in game. "
             "Edit the HTML and click **Launch Game** to hot-reload."
         )