Update app.py

app.py

@@ -13,7 +13,25 @@ import sqlite3
 
 from pymongo.mongo_client import MongoClient
 from pymongo.server_api import ServerApi
+async def direct_regenerate(model, user_input, chatbot):
+    temp_state = {
+        "history": [
+            [{"role": "user", "content": user_input}],
+            [{"role": "user", "content": user_input}]
+        ]
+    }
+    response = await get_bot_response(model, user_input, temp_state, 0)
+    chatbot[-1] = (user_input, response)
+    return "", chatbot
 
+async def regenerate_responses(state, chatbot1, chatbot2):
+    user_input = state["history"][0][-1]["content"]
+    bot1_response = await get_bot_response(state['last_bots'][0], user_input, state, 0)
+    bot2_response = await get_bot_response(state['last_bots'][1], user_input, state, 1)
+    chatbot1[-1] = (user_input, bot1_response)
+    chatbot2[-1] = (user_input, bot2_response)
+    return chatbot1, chatbot2
+    
 password=os.environ.get("MONGODB")
 def reset_database():
     uri = f"mongodb+srv://new-user:{password}@cluster0.xb2urf6.mongodb.net/?retryWrites=true&w=majority&appName=Cluster0"
@@ -236,6 +254,13 @@ with gr.Blocks() as demo:
         with gr.Row():
             submit_btn = gr.Button(value="Send")
             reset_btn = gr.Button(value="Reset")
+        with gr.Row():
+            submit_btn = gr.Button(value="Send")
+            reset_btn = gr.Button(value="Reset")
+            regenerate_btn = gr.Button(value="Regenerate")
+        # ...
+        regenerate_btn.click(regenerate_responses, inputs=[state, chatbot1, chatbot2], outputs=[chatbot1, chatbot2])
+    
         reset_btn.click(clear_chat, inputs=[state], outputs=[state, chatbot1, chatbot2, upvote_btn_a, upvote_btn_b, textbox, submit_btn])
         textbox.submit(user_ask, inputs=[state, chatbot1, chatbot2, textbox], outputs=[state, chatbot1, chatbot2, textbox, upvote_btn_a, upvote_btn_b], queue=True)
         submit_btn.click(user_ask, inputs=[state, chatbot1, chatbot2, textbox], outputs=[state, chatbot1, chatbot2, textbox, upvote_btn_a, upvote_btn_b], queue=True)
@@ -249,7 +274,10 @@ with gr.Blocks() as demo:
         direct_textbox = gr.Textbox(placeholder="Enter your message")
         direct_submit_btn = gr.Button(value="Send")
         direct_reset_btn = gr.Button(value="Reset Chat")  # Add this line
-        
+        direct_regenerate_btn = gr.Button(value="Regenerate")
+        # ...
+        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_reset_btn.click(reset_direct_chat, None, [direct_textbox, direct_chatbot, model_dropdown])