Update app_dialogue.py

app_dialogue.py

@@ -672,14 +672,22 @@ with gr.Blocks(title="IDEFICS Playground", theme=gr.themes.Base()) as demo:
         ],
     )
 
-    def remove_last_turn(chat_history):
+    def remove_last_turn_OLD(chat_history):
         if len(chat_history) == 0:
-            return gr.Update(), gr.Update()
+            return gr.Update(), gr.update()
         last_interaction = chat_history[-1]
         chat_history = chat_history[:-1]
         chat_update = gr.update(value=chat_history)
         text_update = gr.update(value=last_interaction[0])
         return chat_update, text_update
+    def remove_last_turn(chat_history):
+        if len(chat_history) == 0:
+            return None, None
+        last_interaction = chat_history[-1]
+        chat_history = chat_history[:-1]
+        chat_update = gr.update(value=chat_history)
+        text_update = gr.update(value=last_interaction[0])
+        return chat_update, text_update        
 
     regenerate_btn.click(fn=remove_last_turn, inputs=chatbot, outputs=[chatbot, textbox]).then(
         fn=model_inference,