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, num_messages=5):
+def format_prompt(state, bot_index, character_name, character_description, user_name, num_messages=20):
     if character_name is None or character_name.strip() == "":
         character_name = "Ryan"
     if character_description is None or character_description.strip() == "":
@@ -190,8 +190,9 @@ async def user_ask(state, chatbot1, chatbot2, textbox, character_name, character
         {"role": "user", "content": user_input}])
     state["history"][1].extend([
         {"role": "user", "content": user_input}])
-    if len(state["history"])>20:
-        state["history"] = state["history"][-20:]
+    if len(state["history"][0])>20:
+        state["history"][0] = state["history"][0][-20:]
+        state["history"][1] = state["history"][1][-20:]
     # Chat with bots
     bot1_response, bot2_response = await chat_with_bots(user_input, state, character_name, character_description, user_name)
 
@@ -242,17 +243,27 @@ def generate_leaderboard(collection):
 def refresh_leaderboard():
     collection = init_database()
     return generate_leaderboard(collection)
-async def direct_chat(model, user_input, chatbot, character_name, character_description, user_name):
+async def direct_chat(model, user_input, state, chatbot, character_name, character_description, user_name):
     adapter = next(entry['adapter'] for entry in chatbots_data if entry['original_model'] == model)
+
+    if "direct_history" not in state:
+        state["direct_history"] = []
+
+    state["direct_history"].append({"role": "user", "content": user_input})
+
     temp_state = {
         "history": [
-            [{"role": "user", "content": user_input}],
-            [{"role": "user", "content": user_input}]
+            state["direct_history"],
+            state["direct_history"]
         ]
     }
+
     response = await get_bot_response(adapter, user_input, temp_state, 0, character_name, character_description, user_name)
     chatbot.append((user_input, response))
-    return "", chatbot
+
+    state["direct_history"].append({"role": "bot", "content": response})
+
+    return "", chatbot, state
 def reset_direct_chat():
     return [], gr.Textbox.update(value='')
 refresh_leaderboard()
@@ -316,8 +327,8 @@ with gr.Blocks() as demo:
         # ...
 
         direct_regenerate_btn.click(direct_regenerate, inputs=[model_dropdown, direct_textbox, direct_chatbot, character_name, character_description, user_name], outputs=[direct_textbox, direct_chatbot])
-        direct_textbox.submit(direct_chat, inputs=[model_dropdown, direct_textbox, direct_chatbot, character_name, character_description, user_name], outputs=[direct_textbox, direct_chatbot])
-        direct_submit_btn.click(direct_chat, inputs=[model_dropdown, direct_textbox, direct_chatbot, character_name, character_description, user_name], outputs=[direct_textbox, direct_chatbot])
+        direct_textbox.submit(direct_chat, inputs=[model_dropdown, direct_textbox, state, direct_chatbot, character_name, character_description, user_name], outputs=[direct_textbox, direct_chatbot, state])
+        direct_submit_btn.click(direct_chat, inputs=[model_dropdown, direct_textbox, state, direct_chatbot, character_name, character_description, user_name], outputs=[direct_textbox, direct_chatbot, state])
         direct_reset_btn.click(reset_direct_chat, None, [direct_chatbot, direct_textbox])
     with gr.Tab("🏆 Leaderboard"):
         gr.Markdown("## 📊 Check out the top-performing models!")