Update app.py

app.py

@@ -31,7 +31,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 — better model for creative prompts
-IMAGE_MODEL = "black-forest-labs/FLUX.1-schnell"  # nscale
+IMAGE_MODEL = "black-forest-labs/FLUX.1-schnell"  # hf-inference
 
 
 def get_groq_client():
@@ -49,7 +49,7 @@ def get_groq_client():
 def get_image_client():
     if not HF_TOKEN:
         return None
-    return InferenceClient(provider="nscale", api_key=HF_TOKEN)
+    return InferenceClient(provider="hf-inference", api_key=HF_TOKEN)
 
 
 # ---------------------------------------------------------------------------
@@ -260,15 +260,16 @@ def generate_sprites(image_prompts: dict) -> dict:
             sprite_map[sprite_name] = _colored_placeholder(sprite_name.replace(".png", ""))
             continue
         try:
-            # Background gets full size, sprites get 64x64
+            # 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}")
+            print(f"[FLUX] Generated {sprite_name} OK")
         except Exception as exc:
-            print(f"[FLUX] Failed {sprite_name}: {exc}")
-            sprite_map[sprite_name] = _colored_placeholder(sprite_name.replace(".png", ""))
+            error_msg = str(exc)
+            print(f"[FLUX] FAILED {sprite_name}: {error_msg}")
+            sprite_map[sprite_name] = ("ERROR:" + error_msg, _colored_placeholder(sprite_name.replace(".png", "")))[1]
 
     return sprite_map