Add main.py

2026-05-26 08:47:46 +00:00
commit b6cac852f4
1 changed files with 147 additions and 0 deletions
@@ -0,0 +1,147 @@
+"""
+Text-based adventure game powered by OpenAI GPT-3.5.
+
+Usage:
+    python main.py
+
+The game accepts player commands and uses the LLM to generate the next scene or dialogue.
+
+Interrupt feature:
+    Type "pause" to pause the game. The current state will be displayed and you can resume by typing "resume".
+    Type "exit" to quit the game.
+
+Make sure to set the environment variable OPENAI_API_KEY before running.
+"""
+
+import os
+import json
+import sys
+from pathlib import Path
+
+import openai
+
+# Constants
+SYSTEM_PROMPT = (
+    "You are a text-based adventure game narrator. "
+    "Respond to the player's commands with vivid descriptions, dialogue, and actions. "
+    "Maintain continuity and immerse the player in the story. "
+)
+STATE_FILE = Path("game_state.json")
+
+
+class AdventureGame:
+    def __init__(self, model="gpt-3.5-turbo"):
+        self.model = model
+        self.history = []  # list of dicts with role and content
+        self.last_user_input = ""
+        self.load_state()
+
+    def load_state(self):
+        if STATE_FILE.exists():
+            try:
+                data = json.loads(STATE_FILE.read_text())
+                self.history = data.get("history", [])
+                self.last_user_input = data.get("last_user_input", "")
+                print("[Game] Loaded previous state. Type 'resume' to continue.")
+            except Exception as e:
+                print(f"[Game] Failed to load state: {e}")
+
+    def save_state(self):
+        data = {
+            "history": self.history,
+            "last_user_input": self.last_user_input,
+        }
+        STATE_FILE.write_text(json.dumps(data))
+
+    def clear_state(self):
+        if STATE_FILE.exists():
+            STATE_FILE.unlink()
+
+    def generate(self, user_input: str) -> str:
+        self.last_user_input = user_input
+        # Build messages for chat completion
+        messages = [
+            {"role": "system", "content": SYSTEM_PROMPT},
+        ] + self.history + [
+            {"role": "user", "content": user_input},
+        ]
+        try:
+            response = openai.ChatCompletion.create(
+                model=self.model,
+                messages=messages,
+                temperature=0.7,
+                max_tokens=512,
+            )
+            content = response.choices[0].message.content.strip()
+            # Append to history
+            self.history.append({"role": "user", "content": user_input})
+            self.history.append({"role": "assistant", "content": content})
+            self.save_state()
+            return content
+        except Exception as e:
+            return f"[Error] Failed to generate response: {e}"
+
+    def show_state(self):
+        print("\n--- Current Game State ---")
+        if not self.history:
+            print("No history yet.")
+        else:
+            for msg in self.history:
+                role = msg["role"]
+                print(f"{role.capitalize()}: {msg['content']}\n")
+        print("--- End of State ---\n")
+
+    def run(self):
+        print("Welcome to the LLM Adventure Game! Type 'help' for commands.")
+        while True:
+            try:
+                user_input = input("> ").strip()
+            except (EOFError, KeyboardInterrupt):
+                print("\nExiting game.")
+                self.clear_state()
+                sys.exit(0)
+
+            if not user_input:
+                continue
+
+            if user_input.lower() in ("exit", "quit"):
+                print("Goodbye!")
+                self.clear_state()
+                break
+            if user_input.lower() == "help":
+                print("Commands: pause, resume, help, exit")
+                continue
+            if user_input.lower() == "pause":
+                print("Game paused. Type 'resume' to continue or 'exit' to quit.")
+                self.show_state()
+                while True:
+                    try:
+                        cmd = input("(paused)> ").strip().lower()
+                    except (EOFError, KeyboardInterrupt):
+                        print("\nExiting game.")
+                        self.clear_state()
+                        sys.exit(0)
+                    if cmd == "resume":
+                        print("Resuming game...")
+                        break
+                    if cmd == "exit":
+                        print("Goodbye!")
+                        self.clear_state()
+                        sys.exit(0)
+                    if cmd == "help":
+                        print("Commands while paused: resume, exit, help")
+                        continue
+                    print("Unknown command while paused. Type 'resume' or 'exit'.")
+                continue
+            # Regular input
+            response = self.generate(user_input)
+            print(response)
+
+
+if __name__ == "__main__":
+    if not os.getenv("OPENAI_API_KEY"):
+        print("Error: OPENAI_API_KEY environment variable not set.")
+        sys.exit(1)
+    openai.api_key = os.getenv("OPENAI_API_KEY")
+    game = AdventureGame()
+    game.run()