Use whole 2D sprite generator for sprite roles

app.py

@@ -107,6 +107,7 @@ FREE_IMAGE_MODEL = os.environ.get("FREE_IMAGE_MODEL", "segmind/tiny-sd")
 FREE_IMAGE_STEPS = int(os.environ.get("FREE_IMAGE_STEPS", "5"))
 USE_HF_PROMPT_PROVIDER = os.environ.get("USE_HF_PROMPT_PROVIDER", "0") == "1"
 USE_HF_IMAGE_PROVIDER = os.environ.get("USE_HF_IMAGE_PROVIDER", "0") == "1"
+USE_DIFFUSION_FOR_SPRITES = os.environ.get("USE_DIFFUSION_FOR_SPRITES", "0") == "1"
 HF_IMAGE_MODEL = os.environ.get("HF_IMAGE_MODEL", "black-forest-labs/FLUX.1-schnell")
 HF_PROMPT_MODEL = os.environ.get("HF_PROMPT_MODEL", "Qwen/Qwen2.5-Coder-7B-Instruct:fastest")
 HF_PROMPT_ENDPOINT = os.environ.get("HF_PROMPT_ENDPOINT", "https://router.huggingface.co/v1/chat/completions")
@@ -1105,6 +1106,15 @@ def hf_image_png(spec: AssetSpec, index: int, run_id: int) -> tuple[bytes | None
 
 
 def generate_asset(spec: AssetSpec, index: int, run_id: int) -> tuple[str, str, str | None, str]:
+    if not is_background_spec(spec) and not USE_DIFFUSION_FOR_SPRITES:
+        png_content = local_asset_png(spec, index, run_id)
+        return (
+            png_bytes_to_data_uri(png_content),
+            write_gallery_image(png_content, spec.role),
+            None,
+            "shape-aware 2D sprite generator",
+        )
+
     png_content, free_error = free_diffusion_png(spec, index, run_id)
     source = FREE_IMAGE_MODEL
     error = free_error