Update app.py

app.py

@@ -51,7 +51,7 @@ game_state = {
     "start": start,
 }
 
-def run_action(message, history, game_state):
+def run_action12(message, history, game_state):
     
     if(message == 'start game'):
         return game_state['start']
@@ -86,6 +86,50 @@ Your Character:  {game_state['character']}"""
     result = model_output.choices[0].message.content
     return result
 
+def run_action(message, history, game_state):
+    if message.lower() == '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: \
+- Write only 1-3 sentences. \
+- Always write in second person, e.g., "You look north and see..." \
+- Write in present tense."""
+
+    world_info = f"""
+World: {game_state['world']}
+Kingdom: {game_state['kingdom']}
+Town: {game_state['town']}
+Your Character: {game_state['character']}"""
+
+    # Build the conversation context
+    messages = [
+        {"role": "system", "content": system_prompt},
+        {"role": "user", "content": world_info}
+    ]
+
+    for action in history:
+        # Ensure action is in the correct format before appending
+        if isinstance(action, tuple) and len(action) == 2:
+            messages.append({"role": "assistant", "content": action[0]})
+            messages.append({"role": "user", "content": action[1]})
+        else:
+            # If action is not in the correct format, log a warning or handle the case
+            print(f"Warning: Invalid action format: {action}")
+
+    # Add the user's current message
+    messages.append({"role": "user", "content": message})
+
+    # Get the model's response
+    model_output = client.chat.completions.create(
+        model="meta-llama/Llama-3-70b-chat-hf",
+        messages=messages
+    )
+    
+    return model_output.choices[0].message.content
+
+
 def main_loop(message, history):
     return run_action(message, history, game_state)