BasilTh
Deploy updated SLM customer-support chatbot
da2916f
import os
os.environ["OMP_NUM_THREADS"] = "1"
import gradio as gr
from SLM_CService import chat_with_memory, reset_state
def respond(user_message, history):
reply = chat_with_memory(user_message or "")
history = (history or []) + [
{"role":"user","content": user_message or ""},
{"role":"assistant","content": reply},
]
return history
def reset_chat():
reset_state() # <-- clear memory + globals
return [] # clear UI
with gr.Blocks() as demo:
gr.Markdown("# 🛎 Customer Support Chatbot")
chatbot = gr.Chatbot(type="messages")
with gr.Row():
user_in = gr.Textbox(placeholder="Ask about orders, tracking, returns…", scale=5)
send = gr.Button("Send", variant="primary")
reset = gr.Button("🔄 Reset Chat")
send.click(respond, [user_in, chatbot], [chatbot])
user_in.submit(respond, [user_in, chatbot], [chatbot])
reset.click(reset_chat, None, [chatbot]) # <-- real reset
if __name__ == "__main__":
demo.launch(server_name="0.0.0.0", server_port=7860)