Update app.py

app.py

@@ -36,13 +36,35 @@ def respond(message, history: list[tuple[str, str]]):
         response += token
         yield response
 
-# Create the Gradio ChatInterface with Retry, Undo, and Clear buttons
-demo = gr.ChatInterface(
-    fn=respond,  # Function to handle chat responses
-    retry_btn="Retry",  # Add a Retry button
-    undo_btn="Undo",  # Add an Undo button
-    clear_btn="Clear",  # Add a Clear button
-)
+# Custom button actions
+def retry_last_message(history):
+    if history:
+        return history[:-1]  # Remove the last assistant response
+    return history
+
+def undo_last_interaction(history):
+    if history:
+        return history[:-1]  # Remove the last user-assistant interaction
+    return history
+
+def clear_chat(history):
+    return []  # Clear the entire chat history
+
+# Create the Gradio ChatInterface with custom buttons
+with gr.Blocks() as demo:
+    chatbot = gr.Chatbot(label="Chat with DeepSeek-R1")
+    msg = gr.Textbox(label="Your Message")
+    clear = gr.Button("Clear")
+    undo = gr.Button("Undo")
+    retry = gr.Button("Retry")
+
+    # Define button actions
+    clear.click(clear_chat, chatbot, chatbot)
+    undo.click(undo_last_interaction, chatbot, chatbot)
+    retry.click(retry_last_message, chatbot, chatbot)
+
+    # Handle user input and chatbot responses
+    msg.submit(respond, [msg, chatbot], chatbot)
 
 # Launch the Gradio app
 if __name__ == "__main__":