refactor: Enhance prompt handling and session state management in chatbot

pages/chatbot.py

@@ -41,13 +41,22 @@ def show_prompts(prompts: list[str]):
     expander = st.expander("Prompts pré-définis")
 
     for i, p in enumerate(prompts):
-        if expander.button(p, key=f"button_{chapter_num}_{i}_{uuid.uuid4()}"):
-            launchQuery(p)
+        button_key = f"button_{chapter_session_key}_{i}"
 
-def inputSubmut():
-    user_query = st.session_state[f"input_{chapter_session_key}"]
+        # Réinitialisation si la clé existe déjà (quand on change de page)
+        if button_key in st.session_state:
+            del st.session_state[button_key]
+            
+        expander.button(p, key=button_key, on_click=promptSubmit, args=(p,))
+
+def promptSubmit(key,):
+    inputSubmut(key)
+
+
+def inputSubmut(query: str = None):
+    # Get query or find in session state
+    user_query = query if query is not None else st.session_state[f"input_{chapter_session_key}"]
 
-    
     if user_query is not None and user_query != "":
         st.session_state[chapter_session_key]["messages"].append(HumanMessage(content=user_query))
         
@@ -99,7 +108,7 @@ def page():
     st.markdown("<style>iframe{height:50px;}</style>", unsafe_allow_html=True)
 
     # Collpase for default prompts
-    # show_prompts(chapter['prompts'])
+    show_prompts(chapter['prompts'])
 
     # Models selector
     selector.ModelSelector()