Upload 3 files

README.md

@@ -5,28 +5,40 @@ colorTo: purple
 sdk: gradio
 app_file: app.py
 pinned: false
-sdk_version: 6.20.0
 ---
 
 # AI BoardGame Forge
 
-A web app that turns any theme into a mini card game with AI-generated artwork.
+A web app that turns any theme into a mini card game with AI-generated text
+and AI-generated artwork.
 
 Enter a theme, such as "Space Cats" or "Ancient Pirates", or pick one of the
 example themes provided, and the app generates a game title, description,
-rules, and 4 unique cards with AI-generated artwork.
+rules, and 4 unique cards, each with its own AI-generated artwork.
 
 ## How it works
 
-- Game text (title, description, rules, cards) is generated locally using a
-  procedural generator that combines the theme with pre-built word banks and
-  templates.
+- Game text (title, description, rules, cards) is generated by
+  Qwen2.5-0.5B-Instruct, a small, free, open-source language model that
+  runs locally through the Hugging Face `transformers` library. No API key
+  or account is needed.
+- If the model's reply cannot be parsed as valid JSON with exactly 4 cards,
+  the app falls back to a local template generator so it never breaks.
 - Card artwork is generated using Pollinations.ai, a free image generation
-  service that requires no API key.
+  service that also requires no API key. If an image request fails, a
+  themed placeholder image is used instead.
 - Each card is composed into a trading-card layout using Pillow (PIL).
 - The interface is built with Gradio.
 
 ## Files
 
 - `app.py`: the full application code
-- `requirements.txt`: the Python packages needed to run it
+- `requirements.txt`: the Python packages needed to run it
+
+## Notes on hardware
+
+This Space loads a small language model in addition to making image
+requests, so the free CPU tier works but the first response after startup
+may take a bit longer while the model loads into memory. Generating a full
+game typically takes somewhere between 20 and 90 seconds depending on
+available compute.

app.py

@@ -1,22 +1,114 @@
 """
 AI BoardGame Forge
 A Gradio web app that turns any theme into a mini card game with
-AI-generated artwork. Built for Hugging Face Spaces.
+AI-generated text and AI-generated artwork. Built for Hugging Face Spaces.
 """
 
+import json
 import random
 import urllib.parse
 from io import BytesIO
 
 import requests
 import gradio as gr
+import torch
 from PIL import Image, ImageDraw, ImageFont
+from transformers import AutoModelForCausalLM, AutoTokenizer
 
 
 # ------------------------------------------------------------
-# Game text generator
+# Language model setup
 # ------------------------------------------------------------
 
+MODEL_NAME = "Qwen/Qwen2.5-0.5B-Instruct"
+
+print("Loading language model, this can take a minute on first startup...")
+tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
+model = AutoModelForCausalLM.from_pretrained(
+    MODEL_NAME,
+    torch_dtype="auto",
+    device_map="auto",
+)
+print("Model loaded.")
+
+
+def ask_llm(system_prompt, user_prompt, max_new_tokens=500):
+    """Sends a system prompt and a user prompt to the model and returns its text reply."""
+    messages = [
+        {"role": "system", "content": system_prompt},
+        {"role": "user", "content": user_prompt},
+    ]
+
+    prompt_text = tokenizer.apply_chat_template(
+        messages, tokenize=False, add_generation_prompt=True
+    )
+    inputs = tokenizer(prompt_text, return_tensors="pt").to(model.device)
+
+    output_tokens = model.generate(
+        **inputs,
+        max_new_tokens=max_new_tokens,
+        do_sample=True,
+        temperature=0.8,
+        pad_token_id=tokenizer.eos_token_id,
+    )
+
+    new_tokens = output_tokens[0][inputs["input_ids"].shape[1]:]
+    reply = tokenizer.decode(new_tokens, skip_special_tokens=True)
+    return reply.strip()
+
+
+# ------------------------------------------------------------
+# Game text generator (LLM-based, with a template fallback)
+# ------------------------------------------------------------
+
+SYSTEM_PROMPT = """You are a creative board game designer.
+Given a theme, invent a simple, fun mini card game based on it.
+
+Reply with ONLY valid JSON, no extra text and no markdown code fences, in exactly this format:
+
+{
+  "title": "a catchy game title",
+  "description": "2 to 3 sentences describing how the game works",
+  "rules": "3 to 5 short, simple rules a beginner could follow, written as one string",
+  "cards": [
+    {"name": "card name", "type": "card type", "effect": "short special ability"},
+    {"name": "card name", "type": "card type", "effect": "short special ability"},
+    {"name": "card name", "type": "card type", "effect": "short special ability"},
+    {"name": "card name", "type": "card type", "effect": "short special ability"}
+  ]
+}
+
+Keep everything short, simple, and beginner-friendly. Always include exactly 4 cards."""
+
+
+def extract_json(text):
+    """Pulls out the {...} portion of a string, in case the model added extra text around it."""
+    start = text.find("{")
+    end = text.rfind("}")
+    if start == -1 or end == -1:
+        raise ValueError("No JSON object found in the model's reply.")
+    return text[start:end + 1]
+
+
+def generate_game_with_llm(theme):
+    """Asks the language model to design the game and returns it as a dictionary."""
+    user_prompt = f"Theme: {theme}\nDesign the card game now."
+    raw_reply = ask_llm(SYSTEM_PROMPT, user_prompt)
+
+    json_text = extract_json(raw_reply)
+    game_data = json.loads(json_text)
+
+    required_keys = {"title", "description", "rules", "cards"}
+    if not required_keys.issubset(game_data.keys()):
+        raise ValueError("Model reply is missing required fields.")
+    if len(game_data["cards"]) != 4:
+        raise ValueError("Model did not return exactly 4 cards.")
+
+    return game_data
+
+
+# ---- Fallback generator (used only if the LLM output cannot be used) ----
+
 TITLE_TEMPLATES = [
     "{theme} Wars",
     "The {theme} Chronicles",
@@ -24,14 +116,11 @@ TITLE_TEMPLATES = [
     "{theme}: The Card Game",
     "Rise of the {theme}",
     "{theme} Showdown",
-    "Legends of {theme}",
-    "{theme} Uprising",
 ]
 
 DESCRIPTION_TEMPLATES = [
     "In this game, players take on the role of characters caught up in a {theme} scenario. Each player uses cards to outsmart their opponents and be the first to win.",
     "Players compete in a world inspired by {theme}, drawing and playing cards to gain the upper hand. Strategy and a little luck decide the winner.",
-    "Set in a world inspired by {theme}, each player builds a small hand of powerful cards and battles to reach the win condition first.",
 ]
 
 RULES_TEMPLATE = """1. Each player starts with a hand of 4 cards.
@@ -44,7 +133,7 @@ CARD_TYPES = ["Attack", "Defense", "Support", "Trick", "Special", "Boost", "Sabo
 
 NAME_PARTS = [
     "Guardian", "Blade", "Shadow", "Spark", "Wanderer", "Sentinel",
-    "Rebel", "Trickster", "Champion", "Outcast", "Phantom", "Rogue",
+    "Rebel", "Trickster", "Champion", "Outcast",
 ]
 
 EFFECT_POOL = [
@@ -53,16 +142,12 @@ EFFECT_POOL = [
     "Steal one card from an opponent's hand.",
     "Skip your opponent's next turn.",
     "Double your score this round.",
-    "Swap this card with any card in the discard pile.",
-    "Force all players to reveal their hand.",
     "Gain +3 power for the rest of the game.",
-    "Discard your hand and draw a fresh one.",
-    "Copy the effect of the last card played.",
 ]
 
 
-def generate_game(theme):
-    """Builds a full mini card game (title, description, rules, 4 cards) from a theme."""
+def generate_game_template(theme):
+    """Local, non-AI backup generator used only if the language model fails."""
     theme_clean = theme.strip().title()
 
     title = random.choice(TITLE_TEMPLATES).format(theme=theme_clean)
@@ -79,6 +164,18 @@ def generate_game(theme):
     return {"title": title, "description": description, "rules": rules, "cards": cards}
 
 
+def generate_game(theme):
+    """
+    Main entry point. Tries the language model first, and only falls back
+    to the template generator if the model's reply could not be used.
+    """
+    try:
+        return generate_game_with_llm(theme)
+    except Exception as error:
+        print(f"LLM generation could not be used, falling back to templates. Reason: {error}")
+        return generate_game_template(theme)
+
+
 # ------------------------------------------------------------
 # AI artwork generator
 # ------------------------------------------------------------
@@ -110,7 +207,8 @@ def get_font(size, bold=True):
 
 
 def make_placeholder_art(card):
-    """A simple backup image used only if the AI art request fails."""
+    """A simple backup image used if the AI art request fails, or if the
+    LLM invented a card type that has no matching artwork style."""
     colors = TYPE_COLORS.get(card["type"], {"bg": (100, 100, 100)})
     img = Image.new("RGB", (512, 512), colors["bg"])
     draw = ImageDraw.Draw(img)
@@ -162,7 +260,9 @@ def wrap_text(text, font, max_width, draw):
 
 
 def draw_card(card, art_image, theme):
-    """Composes the final card image: background, artwork, title bar, type badge, effect text."""
+    """Composes the final card image: background, artwork, title bar, type badge, effect text.
+    Card type may come from the LLM and not match a preset type; TYPE_COLORS.get()
+    below falls back to a neutral gray frame in that case."""
     width, height = 500, 700
     colors = TYPE_COLORS.get(card["type"], {"bg": (60, 60, 60), "border": (255, 255, 255)})
 
@@ -205,7 +305,7 @@ def draw_card(card, art_image, theme):
 # ------------------------------------------------------------
 
 def forge_full_game(theme):
-    """Full pipeline: theme -> game text -> AI artwork -> designed card images."""
+    """Full pipeline: theme -> LLM-generated game text -> AI artwork -> designed card images."""
     print(f"Generating game for theme: {theme}")
     game_data = generate_game(theme)
     print(f"Game created: {game_data['title']}")
@@ -259,7 +359,7 @@ with gr.Blocks(title="AI BoardGame Forge") as demo:
     gr.Markdown("# AI BoardGame Forge")
     gr.Markdown(
         "Enter a theme, or select one of the example themes below, "
-        "then click Generate Game to create a mini card game with AI-generated artwork."
+        "then click Generate Game to create a mini card game with AI-generated text and artwork."
     )
 
     theme_input = gr.Textbox(

requirements.txt

@@ -1,3 +1,6 @@
 gradio
 requests
 Pillow
+transformers
+torch
+accelerate