Make sprites change with theme

app.py

@@ -2,6 +2,7 @@ import base64
 import hashlib
 import io
 import json
+import math
 import random
 import re
 import tempfile
@@ -377,18 +378,19 @@ def draw_background(spec: AssetSpec, rng: random.Random) -> bytes:
 
 
 def draw_sprite(spec: AssetSpec, rng: random.Random) -> bytes:
-    colors = palette_for(spec.prompt)
+    colors = palette_for(f"{spec.role} {spec.prompt}")
     image = Image.new("RGBA", (spec.width, spec.height), (0, 0, 0, 0))
     draw = ImageDraw.Draw(image, "RGBA")
     cx, cy = spec.width // 2, spec.height // 2
     scale = min(spec.width, spec.height) / 128
+    role = slugify(spec.role)
     role_prompt = f"{spec.role} {spec.prompt}".lower()
     outline = (18, 24, 31, 255)
     body = colors[0] + (255,)
     accent = colors[1] + (255,)
     trim = colors[3] + (255,)
 
-    if any(word in role_prompt for word in ("bullet", "projectile", "laser", "energy")):
+    if role in ("bullet", "projectile", "laser", "shot") or any(word in role_prompt for word in ("bullet", "projectile", "laser", "energy")):
         glow = colors[1] + (85,)
         draw.ellipse((cx - 30 * scale, cy - 42 * scale, cx + 30 * scale, cy + 42 * scale), fill=glow)
         draw.rounded_rectangle(
@@ -405,9 +407,9 @@ def draw_sprite(spec: AssetSpec, rng: random.Random) -> bytes:
         image.save(out, format="PNG")
         return out.getvalue()
 
-    if any(word in role_prompt for word in ("enemy", "drone", "alien", "monster")):
+    if role in ("enemy", "monster", "alien", "drone", "foe", "boss") or any(word in role_prompt for word in ("enemy", "drone", "alien", "monster")):
         draw.ellipse((cx - 38 * scale, cy - 28 * scale, cx + 38 * scale, cy + 28 * scale), fill=outline)
-        draw.ellipse((cx - 32 * scale, cy - 22 * scale, cx + 32 * scale, cy + 22 * scale), fill=(150, 35, 63, 255))
+        draw.ellipse((cx - 32 * scale, cy - 22 * scale, cx + 32 * scale, cy + 22 * scale), fill=body)
         draw.ellipse((cx - 16 * scale, cy - 16 * scale, cx + 16 * scale, cy + 16 * scale), fill=accent)
         for side in (-1, 1):
             draw.polygon(
@@ -419,33 +421,47 @@ def draw_sprite(spec: AssetSpec, rng: random.Random) -> bytes:
                 fill=outline,
             )
             draw.line((cx + side * 14 * scale, cy, cx + side * 58 * scale, cy), fill=trim, width=max(2, int(5 * scale)))
+        for _ in range(5):
+            angle = rng.random() * 6.283
+            r = rng.randint(int(12 * scale), int(30 * scale))
+            px = cx + int(r * math.cos(angle))
+            py = cy + int(r * math.sin(angle))
+            draw.ellipse((px - 3 * scale, py - 3 * scale, px + 3 * scale, py + 3 * scale), fill=trim)
         out = io.BytesIO()
         image.save(out, format="PNG")
         return out.getvalue()
 
-    if any(word in role_prompt for word in ("shooter", "ship", "tank", "hero", "player", "armored")):
+    if role in ("player", "hero", "ship", "tank", "avatar") or any(word in role_prompt for word in ("shooter", "ship", "tank", "hero", "player", "armored")):
         shadow = (int(cx - 36 * scale), int(cy + 34 * scale), int(cx + 36 * scale), int(cy + 48 * scale))
         draw.ellipse(shadow, fill=(0, 0, 0, 55))
+        wing = rng.randint(28, 39)
+        nose = rng.randint(45, 56)
         draw.polygon(
             [
-                (cx, cy - 50 * scale),
-                (cx - 34 * scale, cy + 34 * scale),
+                (cx, cy - nose * scale),
+                (cx - wing * scale, cy + 34 * scale),
                 (cx, cy + 20 * scale),
-                (cx + 34 * scale, cy + 34 * scale),
+                (cx + wing * scale, cy + 34 * scale),
             ],
             fill=outline,
         )
         draw.polygon(
             [
-                (cx, cy - 42 * scale),
-                (cx - 24 * scale, cy + 23 * scale),
+                (cx, cy - (nose - 8) * scale),
+                (cx - (wing - 10) * scale, cy + 23 * scale),
                 (cx, cy + 12 * scale),
-                (cx + 24 * scale, cy + 23 * scale),
+                (cx + (wing - 10) * scale, cy + 23 * scale),
             ],
             fill=body,
         )
         draw.ellipse((cx - 11 * scale, cy - 14 * scale, cx + 11 * scale, cy + 10 * scale), fill=accent)
         draw.line((cx, cy - 42 * scale, cx, cy - 62 * scale), fill=trim, width=max(3, int(6 * scale)))
+        for side in (-1, 1):
+            x1 = cx + side * rng.randint(14, 23) * scale
+            x2 = cx + side * rng.randint(24, 32) * scale
+            y1 = cy + rng.randint(8, 18) * scale
+            y2 = cy + rng.randint(20, 30) * scale
+            draw.rectangle((min(x1, x2), y1, max(x1, x2), y2), fill=trim)
         out = io.BytesIO()
         image.save(out, format="PNG")
         return out.getvalue()