Update app.py

app.py

@@ -95,18 +95,26 @@ def initialize_game_state():
 
 # Initialize the game state
 game_state = initialize_game_state()
+game_running = True  # Flag to manage game status
 
 # Function to process user input and actions
 def run_action(message, history):
-    global game_state  # Access the global game state
+    global game_state, game_running  # Access the global game state and game status
+
+    if not game_running:
+        return "The game has ended. Type 'restart the game' to play again."
 
     if message.lower() == "start game":
         return game_state["start"]
 
-    if message.lower() == "start again":
+    if message.lower() == "restart the game":
         game_state = initialize_game_state()
         return "Game restarted! " + game_state["start"]
 
+    if message.lower() == "exit":
+        game_running = False
+        return "The game has ended. Type 'restart the game' to play again."
+
     system_prompt = """You are an AI Game master. Your job is to write what \
     happens next in a player's adventure game.\
     Instructions: \
@@ -143,12 +151,15 @@ def run_action(message, history):
 
     return model_output.choices[0].message.content
 
-# Gradio ChatInterface with the custom "start again" logic
+
+def main_loop(message, history):
+    return run_action(message, history)
+
 demo = gr.ChatInterface(
-    run_action,
+    main_loop,
     chatbot=gr.Chatbot(
         height=250,
-        placeholder="Type 'start game' to begin or 'start again' to restart the adventure",
+        placeholder="Type 'start game' to begin, 'restart the game' to restart, or 'exit' to end the game.",
         type="messages",
     ),
     textbox=gr.Textbox(
@@ -164,4 +175,3 @@ demo = gr.ChatInterface(
 
 # Launch the game
 demo.launch(share=True, server_name="0.0.0.0")
-