Update app.py

app.py

@@ -26,9 +26,9 @@ def regenerate(chatbot, chat_history_state, system_msg, max_tokens, temperature,
     if not chat_history_state:
         return chatbot, chat_history_state, ""
     
-    # Remove the last item from the chat history
+    # Remove only the last assistant's message from the chat history
     if len(chat_history_state) > 0:
-        chat_history_state.pop(-1)
+        chat_history_state[-1][1] = ""
     
     # Re-run the chat function
     new_history, _, _ = chat(chat_history_state, system_msg, max_tokens, temperature, top_p, top_k, repetition_penalty)
@@ -36,6 +36,7 @@ def regenerate(chatbot, chat_history_state, system_msg, max_tokens, temperature,
     return new_history, new_history, ""
 
 
+
 def chat(history, system_message, max_tokens, temperature, top_p, top_k, repetition_penalty):
     history = history or []