Update app.py

app.py

@@ -16,7 +16,12 @@ def respond(message, chat_history):
     outputs = model.generate(inputs, max_length=200, pad_token_id=tokenizer.eos_token_id)
     response_text = tokenizer.decode(outputs[:, inputs.shape[-1]:][0], skip_special_tokens=True)
     history.append({"role": "assistant", "content": response_text})
-    return history, ""  # reset input box
+    return history, ""
+
+def reset_chat():
+    global history
+    history = []
+    return [], ""  # clears chatbox and input
 
 with gr.Blocks(css="""
 body {background-color: #000 !important; color: #fff !important;}
@@ -25,12 +30,13 @@ body {background-color: #000 !important; color: #fff !important;}
 .gr-chatbot .message.bot {border-color: #aaa; background-color: transparent !important;}
 .gr-textbox textarea {background-color: transparent !important; color: #fff !important; border: 1px solid #555 !important;}
 .gr-textbox textarea::selection {background-color: #0ff !important; color: #000 !important;}
-.gr-button {display: none !important;}
+.gr-button {background-color: #0ff !important; color: #000 !important; border-radius: 8px;}
 footer {display: none !important;}
 """) as demo:
-    chatbot = gr.Chatbot(label="DevMegaBlack AI Chat")
+    chatbot = gr.Chatbot(label="DevMegaBlack")
     msg = gr.Textbox(placeholder="Say something...")
+    reset_btn = gr.Button("Reset Chat")
     msg.submit(respond, [msg, chatbot], [chatbot, msg])
+    reset_btn.click(reset_chat, [], [chatbot, msg])
 
 demo.launch()
-