capstone_1 / interface.py
Chidamma's picture
Upload 6 files
83d3723 verified
import gradio as gr
def create_interface(process_query, clear_context):
with gr.Blocks(title="AI Assistant") as demo:
chatbot = gr.Chatbot(
[],
elem_id="chatbot",
bubble_full_width=False,
height=400
)
with gr.Row():
msg = gr.Textbox(
label="Your Message",
placeholder="Type your message here...",
scale=8
)
submit = gr.Button("Submit", scale=1)
clear = gr.Button("Clear")
def process_message(message, history):
response = process_query(message, history)
history.append((message, response))
return "", history
msg.submit(
process_message,
[msg, chatbot],
[msg, chatbot]
)
submit.click(
process_message,
[msg, chatbot],
[msg, chatbot]
)
clear.click(
clear_context,
None,
[chatbot, msg]
)
return demo