Update app.py

app.py

@@ -22,21 +22,13 @@ def user(message, history):
     history.append([message, ""])
     return "", history
 
-def regenerate(_chatbot, _task_history, system_msg, max_tokens, temperature, top_p, top_k, repetition_penalty):
-    print("Regenerate called")  # Debugging line
-    if not _task_history:
-        yield _chatbot
-        return
-    last_message = _task_history.pop(-1)
-    _chatbot.pop(-1)
-    _task_history.append([last_message[0], ""])
-    new_history, _, _ = chat(_task_history, system_msg, max_tokens, temperature, top_p, top_k, repetition_penalty)
-    print(f"New history: {new_history}")  # Debugging line
-    yield new_history
-
-
-
-
+def regenerate(history, system_msg, max_tokens, temperature, top_p, top_k, repetition_penalty):
+    # Remove the last item from the history
+    if history:
+        history.pop(-1)
+    
+    # Re-execute the chat function
+    return chat(history, system_msg, max_tokens, temperature, top_p, top_k, repetition_penalty)
 
 def chat(history, system_message, max_tokens, temperature, top_p, top_k, repetition_penalty):
     history = history or []
@@ -133,9 +125,11 @@ with gr.Blocks() as demo:
     
     # Stop button remains the same
     stop.click(fn=None, inputs=None, outputs=None, cancels=[submit_click_event], queue=False)
-
-    regen_btn.click(
-        fn=regenerate, inputs=[chatbot, chat_history_state, system_msg, max_tokens, temperature, top_p, top_k, repetition_penalty], outputs=[chatbot], queue=True
+    regen_click_event = regen_btn.click(
+        fn=regenerate, 
+        inputs=[chat_history_state, system_msg, max_tokens, temperature, top_p, top_k, repetition_penalty], 
+        outputs=[chatbot, chat_history_state, message], 
+        queue=True
     )