ui.py

import gradio as gr
from gradio.themes.base import Base

class ChatUI:
    def __init__(self, chat_service):
        self.chat_service = chat_service

    def launch(self):
        with gr.Blocks(theme=Base()) as demo:
            gr.Markdown("## Customer Financial Chatbot")

            customer = gr.Radio(
                choices=self.chat_service.get_customer_names(),
                label="Select Customer"
            )

            question = gr.Textbox(
                label="Ask a question",
                placeholder="What is the total value of all buy transactions?"
            )

            answer = gr.Textbox(lines=8, label="Answer")

            gr.Button("Ask").click(
                fn=self.chat_service.chat,
                inputs=[customer, question],
                outputs=answer
            )

        demo.launch()