Update app.py

app.py

@@ -6,10 +6,29 @@ from pinecone_utilsB import *
 from graph_agentA import agent as agent_A
 from graph_agentB import agent as agent_B
 from functions import *
+from config import *
+from dotenv import load_dotenv
 
-
+def load_existing_indexes():
+    """Se connecter aux index Pinecone existants."""
+    try:
+        # Connexion aux index
+        sparse_index = pc.Index(sparse_index_name)
+        dense_index = pc.Index(dense_index_name)
+        st.write(f"Connexion aux index '{sparse_index_name}' et '{dense_index_name}' réussie.")
+        return sparse_index, dense_index
+    except Exception as e:
+        st.error(f"Erreur lors de la connexion aux index Pinecone : {e}")
+        return None, None
+        
 # Point d'entrée de l'application
 def main():
+    # Charger les index existants au démarrage
+    sparse_index, dense_index = load_existing_indexes()
+    if not sparse_index or not dense_index:
+        st.error("Impossible de se connecter aux index Pinecone. Vérifiez votre configuration.")
+        return
+
     # Initialiser l'historique de chat s'il n'existe pas
     if "chat_history" not in st.session_state:
         st.session_state.chat_history = []
@@ -29,7 +48,4 @@ def main():
             process_query_B(query)
         else:
             process_query_A(query)
-        st.rerun()
-
-if __name__ == "__main__":
-    main()
+        st.rerun()