import logging
import sys

import gradio as gr
from dotenv import load_dotenv

from app.llama_index import Assistant

load_dotenv()

logging.basicConfig(
    format="[%(asctime)s][%(levelname)s][%(name)s] %(message)s",
    datefmt="%d/%m/%Y-%H:%M:%S",
    stream=sys.stdout,
    level=logging.INFO,
)

logger = logging.getLogger("app.app")

title = "Årsopgørelses Assistent"
description = "Assistent til at hjælpe med at besvare spørgsmål omkring årsopgørelsen"
examples = [
    "hvad er vigtigt at vide om årsopgørelsen?",
    "hvordan beregnes årsopgørelsen?",
]

assistant = Assistant(data_dir="data", model_name="gpt-4")


def run(message: str, history: list[str]) -> str:
    response = assistant(message, history=history)
    return response


demo = gr.ChatInterface(fn=run, examples=examples, title=title, description=description)

demo.queue().launch()