File size: 834 Bytes
9cccf74 a4a2f1b 9cccf74 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
import gradio as gr
from builder.agent_runner import run_agent
def build_ui(app, tracer):
with gr.Blocks() as demo:
gr.Markdown("# Datum : Autonomous Data Analysis Agent")
chatbot = gr.Chatbot(type="messages")
user_input = gr.Textbox(label="Ask a question", placeholder="Ex: Show me marketing spend by channel")
submit_btn = gr.Button("Send", variant='primary')
state = gr.State([])
user_input.submit(
lambda m, h: run_agent(app, tracer, m, h),
inputs=[user_input, state],
outputs=[chatbot, state, user_input]
)
submit_btn.click(
lambda m, h: run_agent(app, tracer, m, h),
inputs=[user_input, state],
outputs=[chatbot, state, user_input]
)
return demo
|