Update app.py

app.py

@@ -48,7 +48,7 @@ def randomly_select_from_json(world_data):
         "character": {"name": character_name, "description": character_description},
     }
 
-# Initialize random state
+# Function to initialize or reinitialize the game state
 def initialize_game_state():
     random_state = randomly_select_from_json(world_data)
     world = random_state["world"]
@@ -93,13 +93,20 @@ def initialize_game_state():
         "start": start,
     }
 
-# Game state
+# Initialize the game state
 game_state = initialize_game_state()
 
-def run_action(message, history, game_state):
+# Function to process user input and actions
+def run_action(message, history):
+    global game_state  # Access the global game state
+
     if message.lower() == "start game":
         return game_state["start"]
 
+    if message.lower() == "start again":
+        game_state = initialize_game_state()
+        return "Game restarted! " + 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: \
@@ -114,6 +121,7 @@ def run_action(message, history, game_state):
     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},
@@ -124,8 +132,10 @@ def run_action(message, history, game_state):
             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,
@@ -133,27 +143,25 @@ def run_action(message, history, game_state):
 
     return model_output.choices[0].message.content
 
-# Main Gradio app
-with gr.Blocks() as demo:
-    gr.Markdown("## AI RPG Adventure Game")
-    chatbot = gr.Chatbot(type="messages")
-    msg_input = gr.Textbox(
-        placeholder="Type 'start game' to begin or enter actions...",
-        label="Your Action",
-    )
-    start_again = gr.Button("Start Again")
-
-    def handle_message(message, history):
-        response = run_action(message, history, game_state)
-        return history + [(message, response)]
-
-    def restart_game():
-        global game_state
-        game_state = initialize_game_state()
-        return [("Game restarted!", game_state["start"])]
-
-    msg_input.submit(handle_message, [msg_input, chatbot], chatbot)
-    start_again.click(restart_game, inputs=[], outputs=chatbot)
+# Gradio ChatInterface with the custom "start again" logic
+demo = gr.ChatInterface(
+    run_action,
+    chatbot=gr.Chatbot(
+        height=250,
+        placeholder="Type 'start game' to begin or 'start again' to restart the adventure",
+        type="messages",
+    ),
+    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,
+)
 
 # Launch the game
 demo.launch(share=True, server_name="0.0.0.0")
+