Create database.py

database.py

@@ -0,0 +1,47 @@
+import pixeltable as pxt
+from pixeltable.functions import openai
+import os
+import getpass
+
+# Set up OpenAI API key
+if 'OPENAI_API_KEY' not in os.environ:
+    os.environ['OPENAI_API_KEY'] = getpass.getpass('Enter your OpenAI API key: ')
+
+# Initialize Pixeltable
+pxt.drop_dir('ai_rpg', force=True)
+pxt.create_dir('ai_rpg')
+
+# Create a single table for all game data
+interactions = pxt.create_table(
+    'ai_rpg.interactions',
+    {
+        'session_id': pxt.StringType(),
+        'player_name': pxt.StringType(),
+        'genre': pxt.StringType(),
+        'initial_scenario': pxt.StringType(),
+        'turn_number': pxt.IntType(),
+        'player_input': pxt.StringType(),
+        'timestamp': pxt.TimestampType(),
+    }
+)
+
+# Add computed columns for AI responses
+from src.utils import generate_messages, extract_options
+
+interactions['messages'] = generate_messages(
+    interactions.genre,
+    interactions.player_name,
+    interactions.initial_scenario,
+    interactions.player_input,
+    interactions.turn_number
+)
+
+interactions['ai_response'] = openai.chat_completions(
+    messages=interactions.messages,
+    model='gpt-4o-mini-2024-07-18',
+    max_tokens=500,
+    temperature=0.8
+)
+
+interactions['story_text'] = interactions.ai_response.choices[0].message.content
+interactions['options'] = extract_options(interactions.story_text)