Spaces:
Sleeping
Sleeping
| """ | |
| scripts/demo.py | |
| --------------- | |
| Interactive demonstration for the board game RAG system. | |
| Run (from project root): | |
| .venv/bin/python scripts/demo.py catan | |
| .venv/bin/python scripts/demo.py monopoly | |
| Press Ctrl-C to exit interactive mode. | |
| """ | |
| import os | |
| import sys | |
| sys.path.insert(0, os.path.dirname(os.path.abspath(__file__))) | |
| from model import query_rag | |
| DIVIDER = "=" * 70 | |
| DEMO_QUESTIONS: dict[str, list[dict]] = { | |
| "catan": [ | |
| { | |
| "label": "✅ Easy win — direct factual lookup", | |
| "question": "What happens when you roll a 7?", | |
| }, | |
| { | |
| "label": "✅ Multi-section synthesis — retrieves correctly across chunks", | |
| "question": ( | |
| "My road is 6 segments long and holds the Longest Road card. " | |
| "An opponent builds a settlement in the middle, breaking it into " | |
| "branches of 4 and 2. Do I lose the Longest Road card?" | |
| ), | |
| }, | |
| { | |
| "label": "⚠️ Generation failure — answer IS in context, model still abstains", | |
| "question": "Can players trade resource cards with each other when it is not their turn?", | |
| }, | |
| { | |
| "label": "✅ Hallucination resistance — correctly says rules don't specify", | |
| "question": "Can a player make a binding promise to give resources on a future turn?", | |
| }, | |
| ], | |
| "monopoly": [ | |
| { | |
| "label": "✅ Easy win — direct factual lookup", | |
| "question": "How much money does each player start with?", | |
| }, | |
| { | |
| "label": "✅ Multi-section synthesis — jail + property rules", | |
| "question": "I am in jail. Can I still buy houses and collect rent on my properties?", | |
| }, | |
| { | |
| "label": "⚠️ Common misconception — Free Parking payout", | |
| "question": "What do you receive when you land on Free Parking?", | |
| }, | |
| { | |
| "label": "✅ Hallucination resistance — rules are silent on tie-breaking", | |
| "question": "What happens if two players tie on the initial roll to see who goes first?", | |
| }, | |
| ], | |
| } | |
| def show_question(label: str, question: str, game: str): | |
| print(f"\n{DIVIDER}") | |
| print(f" {label}") | |
| print(DIVIDER) | |
| print(f"\nQ: {question}\n") | |
| result = query_rag(question, game=game, k=3) | |
| print("Retrieved chunks:") | |
| for c in result["retrieved"]: | |
| print(f" [{c['score']:.3f}] {c['title']}") | |
| print(f"\nAnswer:\n{result['answer']}\n") | |
| def main(): | |
| import argparse | |
| parser = argparse.ArgumentParser(description="Interactive RAG demo for a board game.") | |
| parser.add_argument("game", choices=list(DEMO_QUESTIONS.keys()), help="Game to demo") | |
| args = parser.parse_args() | |
| game = args.game | |
| demos = DEMO_QUESTIONS[game] | |
| print(DIVIDER) | |
| print(f" {game.upper()} RAG SYSTEM — Demo") | |
| print(DIVIDER) | |
| print("Loading model and index (one-time)...") | |
| show_question(demos[0]["label"], demos[0]["question"], game) | |
| input("\n[Press Enter to continue to next demo question...]") | |
| for demo in demos[1:]: | |
| show_question(demo["label"], demo["question"], game) | |
| input("[Press Enter to continue...]") | |
| print(DIVIDER) | |
| print(f" INTERACTIVE MODE — ask your own {game.title()} question") | |
| print(" (type 'quit' to exit)") | |
| print(DIVIDER) | |
| while True: | |
| try: | |
| q = input("\nYour question: ").strip() | |
| except (KeyboardInterrupt, EOFError): | |
| print("\nDone.") | |
| break | |
| if not q or q.lower() in {"quit", "exit", "q"}: | |
| print("Done.") | |
| break | |
| result = query_rag(q, game=game, k=3) | |
| print("\nRetrieved chunks:") | |
| for c in result["retrieved"]: | |
| print(f" [{c['score']:.3f}] {c['title']}") | |
| print(f"\nAnswer:\n{result['answer']}") | |
| if __name__ == "__main__": | |
| main() | |