Confirmation handled successfully !

app.py

@@ -251,92 +251,77 @@ def is_neo4j_aura_active(retries: int, wait_seconds: int):
     raise RuntimeError("❌ Impossible de réveiller l'instance Neo4j après plusieurs tentatives.")
 
 def main():
-
     initCarbon()
-
     st.session_state['current_model'] = "mistral-large-latest"
 
-    if "tokens_metrics" not in st.session_state or "chat_history" not in st.session_state :
+    if "tokens_metrics" not in st.session_state or "chat_history" not in st.session_state:
         initialize_conversation()
 
-
-    # """Point d'entrée de l'application."""
-    # Vérifier que les index Pinecone sont prêts
     if not check_indexes_ready():
         return
 
-
     st.title("RAG architectures")
-    st.markdown(
-        """
+    st.markdown("""
         <style>
         div[data-testid="stSelectbox"] {
             width: 200px !important;  
         }
         </style>
-        """,
-        unsafe_allow_html=True
-    )
+        """, unsafe_allow_html=True)
 
     st.markdown("• Basique : *Recherche sémantique*")
     st.markdown("• Intermédiaire : *Rechercher hybride (sémantique + mots clés)*") 
     st.markdown("• Avancée : *Graph de connaissance + Recherche hybride*")
-    
-    st.markdown("")
-    
-    # Init states
+
+    # Initialisation des états
     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
+        st.session_state["architecture"] = "A"
     if "selected_arch" not in st.session_state:
         st.session_state["selected_arch"] = "Basic"
+    if "first_time_advanced" not in st.session_state:
+        st.session_state["first_time_advanced"] = True
+    if "advanced_info_shown" not in st.session_state:
+        st.session_state["advanced_info_shown"] = False
 
-    # Affichage du selectbox
-    selection = st.selectbox(
+    # Selectbox
+    new_selection = st.selectbox(
         "Sélectionnez une architecture RAG :",
         ["Basic", "Intermédiaire", "Avancée"],
         index=["Basic", "Intermédiaire", "Avancée"].index(st.session_state["selected_arch"]),
+        key="arch_selection"
     )
 
-    # 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"] = {
-            "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":
+    # Si la sélection change
+    if new_selection != st.session_state["selected_arch"]:
+        if new_selection == "Avancée":
+            # Message de confirmation pour première fois
+            if st.session_state["first_time_advanced"]:
+                st.info("⚠️ Attention, ce type de fonctionnement est plus précis mais consomme plus de carbone. Souhaitez-vous réellement l'utiliser ?")
+                col1, col2 = st.columns(2)
+                if col1.button("✅ Confirmer"):
                     st.session_state["architecture"] = "C"
-                    st.toast("✅ Architecture 'Avancée' confirmée.")
+                    st.session_state["selected_arch"] = "Avancée"
+                    st.session_state["first_time_advanced"] = False
+                    st.session_state["advanced_info_shown"] = True
+                    st.toast("Mode avancé activé")
+                    st.rerun()
+                if col2.button("❌ Annuler"):
+                    st.session_state["selected_arch"] = "Intermédiaire"
+                    st.rerun()
+                return
+            else:
+                st.session_state["architecture"] = "C"
+                st.session_state["selected_arch"] = "Avancée"
+                st.session_state["advanced_info_shown"] = True
+        else:
+            st.session_state["architecture"] = "A" if new_selection == "Basic" else "B"
+            st.session_state["selected_arch"] = new_selection
+        st.rerun()
 
-        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
+    # Affichage du message d'information persistant
+    if st.session_state["selected_arch"] == "Avancée" and st.session_state["advanced_info_shown"]:
+        st.info("ℹ️ Vous utilisez actuellement l'architecture avancée. Cette option est optimisée pour la précision mais nécessite davantage de ressources.")
+  
 
     display_sidebar()
 
@@ -347,7 +332,7 @@ def main():
     if query:
         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
+                if is_neo4j_aura_active(3, 10):  # 3 retries, 10 seconds each
                     st.toast("✅ Neo4j AuraDB est actif.")
                     process_query(query, "C")
                 else: