import lumen.ai as lmai
import panel as pn
from lumen.sources.duckdb import DuckDBSource

pn.extension(
    "tabulator",
    "codeeditor",
    "vega",
    "filedropper",
    inline=False,
    template="fast",
    notifications=True,
)

lmai.memory["current_source"] = DuckDBSource(
    tables=["olympic_hosts.csv"],
    uri=":memory:",
    initializers=["SET home_directory='/tmp';", "INSTALL httpfs;", "LOAD httpfs;"],
)
llm = lmai.llm.OpenAI()
assistant = lmai.ExplorerUI(llm=llm)
assistant.servable("Lumen.ai")