upd t and m v3

app.py

@@ -54,6 +54,7 @@ if 'inserted' not in st.session_state:
     st.session_state["model"] = "deepseek-ai/DeepSeek-V3"
     st.session_state.max_messages = 50
     st.session_state.messages = []
+    st.session_state.user_data = {}
 
     # user info state
     st.session_state.fields = [
@@ -163,19 +164,19 @@ def form():
 
     if submitted:
 
-        user_data = {key: st.session_state[key] for key in st.session_state.fields}
-        user_data["model"] = st.session_state["model"]
-        user_data["condition"] = st.query_params['p']
-        user_data["start_time"] = st.session_state.start_time
-        user_data["inserted"] = st.session_state.inserted
-        user_data["submission_time"] = datetime.now()
+        st.session_state.user_data = {key: st.session_state[key] for key in st.session_state.fields}
+        st.session_state.user_data["model"] = st.session_state["model"]
+        st.session_state.user_data["condition"] = st.query_params['p']
+        st.session_state.user_data["start_time"] = st.session_state.start_time
+        st.session_state.user_data["inserted"] = st.session_state.inserted
+        st.session_state.user_data["submission_time"] = datetime.now()
         
         from pymongo.mongo_client import MongoClient
         from pymongo.server_api import ServerApi
         with MongoClient(st.secrets["mongo"],server_api=ServerApi('1')) as client:
                 db = client.chat
                 collection = db.app
-                collection.insert_one(user_data)  
+                collection.insert_one(st.session_state.user_data)  
                 st.session_state.inserted += 1
         st.session_state.submitted = True
         setup_messages()
@@ -203,27 +204,21 @@ def submit():
 
     st.text_area('Any feedback?',key="feedback")
     if st.button('Submit', key=None, help=None, use_container_width=True, disabled=st.session_state.score==50 or st.session_state.personalization_score==50):
-        keys = [
-                "user_id", "messages", 
-                "score", "personalization_score", "knowledge_score", 
-                "model", "feedback", 
-                "age", "gender", "education", "residence", "property", "politics",
-                "climate_actions", "impact_open", 
-                "recycling", "ev", "fossil", "aerosol", "diet",
-                "inserted", "start_time", 
-                "convo_start_time"
-            ]
-
-        user_data = {key: st.session_state[key] for key in keys}
-        user_data["condition"] = {st.query_params['p']}
-        user_data["submission_time"] = datetime.now()
+        keys = ["score", "personalization_score", "knowledge_score", 
+                "feedback", 
+                "inserted", "messages", "convo_start_time"]
+
+        st.session_state.user_data.update({key: st.session_state[key] for key in keys})
+        
+        st.session_state.user_data["convo_end_time"] = datetime.now()
         
+        print(st.session_state.user_data)
         from pymongo.mongo_client import MongoClient
         from pymongo.server_api import ServerApi
         with MongoClient(st.secrets["mongo"],server_api=ServerApi('1')) as client:
                 db = client.chat
                 collection = db.app
-                collection.insert_one(user_data)  
+                collection.insert_one(st.session_state.user_data)  
                 st.session_state.inserted += 1
                 
                 st.success('**Your conversation has been submitted! Please proceed with the survey.**', icon="✅")