"""
streamlit web 网页上传
"""
import time

import streamlit as st
from knowledge_base import KnowledgeBaseService

# 标题
st.title("知识库更新服务")

# file_uploader
uploader_file = st.file_uploader(
    "请上传txt文件",
    type="txt",
    accept_multiple_files=False
)

if "service" not in st.session_state:
    st.session_state["service"] = KnowledgeBaseService()

if uploader_file is not None:
    file_name = uploader_file.name
    file_type = uploader_file.type
    file_size = uploader_file.size / 1024

    st.subheader(f"文件名：{file_name}")
    st.write(f"格式: {file_type}, 大小: {file_size:.2f}KB")
    # bytes
    text = uploader_file.getvalue().decode('utf-8')

    with st.spinner("载入知识库中..."):
        time.sleep(1)
        result = st.session_state["service"].upload_by_str(text, file_name)
        st.write(result)