Spaces:
Runtime error
Runtime error
File size: 3,177 Bytes
07645e7 |
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 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 |
import logging
import uuid
import streamlit as st
from st_pages import Page, show_pages
from nh_chatbot import create_engine
from nh_chatbot.etl import utils
from nh_chatbot.utils import filter_no_source_response, get_urls_from_sources
logging.basicConfig(
level=logging.INFO,
filename="app.log",
filemode="a",
format="%(asctime)s - %(name)s - %(levelname)s - Session ID: %(session_id)s - %(message)s",
)
def reset_chat():
st.session_state.chat_engine.reset()
init_chat()
def show_info_text():
st.header("NH Chatbot")
st.markdown("Indskriv dit spørgsmål i feltet placeret i bunden af siden 👇")
st.markdown("")
def init_chat():
st.session_state.messages = []
st.session_state.chat_engine.reset()
with open("app.log", "r") as file:
log_content = file.read()
show_info_text()
show_pages(
[
Page("app.py", "Hjem", "🏠"),
Page("other_pages/settings.py", "Indstillinger", "⚙️"),
Page("other_pages/developer.py", "Udvikler", "🔨"),
]
)
css = """
<style>
[data-testid="stSidebar"]{
min-width: 400px;
max-width: 800px;
}
</style>
"""
st.markdown(css, unsafe_allow_html=True)
@st.cache_resource(show_spinner="Starting engine..")
def load_engine_from_cache():
return create_engine.create_engine()
if "chat_engine" not in st.session_state:
engine = load_engine_from_cache()
st.session_state.chat_engine = engine
# Initialize chat history
if "messages" not in st.session_state:
init_chat()
if "session_id" not in st.session_state:
st.session_state.session_id = uuid.uuid4()
# NOTE: Display chat messages from history on app rerun
for message in st.session_state.messages:
with st.chat_message(message["role"]):
st.markdown(message["content"])
dummy_input = "Kan jeg tilføje oplysninger, dokumenter eller bilag til klagen ved at sende dem via mail efterfølgende?"
dummy_input = "..."
prompt = st.chat_input(dummy_input)
if prompt:
with st.chat_message(name="User"):
st.write(prompt)
with st.spinner("Tænker..."):
response = st.session_state.chat_engine.chat(prompt)
with st.chat_message(name="LLM"):
enhanced_response = f"{response.response}\n"
if len(response.source_nodes) > 0:
# NOTE: Extract references from the source node metadata
urls = get_urls_from_sources(response)
enhanced_response = enhanced_response + "\n*Kildehenvisning:*\n"
for url in set(urls):
enhanced_response = enhanced_response + f"\n- *{url}*"
else:
enhanced_response = filter_no_source_response(response)
st.markdown(enhanced_response)
st.session_state.messages.append(
{"role": "user", "content": prompt},
)
st.session_state.messages.append(
{"role": "llm", "content": enhanced_response},
)
enhanced_response = str(enhanced_response).replace("\n", "\\n")
logging.info(f"user: {prompt}", extra={"session_id": st.session_state.session_id})
logging.info(
f"llm: {enhanced_response}",
extra={"session_id": st.session_state.session_id},
)
|