Fix prompt refresh and broader asset applying

app.py

@@ -1,6 +1,7 @@
 import base64
 import hashlib
 import io
+import json
 import random
 import re
 import tempfile
@@ -145,6 +146,21 @@ def write_gallery_image(content: bytes, role: str) -> str:
 
 
 def palette_for(text: str) -> list[tuple[int, int, int]]:
+    lowered = text.lower()
+    themed = [
+        (("lava", "fire", "volcano", "hell"), [(139, 28, 18), (239, 84, 34), (42, 18, 16), (255, 198, 71)]),
+        (("ice", "snow", "frost", "arctic"), [(75, 151, 191), (187, 235, 255), (24, 63, 96), (240, 252, 255)]),
+        (("desert", "sand", "dune", "western"), [(194, 139, 72), (236, 198, 112), (102, 66, 36), (255, 226, 151)]),
+        (("ocean", "sea", "water", "underwater"), [(25, 103, 155), (57, 191, 205), (8, 43, 85), (177, 239, 224)]),
+        (("forest", "garden", "grass", "jungle"), [(51, 132, 72), (126, 196, 81), (24, 72, 43), (242, 201, 84)]),
+        (("dungeon", "castle", "stone", "crypt"), [(78, 82, 96), (133, 140, 154), (34, 36, 48), (181, 154, 94)]),
+        (("city", "street", "cyber", "neon"), [(40, 52, 87), (39, 230, 226), (19, 21, 38), (239, 67, 154)]),
+        (("sci-fi", "scifi", "space", "arena", "metal", "shooter"), [(30, 216, 236), (235, 64, 96), (18, 22, 31), (245, 192, 51)]),
+    ]
+    for words, colors in themed:
+        if any(word in lowered for word in words):
+            return colors
+
     digest = hashlib.sha256(text.encode("utf-8")).digest()
     base = digest[0] % 360
 
@@ -282,6 +298,59 @@ def draw_background(spec: AssetSpec, rng: random.Random) -> bytes:
         image.save(out, format="PNG")
         return out.getvalue()
 
+    if any(word in prompt for word in ("lava", "fire", "volcano")):
+        image = Image.new("RGBA", (spec.width, spec.height), (52, 23, 18, 255))
+        draw = ImageDraw.Draw(image, "RGBA")
+        for y in range(0, spec.height, 58):
+            draw.rectangle((0, y, spec.width, y + 30), fill=(70, 32, 25, 255))
+        for _ in range(18):
+            x = rng.randint(-80, spec.width)
+            y = rng.randint(0, spec.height)
+            draw.line((x, y, x + rng.randint(120, 320), y + rng.randint(-20, 20)), fill=(255, 111, 43, 220), width=rng.randint(5, 12))
+        for _ in range(45):
+            x = rng.randint(0, spec.width)
+            y = rng.randint(0, spec.height)
+            r = rng.randint(5, 20)
+            draw.ellipse((x - r, y - r, x + r, y + r), fill=(255, 192, 53, rng.randint(70, 140)))
+        out = io.BytesIO()
+        image.save(out, format="PNG")
+        return out.getvalue()
+
+    if any(word in prompt for word in ("dungeon", "castle", "stone", "crypt")):
+        image = Image.new("RGBA", (spec.width, spec.height), (48, 50, 62, 255))
+        draw = ImageDraw.Draw(image, "RGBA")
+        tile = 54
+        for y in range(0, spec.height, tile):
+            for x in range(0, spec.width, tile):
+                shade = rng.randint(48, 78)
+                draw.rectangle((x, y, x + tile - 2, y + tile - 2), fill=(shade, shade + 2, shade + 8, 255), outline=(27, 29, 38, 190))
+        for _ in range(22):
+            x = rng.randint(0, spec.width)
+            y = rng.randint(0, spec.height)
+            draw.ellipse((x - 8, y - 8, x + 8, y + 8), fill=(185, 146, 80, 110))
+        out = io.BytesIO()
+        image.save(out, format="PNG")
+        return out.getvalue()
+
+    if any(word in prompt for word in ("ocean", "sea", "water", "underwater")):
+        image = Image.new("RGBA", (spec.width, spec.height), (8, 54, 92, 255))
+        draw = ImageDraw.Draw(image, "RGBA")
+        for y in range(spec.height):
+            mix = y / max(1, spec.height - 1)
+            draw.line((0, y, spec.width, y), fill=(int(16 + 6 * mix), int(99 + 62 * mix), int(151 + 58 * mix), 255))
+        for _ in range(26):
+            x = rng.randint(-100, spec.width)
+            y = rng.randint(0, spec.height)
+            draw.arc((x, y, x + 180, y + 70), 0, 180, fill=(165, 232, 225, 90), width=3)
+        for _ in range(38):
+            x = rng.randint(0, spec.width)
+            y = rng.randint(0, spec.height)
+            r = rng.randint(2, 8)
+            draw.ellipse((x - r, y - r, x + r, y + r), outline=(205, 247, 245, 120), width=2)
+        out = io.BytesIO()
+        image.save(out, format="PNG")
+        return out.getvalue()
+
     for y in range(spec.height):
         mix = y / max(1, spec.height - 1)
         color = tuple(int(colors[2][i] * (1 - mix) + colors[0][i] * mix) for i in range(3))
@@ -480,16 +549,40 @@ def replacement_names(spec: AssetSpec) -> set[str]:
         names.update({"player.png", "sprite_player.jpg", "hero.png"})
     if slug == "enemy":
         names.update({"enemy.png", "monster.png", "sprite_enemy.jpg"})
+    if slug == "bullet":
+        names.update({"bullet.png", "projectile.png", "laser.png", "shot.png", "sprite_bullet.jpg"})
     return names
 
 
+def asset_aliases(spec: AssetSpec) -> list[str]:
+    slug = slugify(spec.role)
+    aliases = {slug, spec.role.strip().lower(), spec.filename.lower()}
+    prompt = spec.prompt.lower()
+    role_groups = {
+        "player": ("player", "hero", "character", "avatar", "ship", "shooter", "tank", "knight", "wizard"),
+        "enemy": ("enemy", "monster", "alien", "drone", "foe", "zombie", "boss", "hazard"),
+        "bullet": ("bullet", "projectile", "laser", "shot", "missile", "beam", "ammo"),
+        "coin": ("coin", "gem", "seed", "star", "pickup", "collectible", "key", "orb"),
+        "background": ("background", "backdrop", "bg", "map", "level", "arena", "scene", "world", "floor"),
+    }
+    for group, words in role_groups.items():
+        if group in slug or any(word in prompt for word in words):
+            aliases.update(words)
+    return sorted(alias for alias in aliases if alias)
+
+
 def embed_assets(html_code: str, assets: dict[str, str], specs: list[AssetSpec]) -> str:
     output = html_code
     manifest_lines = ["<!-- Embedded game assets generated by Image Generator for HTML Games"]
     background_uri = None
+    asset_map: dict[str, str] = {}
+    alias_map: dict[str, list[str]] = {}
 
     for spec in specs:
         data_uri = assets[spec.role]
+        slug = slugify(spec.role)
+        asset_map[slug] = data_uri
+        alias_map[slug] = asset_aliases(spec)
         manifest_lines.append(f"{spec.role}: {spec.filename}")
         if is_background_spec(spec) and background_uri is None:
             background_uri = data_uri
@@ -500,15 +593,52 @@ def embed_assets(html_code: str, assets: dict[str, str], specs: list[AssetSpec])
 
     manifest_lines.append("-->")
     manifest = "\n".join(manifest_lines) + "\n"
-    asset_map = ",\n".join(
-        f"    {slugify(spec.role)!r}: {assets[spec.role]!r}" for spec in specs
-    )
+    asset_json = json.dumps(asset_map)
+    alias_json = json.dumps(alias_map)
+    background_json = json.dumps(background_uri)
     helper_script = f"""<script>
-  window.GENERATED_GAME_ASSETS = {{
-{asset_map}
-  }};
+(function () {{
+  var ASSETS = {asset_json};
+  var ALIASES = {alias_json};
+  window.GENERATED_GAME_ASSETS = ASSETS;
+
+  function basename(value) {{
+    return String(value || "").split("?")[0].split("#")[0].split("/").pop().toLowerCase();
+  }}
+
+  function pickAsset(value) {{
+    var text = String(value || "").toLowerCase();
+    if (!text || text.indexOf("data:image/") === 0) return value;
+    var file = basename(text);
+    for (var role in ASSETS) {{
+      var aliases = ALIASES[role] || [role];
+      for (var i = 0; i < aliases.length; i++) {{
+        var alias = String(aliases[i]).toLowerCase();
+        if (!alias) continue;
+        if (file === alias || file === alias + ".png" || file === "sprite_" + alias + ".png") return ASSETS[role];
+        if (file.indexOf(alias) !== -1 || text.indexOf("/" + alias) !== -1 || text.indexOf("_" + alias) !== -1) return ASSETS[role];
+      }}
+    }}
+    return value;
+  }}
+
+  var descriptor = Object.getOwnPropertyDescriptor(HTMLImageElement.prototype, "src");
+  if (descriptor && descriptor.set && !HTMLImageElement.prototype.__generatedAssetMapper) {{
+    Object.defineProperty(HTMLImageElement.prototype, "src", {{
+      get: function () {{ return descriptor.get.call(this); }},
+      set: function (value) {{ descriptor.set.call(this, pickAsset(value)); }},
+      configurable: true,
+      enumerable: descriptor.enumerable
+    }});
+    HTMLImageElement.prototype.__generatedAssetMapper = true;
+  }}
+
   window.addEventListener("DOMContentLoaded", function () {{
-    var background = {background_uri!r};
+    document.querySelectorAll("img").forEach(function (img) {{
+      var mapped = pickAsset(img.getAttribute("src") || img.src);
+      if (mapped !== (img.getAttribute("src") || img.src)) img.src = mapped;
+    }});
+    var background = {background_json};
     if (!background) return;
     document.querySelectorAll("canvas").forEach(function (canvas) {{
       canvas.style.backgroundImage = "url(" + background + ")";
@@ -516,6 +646,7 @@ def embed_assets(html_code: str, assets: dict[str, str], specs: list[AssetSpec])
       canvas.style.backgroundPosition = "center";
     }});
   }});
+}})();
 </script>"""
 
     if "</head>" in output: