Update app.py

app.py

@@ -43,36 +43,47 @@ def chat_bot_response(user_message, history):
     history.append((user_message, response))
     return "", history
 
+def clear_chat():
+    return "", []
+
+def clear_history_fn():
+    memory.clear()
+    return "", []
+
 # Gradio ChatGPT-like interface
 with gr.Blocks(css="""
 body {
-    background-color: #f7f7f8;
+    background-color: #0f172a;
     font-family: 'Segoe UI', sans-serif;
+    color: #e5e7eb;
 }
 .chatbox {
-    background-color: #ffffff;
-    border: 1px solid #ddd;
-    border-radius: 10px;
+    background-color: #1f2937;
+    border: 1px solid #334155;
+    border-radius: 12px;
     padding: 10px;
     max-height: 500px;
     overflow-y: auto;
 }
 .message.user {
-    background-color: #e8f0fe;
-    color: #000;
+    background-color: #2563eb;
+    color: #fff;
     padding: 8px 12px;
     border-radius: 10px;
     margin: 5px 0;
     align-self: flex-end;
 }
 .message.bot {
-    background-color: #f1f1f1;
-    color: #000;
+    background-color: #374151;
+    color: #fff;
     padding: 8px 12px;
     border-radius: 10px;
     margin: 5px 0;
     align-self: flex-start;
 }
+button {
+    margin-top: 10px;
+}
 """) as demo:
 
     gr.HTML("<h1 style='text-align: center;'>🤖 Ultron - ChatGPT Style Assistant</h1>")
@@ -80,8 +91,14 @@ body {
     chatbot = gr.Chatbot(elem_classes="chatbox", value=[])
     msg = gr.Textbox(placeholder="Ask Ultron anything...", show_label=False)
 
+    with gr.Row():
+        clear_btn = gr.Button("Clear Chat")
+        reset_memory_btn = gr.Button("Clear History")
+
     msg.submit(chat_bot_response, inputs=[msg, chatbot], outputs=[msg, chatbot])
+    clear_btn.click(clear_chat, outputs=[msg, chatbot])
+    reset_memory_btn.click(clear_history_fn, outputs=[msg, chatbot])
 
 # Run the app
 if __name__ == "__main__":
-    demo.launch(debug=True)
+    demo.launch(debug=True)