Update app.py

app.py

@@ -156,6 +156,8 @@ def main_loop1(message, history):
     return run_action(message, history)
 
 
+# Function to process user input and actions
+
 # 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
@@ -166,24 +168,28 @@ def run_action(message, history):
     # Handle exit and restart logic
     if not game_running:
         response = "The game has ended. Type 'restart the game' to play again."
-        history.append((message, response))
+        history.append({"role": "user", "content": message})
+        history.append({"role": "assistant", "content": response})
         return history
 
     if message.lower() == "start game":
         response = game_state["start"]
-        history.append((message, response))
+        history.append({"role": "user", "content": message})
+        history.append({"role": "assistant", "content": response})
         return history
 
     if message.lower() == "restart the game":
         game_state = initialize_game_state()
         response = "Game restarted! " + game_state["start"]
-        history.append((message, response))
+        history.append({"role": "user", "content": message})
+        history.append({"role": "assistant", "content": response})
         return history
 
     if message.lower() == "exit":
         game_running = False
         response = "The game has ended. Type 'restart the game' to play again."
-        history.append((message, response))
+        history.append({"role": "user", "content": message})
+        history.append({"role": "assistant", "content": response})
         return history
 
     # Prepare the system prompt for the AI model
@@ -201,12 +207,13 @@ def run_action(message, history):
     Town: {game_state['town']}
     Your Character: {game_state['character']}"""
 
-    # Convert history to messages for the AI model
+    # Prepare the messages for the AI model
     messages = [
         {"role": "system", "content": system_prompt},
         {"role": "user", "content": world_info},
     ]
 
+    # Convert Gradio's history format (list of tuples) to the required AI model format
     for user_input, assistant_response in history:
         messages.append({"role": "user", "content": user_input})
         messages.append({"role": "assistant", "content": assistant_response})
@@ -214,7 +221,7 @@ def run_action(message, history):
     # Add the user's current action
     messages.append({"role": "user", "content": message})
 
-    # Get the model's response
+    # Call the AI model to get the response
     model_output = client.chat.completions.create(
         model="meta-llama/Llama-3-70b-chat-hf",
         messages=messages,
@@ -222,16 +229,45 @@ def run_action(message, history):
 
     response = model_output.choices[0].message.content
 
-    # Append the user's input and assistant's response to history
-    history.append((message, response))
+    # Append the user input and the AI response to history
+    history.append({"role": "user", "content": message})
+    history.append({"role": "assistant", "content": response})
 
     return history
 
-# Main loop to process the conversation
+
+# Main loop to process user input and maintain history
 def main_loop(message, history):
-    history = run_action(message, history)
-    return history, history
+    history = run_action(message, history or [])
+    display_history = [
+        (item["content"], history[i + 1]["content"])
+        for i, item in enumerate(history[:-1])
+        if item["role"] == "user"
+    ]
+    return display_history, history
+
 
+# Gradio ChatInterface
+demo = gr.ChatInterface(
+    main_loop,
+    chatbot=gr.Chatbot(
+        height=300,
+        placeholder="Type 'start game' to begin, 'restart the game' to restart, or 'exit' to end the game.",
+        type="messages",  # Ensures proper rendering
+    ),
+    textbox=gr.Textbox(
+        placeholder="What do you do next?",
+        container=False,
+        scale=7,
+    ),
+    title="AI RPG",
+    theme="Monochrome",
+    examples=["Look around", "Continue the story"],
+    cache_examples=False,
+)
+
+# Launch the Gradio app
+demo.launch(share=True, server_name="0.0.0.0")