fix version torch

requirements.txt

@@ -6,11 +6,11 @@ langchain-community
 sentence-transformers
 chromadb
 transformers>=4.36.0
-torch>=2.1.0
+torch==2.0.1
 sentencepiece
 tqdm
 accelerate
 
 
-
-
+pandas
+langchain-core

src/streamlit_app.py

@@ -68,30 +68,6 @@ similarity_threshold = st.sidebar.slider(
     step=5
 )
 
-def get_base_stats(source_key, source_dir):
-    # Nombre de documents indexés dans la base vectorielle
-    try:
-        embeddings = get_local_embeddings()
-        db_path = os.path.abspath("./db")
-        db = Chroma(persist_directory=db_path, embedding_function=embeddings)
-        # On compte les documents dont la métadonnée 'source' correspond
-        docs = db.get(where={"source": source_key})
-        nb_docs = len(docs.get("ids", []))
-    except Exception:
-        nb_docs = 0
-    # Date de dernière indexation = fichier le plus récent du dossier source
-    last_index_date = None
-    try:
-        dir_path = os.path.join("data", source_dir)
-        if os.path.exists(dir_path):
-            files = [os.path.join(dir_path, f) for f in os.listdir(dir_path) if os.path.isfile(os.path.join(dir_path, f))]
-            if files:
-                last_mtime = max(os.path.getmtime(f) for f in files)
-                last_index_date = datetime.datetime.fromtimestamp(last_mtime).strftime("%d/%m/%Y %H:%M")
-    except Exception:
-        pass
-    return nb_docs, last_index_date
-
 # Choix multi-bases avec checkbox
 st.sidebar.markdown("**Bases de documents à interroger :**")
 base_options = [
@@ -103,15 +79,10 @@ selected_bases = [
     key for label, key, _ in base_options if st.sidebar.checkbox(label, value=True)
 ]
 
-# Affichage des stats pour chaque base
-st.sidebar.markdown(":mag: **Statistiques des bases**")
-for label, key, data_dir in base_options:
-    nb_docs, last_index = get_base_stats(key, data_dir)
-    st.sidebar.markdown(f"- **{label}** : {nb_docs} document(s) indexé(s)")
-    if last_index:
-        st.sidebar.markdown(f"  <span style='color:gray;font-size:90%'>Dernière indexation : {last_index}</span>", unsafe_allow_html=True)
-    else:
-        st.sidebar.markdown(f"  <span style='color:gray;font-size:90%'>Aucune indexation détectée</span>", unsafe_allow_html=True)
+# Vérification qu'au moins une base est sélectionnée
+if not selected_bases:
+    st.sidebar.warning("⚠️ Veuillez sélectionner au moins une base de documents pour continuer.")
+    st.stop()  
 
 # Affichage des modèles utilisés (en bas de la sidebar)
 st.sidebar.markdown("---")