Update app.py

app.py

@@ -72,7 +72,7 @@ import os
 
 
 # Function to get bot response
-def format_prompt(state, bot_index, character_name, character_description, user_name):
+def format_prompt(state, bot_index, character_name, character_description, user_name, num_messages=5):
     if character_name is None or character_name.strip() == "":
         character_name = "Ryan"
     if character_description is None or character_description.strip() == "":
@@ -81,7 +81,11 @@ def format_prompt(state, bot_index, character_name, character_description, user_
         user_name = "You"
 
     prompt = f"{character_description}\n"
-    for message in state["history"][bot_index]:
+    
+    # Get the last num_messages messages from the conversation history
+    recent_messages = state["history"][bot_index][-num_messages:]
+    
+    for message in recent_messages:
         if message['role'] == 'user':
             prompt += f"{user_name}: {message['content']}\n"
         else: