Change session state initialisation

app.py

@@ -21,19 +21,19 @@ def init_app():
     if "user" in query_params:
         st.session_state["id_user"] = query_params["user"]
 
-            
-    if len(st.session_state) == 0:
-        # Define Vectore store strategy
-        pinecone_connector = PineconeConnector()
-        vs_manager = VectoreStoreManager(pinecone_connector)
-
+    # Initialize session state keys if they are not already defined
+    if "retrived_documents" not in st.session_state:
         st.session_state["retrived_documents"] = []
+    if "messages" not in st.session_state:
         st.session_state["messages"] = []
+    if "remove_undefined_value" not in st.session_state:
         st.session_state["remove_undefined_value"] = True
+    if "assistant" not in st.session_state:
+        # Define Vectore store strategy
+        pinecone_connector = PineconeConnector()
+        vs_manager = VectoreStoreManager(pinecone_connector)
         st.session_state["assistant"] = Rag(vectore_store=vs_manager)
-        st.session_state["data_dict"] = config['variables']
-        st.session_state["prompt_system"] = config['prompt_system']
-
+    if "data_dict" not in st.session_state:
         if 'parts' in config['variables']:
             # Flatten structure by adding part name to each field
             st.session_state["data_dict"] = [
@@ -44,6 +44,9 @@ def init_app():
         else:
             # Initialize session state with single list of variables
             st.session_state["data_dict"] = [{**field} for field in config["variables"]]
+    
+    if "prompt_system" not in st.session_state:
+        st.session_state["prompt_system"] = config['prompt_system']
 
 
 def main():