import gradio as gr
from src.agent.core import chat

def build_ui() -> gr.Blocks:
    with gr.Blocks() as demo:
        gr.Markdown("# My Agent")
        inp = gr.Textbox(label="Message", placeholder="Napisz…")
        out = gr.Markdown()

        def _run(msg: str) -> str:
            return chat(msg)

        gr.Button("Send").click(_run, inputs=inp, outputs=out)
    return demo