Role-playing-game app iteration zero

app.py

@@ -1,148 +1,28 @@
 import gradio as gr
-import os
-demo = None #added to allow restart
+import 
+import json
+#from helper import (
+#    get_game_state,
+#    run_action,
+#    is_safe,
+#    detect_inventory_changes,
+#    update_inventory,
+#)
+
+# Global variable to allow demo restart
+demo = None
 
-# Create a Game UI with Gradio
-def start_game(main_loop, share=False):
-    # added code to support restart
-    global demo
-    # If demo is already running, close it first
-    if demo is not None:
-        demo.close()
-
-    demo = gr.ChatInterface(
-        main_loop,
-        chatbot=gr.Chatbot(height=250, placeholder="Type 'start game' to begin"),
-        textbox=gr.Textbox(placeholder="What do you do next?", container=False, scale=7),
-        title="AI RPG",
-        # description="Ask Yes Man any question",
-        theme="soft",
-        examples=["Look around", "Continue the story"],
-        cache_examples=False,
-        retry_btn="Retry",
-        undo_btn="Undo",
-        clear_btn="Clear",
-                           )
-    demo.launch(share=share, server_name="0.0.0.0")
-
-def test_main_loop(message, history):
-    return 'Entered Action: ' + message
-
-start_game(test_main_loop)
-
-from helper import load_world, save_world
-from together import Together
-from helper import get_together_api_key, load_env
-
-# Generating an Initial Start
-client = Together(api_key=get_together_api_key())
-
-world = load_world('../shared_data/Kyropeia.json')
-kingdom = world['kingdoms']['Eldrida']
-town = kingdom['towns']["Luminaria"]
-character = town['npcs']['Elwyn Stormbringer']
-
-system_prompt = """You are an AI Game master. Your job is to create a 
-start to an adventure based on the world, kingdom, town and character 
-a player is playing as. 
-Instructions:
-You must only use 2-4 sentences \
-Write in second person. For example: "You are Jack" \
-Write in present tense. For example "You stand at..." \
-First describe the character and their backstory. \
-Then describes where they start and what they see around them."""
-world_info = f"""
-World: {world}
-Kingdom: {kingdom}
-Town: {town}
-Your Character: {character}
-"""
-
-model_output = client.chat.completions.create(
-    model="meta-llama/Llama-3-70b-chat-hf",
-    temperature=1.0,
-    messages=[
-        {"role": "system", "content": system_prompt},
-        {"role": "user", "content": world_info + '\nYour Start:'}
-    ],
-)
-
-
-start = model_output.choices[0].message.content
-print(start)
-world['start'] = start
-#save_world(world, '../shared_data/Kyropeia.json')  # preserve video version
-save_world(world, '../shared_data/YourWorld_L1.json')
-
-# Creating the Main Action Loop
-def run_action(message, history, game_state):
-    
-    if(message == 'start game'):
-        return game_state['start']
-
-    system_prompt = """You are an AI Game master. Your job is to write what \
-happens next in a player's adventure game.\
-Instructions: \
-You must on only write 1-3 sentences in response. \
-Always write in second person present tense. \
-Ex. (You look north and see...)"""
-    
-    world_info = f"""
-World: {game_state['world']}
-Kingdom: {game_state['kingdom']}
-Town: {game_state['town']}
-Your Character:  {game_state['character']}"""
-
-    messages = [
-        {"role": "system", "content": system_prompt},
-        {"role": "user", "content": world_info}
-    ]
-    for action in history:
-        messages.append({"role": "assistant", "content": action[0]})
-        messages.append({"role": "user", "content": action[1]})
-
-    messages.append({"role": "user", "content": message})
-    model_output = client.chat.completions.create(
-        model="meta-llama/Llama-3-70b-chat-hf",
-        messages=messages
-    )
-    
-    result = model_output.choices[0].message.content
-    return result
-
-    game_state = {
-    "world": world['description'],
-    "kingdom": kingdom['description'],
-    "town": town['description'],
-    "character": character['description'],
-    "start": start,
+# Moderate Content Policy
+# Set the content policy
+mature_content_policy = {
+    "policy": """O2: Sexual Content.
+Should not
+- Write about graphic or illegal content.
+Can
+- Write about mature content between adults"""
 }
 
-def main_loop(message, history):
-    return run_action(message, history, game_state)
-
-# Launch and Share!
-start_game(main_loop, True)
-
-
-# Performing Safety Checks with the Default Content Policy
-from helper import get_game_state
-
-game_state = get_game_state()
-character = game_state["character"]
-print("Character Description:", character)
-
-
-safety_check = client.chat.completions.create(
-    model="Meta-Llama/LlamaGuard-2-8b",
-    messages=[{"role": "user", "content": character}],
-)
-
-print(f"prompt safety check: {safety_check.choices[0].message.content}")
-print(" ")
-
 # Safe Content Policy
-# Set the content policy
 safe_content_policy = {
     "policy": """O1: Violence and Hate.
 Should not
@@ -172,71 +52,42 @@ Should not
 - Write stories that depicts, glorifies or encourages drug use."""
 }
 
-heck_text = "Kill the troll."
-
-prompt = f"""[INST] Task: Check if there is unsafe content in 'user' messages in conversations according to our safety policy with the below categories.
-
-<BEGIN UNSAFE CONTENT CATEGORIES>
-{safe_content_policy['policy']}
-<END UNSAFE CONTENT CATEGORIES>
-
-<BEGIN CONVERSATION>
-user: {check_text}
-
-Provide your safety assessment for user in the above conversation:
-- First line must read 'safe' or 'unsafe'.
-- If unsafe, a second line must include a comma-separated list of violated categories. [/INST]"""
-
-print(prompt)
-
-response = client.completions.create(
-    model="Meta-Llama/LlamaGuard-2-8b",
-    prompt=prompt,
-)
-print(response.choices[0].text)
-
-# Moderate Content Policy
-# Set the content policy
-mature_content_policy = {
-    "policy": """O2: Sexual Content.
+everyone_content_policy = {
+    "policy": """O1: Violence and Hate.
 Should not
-- Write about graphic or illegal content.
+- Express hateful, derogatory, or demeaning sentiments against people based on sensitive personal characteristics.
+- Use slurs based on personal characteristics
+- Write gruesome / graphic violence.
 Can
-- Write about mature content between adults"""
-}
-
-check_text = "Kill the troll."
-
-prompt = f"""[INST] Task: Check if there is unsafe 
-content in 'user' messages in conversations according 
-to our safety policy with the below categories.
-
-<BEGIN UNSAFE CONTENT CATEGORIES>
-{mature_content_policy['policy']}
-<END UNSAFE CONTENT CATEGORIES>
-
-<BEGIN CONVERSATION>
-user: {check_text}
-
-Provide your safety assessment for user in the above conversation:
-- First line must read 'safe' or 'unsafe'.
-- If unsafe, a second line must include a comma-separated list of 
-violated categories. [/INST]"""
-
-response = client.completions.create(
-    model="Meta-Llama/LlamaGuard-2-8b",
-    prompt=prompt,
-)
-print(response.choices[0].text)
+- Write tame fantasy violence
+O2: Sexual Content.
+Should not
+- Write any somewhat sexual content that mentions sexual functions or sexual body parts.
+- Write any content that would be inappropriate for kids.
+Can
+- Write romantic content (including kissing, flirting etc...)
+O3: Self-Harm.
+Should not
+- Encourage people to harm themselves.
+- Romanticize or condone self-harm or suicide.
+- Write story depicting suicide.
+O4: Profanity.
+Should not
+- Write with any profane language that wouldn't be appropriate for kids.
+Can
+- Write with kid appropriate language
+O5: Regulated or Controlled Substances.
+Should not
+- Write stories that depicts, glorifies or encourages drug use."""
 
-# Define helper functions
-def is_safe (message):
-    prompt = f"""[INST] Task: Check if there is unsafe content in 
+prompt = f"""[INST] Task: Check if there is unsafe content in 
     'user' messages in conversations according to our safety policy 
     with the below categories.
     
     <BEGIN UNSAFE CONTENT CATEGORIES>
     {safe_content_policy['policy']}
+    {everyone_content_policy['policy']}
+    {mature_content_policy['policy']}
     <END UNSAFE CONTENT CATEGORIES>
     
     <BEGIN CONVERSATION>
@@ -246,35 +97,10 @@ def is_safe (message):
     - First line must read 'safe' or 'unsafe'.
     - If unsafe, a second line must include a comma-separated list of 
     violated categories. [/INST]"""
-
-    response = client.completions.create(
-        model="Meta-Llama/LlamaGuard-2-8b",
-        prompt=prompt,
-    )
-
-    result = response.choices[0].text
-    return result.strip() == 'safe'
-
-from helper import run_action, start_game, get_game_state
-
-game_state = get_game_state()
-
-def main_loop(message, history):
-
-    if not is_safe(message):
-        return 'Invalid action.'
     
-    result = run_action(message, history, game_state)
-    safe = is_safe(result)
-    if(safe):
-        return result # only if safe?
-    else:
-        return 'Invalid output.'
-
-start_game(main_loop, True)
 
 # Define Inventory Detector
-system_prompt = """You are an AI Game Assistant. \
+inventory_detector_system_prompt = """You are an AI Game Assistant. \
 Your job is to detect changes to a player's \
 inventory based on the most recent story and game state.
 If a player picks up, or gains an item add it to the inventory \
@@ -301,17 +127,103 @@ Inventory Updates:
 }
 """
 
+system_prompt = """You are an AI Game master. Your job is to create a 
+start to an adventure based on the world, kingdom, town and character 
+a player is playing as. 
+Instructions:
+You must only use 2-4 sentences \
+Write in second person. For example: "You are Jack" \
+Write in present tense. For example "You stand at..." \
+First describe the character and their backstory. \
+Then describes where they start and what they see around them."""
+world_info = f"""
+World: {world}
+Kingdom: {kingdom}
+Town: {town}
+Your Character: {character}
+"""
+
+system_prompt = """You are an AI Game master. Your job is to write what \
+happens next in a player's adventure game.\
+Instructions: \
+You must on only write 1-3 sentences in response. \
+Always write in second person present tense. \
+Ex. (You look north and see...) \
+Don't let the player use items they don't have in their inventory.
+"""
+
+# Initialize game state
+def initialize_game_state():
+    world = load_world('../shared_data/Kyropeia.json')
+    kingdom = world['kingdoms']['Eldrida']
+    town = kingdom['towns']["Luminaria"]
+    character = town['npcs']['Elwyn Stormbringer']
+
+    # Generate initial story
+    client = Together(api_key=get_together_api_key())
+    system_prompt = """You are an AI Game master. Your job is to create a 
+start to an adventure based on the world, kingdom, town and character 
+a player is playing as. 
+Instructions:
+You must only use 2-4 sentences
+Write in second person. For example: "You are Jack"
+Write in present tense. For example "You stand at..."
+First describe the character and their backstory.
+Then describe where they start and what they see around them."""
+    world_info = f"""
+World: {world}
+Kingdom: {kingdom}
+Town: {town}
+Your Character: {character}
+"""
+    model_output = client.chat.completions.create(
+        model="meta-llama/Llama-3-70b-chat-hf",
+        temperature=1.0,
+        messages=[
+            {"role": "system", "content": system_prompt},
+            {"role": "user", "content": world_info + '\nYour Start:'}
+        ],
+    )
+    start = model_output.choices[0].message.content
+    world['start'] = start
+    save_world(world, '../shared_data/YourWorld_L1.json')
+
+    # Return game state
+    return {
+        "world": world['description'],
+        "kingdom": kingdom['description'],
+        "town": town['description'],
+        "character": character['description'],
+        "start": start,
+        "inventory": {
+            "cloth pants": 1,
+            "cloth shirt": 1,
+            "goggles": 1,
+            "leather bound journal": 1,
+            "gold": 5
+        }
+    }
+
+def is_safe(message):
+    # Build the prompt with embedded values
+    client = Together(api_key=get_together_api_key())
+
+    response = client.completions.create(
+        model="Meta-Llama/LlamaGuard-2-8b",
+        prompt=prompt,
+    )
+    result = response.choices[0].text
+    return result.strip() == 'safe'
+
+
 def detect_inventory_changes(game_state, output):
-    
-    inventory = game_state['inventory']
     messages = [
-        {"role": "system", "content": system_prompt},
-        {"role": "user", "content": 
-         f'Current Inventory: {str(inventory)}'},
-        
+        {"role": "system", "content": inventory_detector_system_prompt},
+        {"role": "user", "content": f'Current Inventory: {json.dumps(game_state['inventory'])}'},
         {"role": "user", "content": f'Recent Story: {output}'},
         {"role": "user", "content": 'Inventory Updates'}
     ]
+
     chat_completion = client.chat.completions.create(
         # response_format={"type": "json_object", "schema": InventoryUpdate.model_json_schema()},
         model="meta-llama/Llama-3-70b-chat-hf",
@@ -319,22 +231,9 @@ def detect_inventory_changes(game_state, output):
         messages=messages
     )
     response = chat_completion.choices[0].message.content
-    result = json.loads(response)
-    return result['itemUpdates']
-
-    game_state = get_game_state()
-game_state['inventory'] = {
-    "cloth pants": 1,
-    "cloth shirt": 1,
-    "gold": 5
-}
-
-result = detect_inventory_changes(game_state, 
-"You buy a sword from the merchant for 5 gold")
-
-print(result)
-
+    return json.loads(response)['itemUpdates']    
 
+# Update inventory
 def update_inventory(inventory, item_updates):
     update_msg = ''
     
@@ -357,39 +256,31 @@ def update_inventory(inventory, item_updates):
             
     return update_msg
 
-# Now include inventory in the story
+# Main action loop (include inventory in the story)
 def run_action(message, history, game_state):
     
     if(message == 'start game'):
         return game_state['start']
-        
-    system_prompt = """You are an AI Game master. Your job is to write what \
-happens next in a player's adventure game.\
-Instructions: \
-You must on only write 1-3 sentences in response. \
-Always write in second person present tense. \
-Ex. (You look north and see...) \
-Don't let the player use items they don't have in their inventory.
-"""
 
     world_info = f"""
 World: {game_state['world']}
 Kingdom: {game_state['kingdom']}
 Town: {game_state['town']}
-Your Character:  {game_state['character']}
+Your Character: {game_state['character']}
 Inventory: {json.dumps(game_state['inventory'])}"""
-
+        
     messages = [
         {"role": "system", "content": system_prompt},
         {"role": "user", "content": world_info}
     ]
-
     for action in history:
         messages.append({"role": "assistant", "content": action[0]})
         messages.append({"role": "user", "content": action[1]})
-           
+
     messages.append({"role": "user", "content": message})
+
     client = Together(api_key=get_together_api_key())
+
     model_output = client.chat.completions.create(
         model="meta-llama/Llama-3-70b-chat-hf",
         messages=messages
@@ -398,30 +289,61 @@ Inventory: {json.dumps(game_state['inventory'])}"""
     result = model_output.choices[0].message.content
     return result
 
-# Integrate into the Game
-from helper import start_game, get_game_state, is_safe
-game_state = get_game_state(inventory={
-    "cloth pants": 1,
-    "cloth shirt": 1,
-    "goggles": 1,
-    "leather bound journal": 1,
-    "gold": 5
-})
-
+# Gradio main loop
 def main_loop(message, history):
-    output = run_action(message, history, game_state)
-    
-    safe = is_safe(output)
-    if not safe:
-        return 'Invalid Output'
-
-    item_updates = detect_inventory_changes(game_state, output)
-    update_msg = update_inventory(
-        game_state['inventory'], 
-        item_updates
+   """Main game loop that processes player actions"""
+    game_state = initialize_game_state()
+
+    # Check if input message is safe
+    if not is_safe(message):
+        return 'Invalid action.'
+
+    result = run_action(message, history, game_state)
+
+    # Check if output is safe
+    if not is_safe(result):
+        return 'Invalid output.'
+
+    # Detect and update inventory changes
+    try:
+        item_updates = detect_inventory_changes(game_state, result)
+        update_msg = update_inventory(game_state['inventory'], item_updates)
+        result += update_msg
+    except Exception as e:
+        # If inventory detection fails, continue without inventory updates
+        print(f"Inventory detection failed: {e}")
+        
+    return result
+
+# Start the Gradio UI Game
+def start_game(main_loop, share=False):
+    """Create and launch the Gradio interface for the RPG game"""
+    # added code to support restart
+    global demo
+    # If demo is already running, close it first
+    if demo is not None:
+        demo.close()
+        
+    demo = gr.ChatInterface(
+        main_loop,
+        chatbot=gr.Chatbot(height=250, placeholder="Type 'start game' to begin"),
+        textbox=gr.Textbox(placeholder="What do you do next?", container=False, scale=7),
+        title="AI RPG",
+        theme="soft",
+        examples=["Look around", "Continue the story"],
+        cache_examples=False,
+        retry_btn="Retry",
+        undo_btn="Undo",
+        clear_btn="Clear",
     )
-    output += update_msg
+    demo.launch(share=False, server_name="0.0.0.0")
 
-    return output
+# Launch the app if run directly
+if __name__ == "__main__":
+    # Launch the game
+    start_game(main_loop)
 
-start_game(main_loop, True)
+    #world = load_world('../shared_data/Kyropeia.json')
+    #kingdom = world['kingdoms']['Eldrida']
+    #town = kingdom['towns']["Luminaria"]
+    #character = town['npcs']['Elwyn Stormbringer']