Update app.py

app.py

@@ -1,19 +1,21 @@
 """
-Text-to-Game Generator - Groq Edition (Truly Free)
+Text-to-Game Generator - Groq + FLUX Edition
 - Code  : llama-3.1-8b-instant via Groq API (free, no credit card)
-- Images: canvas-drawn graphics only (zero API, zero quota, zero cost)
+- Images: black-forest-labs/FLUX.1-schnell via nscale (HF Inference API, free with HF_TOKEN)
 
-No GPU needed. No compilation. No HF quota. No credit card.
-Setup: add GROQ_API_KEY as a Space secret.
+Setup: add GROQ_API_KEY and HF_TOKEN as Space secrets.
 """
 
 import os
 import re
+import io
 import base64
 import traceback
 
 import gradio as gr
 from openai import OpenAI
+from huggingface_hub import InferenceClient
+from PIL import Image
 
 # ---------------------------------------------------------------------------
 # Google AI Studio client (OpenAI-compatible endpoint)
@@ -43,7 +45,21 @@ def get_client():
 
 
 # ---------------------------------------------------------------------------
-# Game type configs — canvas-drawn graphics, no image files
+# Image model — FLUX.1-schnell via nscale (free with HF_TOKEN)
+# ---------------------------------------------------------------------------
+
+IMAGE_MODEL = "black-forest-labs/FLUX.1-schnell"
+HF_TOKEN    = os.environ.get("HF_TOKEN", "")
+
+
+def get_image_client():
+    if not HF_TOKEN:
+        return None
+    return InferenceClient(provider="nscale", api_key=HF_TOKEN)
+
+
+# ---------------------------------------------------------------------------
+# Game type configs — uses sprite_NAME.png when HF_TOKEN set, else canvas
 # ---------------------------------------------------------------------------
 
 GAME_TYPES = {
@@ -146,6 +162,69 @@ THEME_EXAMPLES = {
     "Surprise Me!":     [["A sentient library that rearranges itself"], ["Deep sea bioluminescent creatures"], ["Retro space diner on a comet"]],
 }
 
+# ---------------------------------------------------------------------------
+# Sprite helpers
+# ---------------------------------------------------------------------------
+
+def _pil_to_data_uri(pil_image: Image.Image, size: int = 64) -> str:
+    img = pil_image.resize((size, size), Image.LANCZOS)
+    buf = io.BytesIO()
+    img.save(buf, format="PNG")
+    return "data:image/png;base64," + base64.b64encode(buf.getvalue()).decode()
+
+
+def _colored_placeholder(name: str) -> str:
+    colours = ["#e74c3c", "#3498db", "#2ecc71", "#f39c12", "#9b59b6", "#1abc9c"]
+    colour  = colours[abs(hash(name)) % len(colours)]
+    label   = name.replace("sprite_", "")[:6]
+    svg = (
+        '<svg xmlns="http://www.w3.org/2000/svg" width="64" height="64">'
+        '<rect width="64" height="64" fill="' + colour + '" rx="8"/>'
+        '<text x="32" y="38" font-size="10" text-anchor="middle" fill="white">' + label + '</text>'
+        '</svg>'
+    )
+    return "data:image/svg+xml;base64," + base64.b64encode(svg.encode()).decode()
+
+
+def _sprite_prompt(sprite_name: str, theme: str) -> str:
+    label = sprite_name.replace("sprite_", "").replace("_", " ")
+    return (
+        "Pixel-art game sprite: " + label + ". Theme: " + theme + ". "
+        "Vibrant colours, clear silhouette, plain dark background, 64x64 pixel style."
+    )
+
+
+def _find_sprite_filenames(html_code: str) -> list:
+    pattern = r"""(?:src\s*=\s*['"]|\.src\s*=\s*['"])\s*(sprite_[a-zA-Z0-9_]+\.png)\s*['"]"""
+    return list(dict.fromkeys(re.findall(pattern, html_code)))[:3]
+
+
+def _inject_sprites(html_code: str, sprite_map: dict) -> str:
+    for fname, data_uri in sprite_map.items():
+        html_code = html_code.replace('"' + fname + '"', '"' + data_uri + '"')
+        html_code = html_code.replace("'" + fname + "'", "'" + data_uri + "'")
+    return html_code
+
+
+def generate_sprites(sprite_names: list, theme: str) -> dict:
+    if not sprite_names:
+        return {}
+    client = get_image_client()
+    if not client:
+        return {n: _colored_placeholder(n.replace(".png","")) for n in sprite_names}
+    mapping = {}
+    for fname in sprite_names:
+        name_no_ext = fname.replace(".png", "")
+        prompt = _sprite_prompt(name_no_ext, theme)
+        try:
+            pil_img = client.text_to_image(prompt, model=IMAGE_MODEL)
+            mapping[fname] = _pil_to_data_uri(pil_img, size=64)
+        except Exception as exc:
+            print("[FLUX] Failed '" + fname + "': " + str(exc))
+            mapping[fname] = _colored_placeholder(name_no_ext)
+    return mapping
+
+
 # ---------------------------------------------------------------------------
 # Code generation
 # ---------------------------------------------------------------------------
@@ -156,13 +235,22 @@ def generate_game_code(game_type: str, theme: str, temperature: float, max_new_t
 
     template    = GAME_TYPES[game_type]["prompt_template"]
     user_prompt = template.format(theme=theme.strip())
-    system_msg  = (
-        "You are an expert HTML5 game developer. "
-        "Write a complete, working, single-file HTML5 game using only canvas 2D drawing. "
-        "Never use image files or external resources. "
-        "Draw everything with canvas shapes, colors, and text. "
-        "Output ONLY the raw HTML - no markdown fences, no explanation."
-    )
+    use_sprites = bool(HF_TOKEN)
+
+    if use_sprites:
+        system_msg = (
+            "You are an expert HTML5 game developer. "
+            "Write a complete, working, single-file HTML5 game using canvas and vanilla JavaScript. "
+            "Use new Image() with src='sprite_NAME.png' for every visual asset (max 3 sprites). "
+            "Output ONLY the raw HTML - no markdown fences, no explanation."
+        )
+    else:
+        system_msg = (
+            "You are an expert HTML5 game developer. "
+            "Write a complete, working, single-file HTML5 game using only canvas 2D drawing. "
+            "Never use image files. Draw everything with canvas shapes, colors, and text. "
+            "Output ONLY the raw HTML - no markdown fences, no explanation."
+        )
 
     try:
         client = get_client()
@@ -181,7 +269,20 @@ def generate_game_code(game_type: str, theme: str, temperature: float, max_new_t
         if "<html" not in code.lower() and "<!doctype" not in code.lower():
             code = _wrap_in_html(code, theme)
 
-        status = "Done! Game generated. Click Launch Game to play."
+        if use_sprites:
+            sprite_names = _find_sprite_filenames(code)
+            sprite_map   = generate_sprites(sprite_names, theme.strip())
+            code         = _inject_sprites(code, sprite_map)
+            n_real       = sum(1 for v in sprite_map.values() if "image/png" in v)
+            n_fallback   = len(sprite_map) - n_real
+            status = (
+                "Done! " + str(len(sprite_names)) + " sprite(s) - "
+                + str(n_real) + " by FLUX.1-schnell, "
+                + str(n_fallback) + " fallback. Click Launch Game to play."
+            )
+        else:
+            status = "Done! Game generated (canvas graphics). Click Launch Game to play."
+
         return code, status, _build_preview(code)
 
     except Exception as exc: