Update app.py

app.py

@@ -66,28 +66,31 @@ game_running = False  # Flag to manage game status
 
 # Function to process user input and actions
 def run_action(message, history):
-    global game_state, game_running  # Access the global game state and game status
+    global game_state, game_running  # Access global variables
 
-    if not game_running:
-        return "The game has ended. Type 'restart the game' to play again."
+    if message.lower() == "restart the game":
+        game_state = initialize_game_state()
+        game_running = True  # Set the game as running
+        return "Game restarted! " + game_state["start"]
 
     if message.lower() == "start game":
+        game_running = True  # Set the game as running
         return game_state["start"]
 
-    if message.lower() == "restart the game":
-        game_state = initialize_game_state()
-        return "Game restarted! " + game_state["start"]
+    if not game_running:
+        return "The game has ended. Type 'restart the game' to play again."
 
     if message.lower() == "exit":
         game_running = False
         return "The game has ended. Type 'restart the game' to play again."
 
+    # The rest of the function remains the same
     system_prompt = """You are an AI Game master. Your job is to write what \
-    happens next in a player's adventure game.\
+    happens next in a player's adventure game. \
     Instructions: \
     - Write only 1-3 sentences. \
-    - Please use simple and clear language that is easy for children to understand.
-    - Always write in second person, e.g., "You look north and see..." \
+    - Please use simple and clear language that is easy for children to understand. \
+    - Always write in second person, e.g., 'You look north and see...' \
     - Write in present tense."""
 
     world_info = f"""
@@ -96,7 +99,6 @@ def run_action(message, history):
     Town: {game_state['town']}
     Your Character: {game_state['character']}"""
 
-    # Build the context for the conversation
     messages = [
         {"role": "system", "content": system_prompt},
         {"role": "user", "content": world_info},
@@ -107,10 +109,8 @@ def run_action(message, history):
             messages.append({"role": "assistant", "content": action[0]})
             messages.append({"role": "user", "content": action[1]})
 
-    # Add the user's current action
     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,