Update app.py

app.py

@@ -20,8 +20,21 @@ def generate(message, history,temperature=0.7,max_tokens=1024):
 mychatbot = gr.Chatbot(
 avatar_images=["user.png", "bots.png"], bubble_full_width=False, show_label=False, show_copy_button=True, likeable=True,)
         
-iface = gr.ChatInterface(fn=generate, chatbot=mychatbot, retry_btn=retry_last_message, undo_btn=None)
+        def retry(message, history):
+    # Remove the last user message from the history
+    history = history[:-1]
+    # Generate a new response using the updated history
+    return generate(message, history)
 
+        def undo(message, history):
+    # Remove the last user message and the last bot response from the history
+    history = history[:-2]
+    # Return the updated history
+    return history
+        
+        
+iface = gr.ChatInterface(fn=generate, chatbot=mychatbot, retry_btn=retry, undo_btn=undo)
+        
 with gr.Blocks() as demo:
     gr.HTML("<center><h1>Chat with Solar</h1></center>")
     iface.render()