Improve top-down shooter asset matching

app.py

@@ -178,6 +178,40 @@ def draw_background(spec: AssetSpec, rng: random.Random) -> bytes:
     draw = ImageDraw.Draw(image, "RGBA")
     prompt = spec.prompt.lower()
 
+    if any(word in prompt for word in ("sci-fi", "scifi", "arena", "metal", "hazard", "grid", "shooter")):
+        image = Image.new("RGBA", (spec.width, spec.height), (18, 22, 31, 255))
+        draw = ImageDraw.Draw(image, "RGBA")
+        panel = 64
+        for y in range(0, spec.height, panel):
+            for x in range(0, spec.width, panel):
+                shade = 24 + ((x // panel + y // panel) % 2) * 12
+                draw.rectangle(
+                    (x, y, x + panel - 2, y + panel - 2),
+                    fill=(shade, shade + 4, shade + 14, 255),
+                    outline=(58, 74, 92, 170),
+                )
+        for x in range(0, spec.width, 96):
+            draw.line((x, 0, x, spec.height), fill=(34, 215, 236, 95), width=2)
+        for y in range(0, spec.height, 96):
+            draw.line((0, y, spec.width, y), fill=(34, 215, 236, 95), width=2)
+        for _ in range(8):
+            x = rng.randint(0, spec.width - 120)
+            y = rng.randint(0, spec.height - 34)
+            draw.rectangle((x, y, x + 120, y + 34), fill=(36, 42, 53, 235), outline=(110, 125, 142, 210))
+            for stripe in range(0, 120, 24):
+                draw.polygon(
+                    [(x + stripe, y), (x + stripe + 12, y), (x + stripe - 12, y + 34), (x + stripe - 24, y + 34)],
+                    fill=(245, 192, 51, 210),
+                )
+        for _ in range(18):
+            x = rng.randint(0, spec.width)
+            y = rng.randint(0, spec.height)
+            r = rng.randint(10, 32)
+            draw.ellipse((x - r, y - r, x + r, y + r), outline=(224, 54, 85, 100), width=2)
+        out = io.BytesIO()
+        image.save(out, format="PNG")
+        return out.getvalue()
+
     if any(word in prompt for word in ("space", "star", "nebula", "galaxy", "cosmic")):
         image = Image.new("RGBA", (spec.width, spec.height), (3, 4, 18, 255))
         draw = ImageDraw.Draw(image, "RGBA")
@@ -279,14 +313,77 @@ def draw_sprite(spec: AssetSpec, rng: random.Random) -> bytes:
     draw = ImageDraw.Draw(image, "RGBA")
     cx, cy = spec.width // 2, spec.height // 2
     scale = min(spec.width, spec.height) / 128
-
-    shadow = (int(cx - 36 * scale), int(cy + 34 * scale), int(cx + 36 * scale), int(cy + 48 * scale))
-    draw.ellipse(shadow, fill=(0, 0, 0, 65))
+    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")):
+        glow = colors[1] + (85,)
+        draw.ellipse((cx - 30 * scale, cy - 42 * scale, cx + 30 * scale, cy + 42 * scale), fill=glow)
+        draw.rounded_rectangle(
+            (cx - 9 * scale, cy - 40 * scale, cx + 9 * scale, cy + 38 * scale),
+            radius=int(9 * scale),
+            fill=outline,
+        )
+        draw.rounded_rectangle(
+            (cx - 5 * scale, cy - 35 * scale, cx + 5 * scale, cy + 33 * scale),
+            radius=int(5 * scale),
+            fill=accent,
+        )
+        out = io.BytesIO()
+        image.save(out, format="PNG")
+        return out.getvalue()
+
+    if 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 - 16 * scale, cy - 16 * scale, cx + 16 * scale, cy + 16 * scale), fill=accent)
+        for side in (-1, 1):
+            draw.polygon(
+                [
+                    (cx + side * 28 * scale, cy - 8 * scale),
+                    (cx + side * 56 * scale, cy - 28 * scale),
+                    (cx + side * 48 * scale, cy + 20 * scale),
+                ],
+                fill=outline,
+            )
+            draw.line((cx + side * 14 * scale, cy, cx + side * 58 * scale, cy), fill=trim, width=max(2, int(5 * scale)))
+        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")):
+        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))
+        draw.polygon(
+            [
+                (cx, cy - 50 * scale),
+                (cx - 34 * scale, cy + 34 * scale),
+                (cx, cy + 20 * scale),
+                (cx + 34 * scale, cy + 34 * scale),
+            ],
+            fill=outline,
+        )
+        draw.polygon(
+            [
+                (cx, cy - 42 * scale),
+                (cx - 24 * scale, cy + 23 * scale),
+                (cx, cy + 12 * scale),
+                (cx + 24 * 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)))
+        out = io.BytesIO()
+        image.save(out, format="PNG")
+        return out.getvalue()
+
+    shadow = (int(cx - 36 * scale), int(cy + 34 * scale), int(cx + 36 * scale), int(cy + 48 * scale))
+    draw.ellipse(shadow, fill=(0, 0, 0, 65))
+
     body_box = (cx - 26 * scale, cy - 18 * scale, cx + 26 * scale, cy + 34 * scale)
     head_box = (cx - 20 * scale, cy - 52 * scale, cx + 20 * scale, cy - 12 * scale)
     draw.rounded_rectangle(body_box, radius=int(12 * scale), fill=outline)