Update app.py

app.py

@@ -13,14 +13,14 @@ import sqlite3
 
 from pymongo.mongo_client import MongoClient
 from pymongo.server_api import ServerApi
-async def direct_regenerate(model, user_input, chatbot):
+async def direct_regenerate(model, user_input, chatbot, character_name, character_description, user_name):
     adapter = next(entry['adapter'] for entry in chatbots_data if entry['original_model'] == model)
     temp_state = {
         "history": [
             [{"role": "user", "content": chatbot[-1][0]}]  # Keep the user's last message
         ]
     }
-    response = await get_bot_response(adapter, user_input, temp_state, 0)
+    response = await get_bot_response(adapter, user_input, temp_state, 0, character_name, character_description, user_name)
     chatbot[-1] = (chatbot[-1][0], response)  # Update only the assistant's response
     return "", chatbot
 password=os.environ.get("MONGODB")
@@ -242,7 +242,7 @@ def generate_leaderboard(collection):
 def refresh_leaderboard():
     collection = init_database()
     return generate_leaderboard(collection)
-async def direct_chat(model, user_input, chatbot):
+async def direct_chat(model, user_input, chatbot, character_name, character_description, user_name):
     adapter = next(entry['adapter'] for entry in chatbots_data if entry['original_model'] == model)
     temp_state = {
         "history": [
@@ -250,7 +250,7 @@ async def direct_chat(model, user_input, chatbot):
             [{"role": "user", "content": user_input}]
         ]
     }
-    response = await get_bot_response(adapter, user_input, temp_state, 0)
+    response = await get_bot_response(adapter, user_input, temp_state, 0, character_name, character_description, user_name)
     chatbot.append((user_input, response))
     return "", chatbot
 def reset_direct_chat():
@@ -315,9 +315,8 @@ with gr.Blocks() as demo:
 
         # ...
 
-        direct_regenerate_btn.click(direct_regenerate, inputs=[model_dropdown, direct_textbox, direct_chatbot], outputs=[direct_textbox, direct_chatbot])
-        direct_textbox.submit(direct_chat, inputs=[model_dropdown, direct_textbox, direct_chatbot], outputs=[direct_textbox, direct_chatbot])
-        direct_submit_btn.click(direct_chat, inputs=[model_dropdown, direct_textbox, direct_chatbot], outputs=[direct_textbox, direct_chatbot])
+        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_reset_btn.click(reset_direct_chat, None, [direct_textbox, direct_chatbot, model_dropdown])