File size: 873 Bytes
e64f61b |
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 27 28 29 30 31 32 33 34 35 36 37 38 |
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()
|