AuraDB instance dynamic activation added successfully !!

app.py

@@ -231,6 +231,25 @@ def display_chat_history():
             with st.chat_message("assistant"):
                 st.markdown(message["content"])
 
+def is_neo4j_aura_active(retries: int, wait_seconds: int):
+    """
+    Tente de se connecter à l'instance Neo4j. Si elle est en veille,
+    la fonction essaiera de se reconnecter jusqu'à ce que l'instance soit réveillée.
+
+    Args:
+        retries (int): Nombre de tentatives avant d'abandonner.
+        wait_seconds (int): Temps d'attente entre les tentatives.
+    """
+    for attempt in range(1, retries + 1):
+        try:
+            with neo4j_driver.session() as session:
+                session.run("RETURN 1") # Simple requête pour réveiller l’instance
+            return True
+        except Exception as e:
+            print(f"⚠️ Tentative {attempt}/{retries} : l'instance est en veille. Nouvel essai dans {wait_seconds}s...")
+            time.sleep(wait_seconds)
+    raise RuntimeError("❌ Impossible de réveiller l'instance Neo4j après plusieurs tentatives.")
+
 def main():
 
     initCarbon()
@@ -265,20 +284,59 @@ def main():
     
     st.markdown("")
     
-
-    architecture = st.selectbox(
-        "Sélectionnez une architecture", 
+    # Init states
+    if "architecture" not in st.session_state:
+        st.session_state["architecture"] = None
+    if "confirmation_needed" not in st.session_state:
+        st.session_state["confirmation_needed"] = False
+    if "user_response" not in st.session_state:
+        st.session_state["user_response"] = None
+    if "selected_arch" not in st.session_state:
+        st.session_state["selected_arch"] = "Basic"
+
+    # Affichage du selectbox
+    selection = st.selectbox(
+        "Sélectionnez une architecture RAG :",
         ["Basic", "Intermédiaire", "Avancée"],
-        on_change=st.session_state.chat_history.clear
-    )        
-
-    # Sauvegarder l'architecture dans st.session_state
-    if architecture == "Intermédiaire":
-        st.session_state["architecture"] = "B"
-    elif architecture == "Avancée":
-        st.session_state["architecture"] = "C"
+        index=["Basic", "Intermédiaire", "Avancée"].index(st.session_state["selected_arch"]),
+    )
+
+    # Gestion de la logique
+    if selection == "Avancée":
+        st.session_state["confirmation_needed"] = True
+        st.session_state["architecture"] = None
+        st.session_state["selected_arch"] = "Avancée"
     else:
-        st.session_state["architecture"] = "A"
+        st.session_state["architecture"] = {
+            "Basic": "A",
+            "Intermédiaire": "B"
+        }[selection]
+        st.session_state["confirmation_needed"] = False
+        st.session_state["user_response"] = None
+        st.session_state["selected_arch"] = selection
+
+    # Popup confirmation si avancée
+    if st.session_state["confirmation_needed"]:
+        st.toast("⚠️ Attention, ce type de fonctionnement **Avancée** est plus précis, mais consomme plus de carbone. Souhaitez-vous réellement l'utiliser ?")
+        st.session_state["user_response"] = st.radio(
+            "Confirmez votre choix :",
+            options=["Oui", "Non"],
+            index=None,
+            key="confirm_radio"
+        )
+
+        # Ne rien faire tant que l'utilisateur n'a rien choisi
+        if st.session_state["user_response"] == "Oui":
+                    st.session_state["architecture"] = "C"
+                    st.toast("✅ Architecture 'Avancée' confirmée.")
+
+        elif st.session_state["user_response"] == "Non":
+            st.session_state["selected_arch"] = "Intermédiaire"
+            st.session_state["architecture"] = "B"
+            st.session_state["confirmation_needed"] = False
+            st.session_state["user_response"] = None
+            st.success("✅ Retour à l'architecture 'Intermédiaire'.")
+            st.rerun()  # 🔁 Forcer le widget à se recréer
 
     display_sidebar()
 
@@ -287,7 +345,15 @@ def main():
     
     query = st.chat_input("Posez votre question ici:")
     if query:
-        process_query(query, st.session_state["architecture"])
+        if st.session_state.get("architecture") == "C":
+            #with st.spinner("⏳ Activation de l'instance Neo4j AuraDB en cours..."):
+                if is_neo4j_aura_active(10, 60):  # 10 retries, 60 seconds each
+                    st.toast("✅ Neo4j AuraDB est actif.")
+                    process_query(query, "C")
+                else:
+                    st.error("❌ Échec de l'activation de Neo4j. Veuillez réessayer plus tard.")
+        else:
+            process_query(query, st.session_state["architecture"])
 
            
 def calculate_tokens_cost():