Update app.py

app.py

@@ -22,13 +22,17 @@ def user(message, history):
     history.append([message, ""])
     return "", history
 
-def regenerate(history, system_msg, max_tokens, temperature, top_p, top_k, repetition_penalty):
-    # Remove the last item from the history
-    if history:
-        del history[-1]
+def regenerate(chatbot, chat_history_state, system_msg, max_tokens, temperature, top_p, top_k, repetition_penalty):
+    if not chat_history_state:
+        return chatbot, chat_history_state, ""
+    
+    # Remove the last item from the chat history
+    if len(chat_history_state) > 0:
+        chat_history_state.pop(-1)
+    
+    # Re-run the chat function
+    new_history, _, _ = chat(chat_history_state, system_msg, max_tokens, temperature, top_p, top_k, repetition_penalty)
     
-    # Re-execute the chat function
-    new_history, _, _ = chat(history, system_msg, max_tokens, temperature, top_p, top_k, repetition_penalty)
     return new_history, new_history, ""
 
 
@@ -129,12 +133,11 @@ with gr.Blocks() as demo:
     stop.click(fn=None, inputs=None, outputs=None, cancels=[submit_click_event], queue=False)
     regen_click_event = regen_btn.click(
         fn=regenerate, 
-        inputs=[chat_history_state, system_msg, max_tokens, temperature, top_p, top_k, repetition_penalty], 
+        inputs=[chatbot, chat_history_state, system_msg, max_tokens, temperature, top_p, top_k, repetition_penalty], 
         outputs=[chatbot, chat_history_state, message], 
         queue=True
     )
 
 
-
 demo.queue(max_size=128, concurrency_count=2)
 demo.launch()