File size: 960 Bytes
a3d26e6 444d7f6 a3d26e6 |
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 |
import os
import tempfile
import streamlit as st
def uploadToDb():
for file in st.session_state["file_uploader_commun"]:
with tempfile.NamedTemporaryFile(delete=False) as tf:
tf.write(file.getbuffer())
file_path = tf.name
with st.session_state["ingestion_spinner"], st.spinner(f"Chargement {file.name}"):
st.session_state["assistant"].ingestToDb(file_path, filename=file.name)
os.remove(file_path)
def page():
st.subheader("Montez des documents communs")
st.file_uploader(
"Télécharger un documents",
type=["pdf"],
key="file_uploader_commun",
accept_multiple_files=True,
on_change=uploadToDb,
)
st.session_state["ingestion_spinner"] = st.empty()
st.divider()
st.write("Documents dans la base de données")
for doc in st.session_state["assistant"].vector_store.getDocs():
st.write(" - "+doc)
page() |