Fix undo button

Tabs/Gemini_Chabot_Stable.py

@@ -64,6 +64,9 @@ chat = model.start_chat(history=[])
 
 def clear_chat_history():
     chat.history = []
+    
+def undo_chat():
+    last_send, last_received = chat.rewind()
 
 def transform_history(history):
     new_history = []
@@ -103,9 +106,14 @@ clear_chat_button = gr.ClearButton(
     value="🗑️ Clear"
 )
 
+undo_chat_button = gr.Button(
+    value="↩️ Undo"
+)
+
 gemini_chatbot = gr.ChatInterface(
     fn=chatbot_stable,
     chatbot=gemini_chatbot_interface,
     multimodal=True,
-    clear_btn=clear_chat_button
+    clear_btn=clear_chat_button,
+    undo_btn=undo_chat_button
 )

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-from Tabs.Gemini_Chabot_Stable import gemini_chatbot, clear_chat_button, clear_chat_history, TITLE, NOTICE, ERRORS, FUTURE_IMPLEMENTATIONS, ABOUT
+from Tabs.Gemini_Chabot_Stable import gemini_chatbot, clear_chat_button, undo_chat_button, clear_chat_history, undo_chat, TITLE, NOTICE, ERRORS, FUTURE_IMPLEMENTATIONS, ABOUT
 
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     # ============================== Stable - START ==============================
@@ -14,6 +14,9 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         clear_chat_button.click(
             fn=clear_chat_history
         )
+        undo_chat_button.click(
+            fn=undo_chat
+        )
     # ============================== Stable - END ==============================
 
     with gr.Tab("Chat with Gemma 2"):