Fiw stream nicely, without reloading page

app.py

@@ -31,6 +31,9 @@ def check_indexes_ready():
 def process_query(query, architecture: Literal["A", "B"]):
     """Traite la requête de l'utilisateur avec l'architecture A."""
 
+    # Reload conversation
+    display_chat_history()
+
     config = { "metadata": { "architecture": architecture }, "tags": ["arch_"+architecture] }
 
     if architecture == "A":
@@ -40,27 +43,32 @@ def process_query(query, architecture: Literal["A", "B"]):
 
     st.session_state.chat_history.append({"role": "user", "content": query})
 
-    with st.spinner("Recherche en cours..."):
-        initial_state = {"query": query, "messages": [], "relevant_docs": [], "response": ""}
+    # with st.spinner("Recherche en cours..."):
+    initial_state = {"query": query, "messages": [], "relevant_docs": [], "response": ""}
+
+    with st.chat_message("user"):
+        st.markdown(query)
+
+    full_response = ""
+
+    events = agent.stream(initial_state, config=config, stream_mode="messages")
+    
+    # Ajouter le message du chatbot avec streaming
+    with st.chat_message("assistant"):
+        response_placeholder = st.empty()
 
-        # Conteneur pour le streaming
-        message_placeholder = st.empty()
-        response = "" 
 
-        events = agent.stream(initial_state, config=config, stream_mode="messages")
-        
-        # Affichage en temps réel des messages
         for event in events:
             for message in event:
                 
                 if isinstance(message, AIMessageChunk):
                     # Vérifiez le type de message et affichez son contenu
                     if hasattr(message, 'content'):
-                        response += message.content
-                        message_placeholder.markdown(response + "▌")
-        
-        # response = result["response"]
-        st.session_state.chat_history.append({"role": "assistant", "content": response})
+                        full_response += message.content
+                        response_placeholder.markdown(full_response)
+    
+    
+    st.session_state.chat_history.append({"role": "assistant", "content": full_response})
 
 def display_sidebar():
     """Affiche la barre latérale."""
@@ -106,8 +114,8 @@ def main():
         ["Basic", "Intermédiaire", "Avancée"],
         on_change=st.session_state.chat_history.clear
     )        
+
     display_sidebar()
-    display_chat_history()
     
     query = st.chat_input("Posez votre question ici:")
     if query:
@@ -118,7 +126,5 @@ def main():
 
         process_query(query, architecture)
 
-        st.rerun()
-
 if __name__ == "__main__":
     main()