Update app.py

app.py

@@ -250,28 +250,33 @@ def _colored_placeholder(name: str) -> str:
     return "data:image/svg+xml;base64," + base64.b64encode(svg.encode()).decode()
 
 
-def generate_sprites(image_prompts: dict) -> dict:
-    """Generate actual images from enhanced prompts via FLUX.1-schnell."""
+def generate_sprites(image_prompts: dict) -> tuple:
+    """Generate actual images from enhanced prompts via FLUX.1-schnell.
+    Returns (sprite_map, errors_list)"""
     image_client = get_image_client()
     sprite_map = {}
+    errors = []
 
-    for sprite_name, prompt in image_prompts.items():
-        if not image_client:
+    if not image_client:
+        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", ""))
-            continue
+        return sprite_map, errors
+
+    for sprite_name, prompt in image_prompts.items():
         try:
-            # Background gets full canvas size, sprites get 64x64
             is_bg = "background" in sprite_name
             pil_img = image_client.text_to_image(prompt, model=IMAGE_MODEL)
             size    = None if is_bg else (64, 64)
             sprite_map[sprite_name] = _pil_to_data_uri(pil_img, size=size)
-            print(f"[FLUX] Generated {sprite_name} OK")
+            print(f"[FLUX] 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}")
-            sprite_map[sprite_name] = ("ERROR:" + error_msg, _colored_placeholder(sprite_name.replace(".png", "")))[1]
+            sprite_map[sprite_name] = _colored_placeholder(sprite_name.replace(".png", ""))
 
-    return sprite_map
+    return sprite_map, errors
 
 # ---------------------------------------------------------------------------
 # Step 3: Inject sprites into HTML
@@ -339,17 +344,25 @@ def generate_game_code(game_type: str, theme: str, temperature: float, max_new_t
         image_prompts = generate_image_prompts(theme.strip(), game_type)
 
         # -- Sprite generation (FLUX.1-schnell) --
-        sprite_map = generate_sprites(image_prompts)
+        sprite_map, sprite_errors = generate_sprites(image_prompts)
 
         # -- Inject sprites --
         final_code = _inject_sprites(code, sprite_map)
 
         n_real     = sum(1 for v in sprite_map.values() if "image/png" in v)
         n_fallback = len(sprite_map) - n_real
-        status = (
-            f"Done! {n_real} sprite(s) generated by FLUX, "
-            f"{n_fallback} fallback. Click Launch Game to play."
-        )
+
+        if sprite_errors:
+            error_detail = " | ".join(sprite_errors)
+            status = (
+                f"Code generated. Images: {n_real} FLUX, {n_fallback} fallback. "
+                f"FLUX errors: {error_detail}"
+            )
+        else:
+            status = (
+                f"Done! {n_real} sprite(s) generated by FLUX. "
+                "Click Launch Game to play."
+            )
 
         # Show the enhanced prompts that were used
         prompt_summary = "\n\n".join(