fix submission

app.py

@@ -196,36 +196,20 @@ 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.user_id=="" or st.session_state.score==50 or st.session_state.personalization_score==50):
-        submission_date = datetime.now() #.strftime("%Y-%m-%d %H:%M:%S")
-
-        user_data={"user_id":st.session_state.user_id,
-                    "conversation":st.session_state.messages,
-                    "score":st.session_state.score,
-                    "personalization_score":st.session_state.personalization_score,
-                    "knowledge_score":st.session_state.knowledge_score,
-                    "model":st.session_state["model"],
-                    #"user_info":st.session_state.user_info,
-                    "feedback":st.session_state.feedback,
-                    "condition":f"p{st.query_params['p']}",
-                    "age":st.session_state.age,
-                    "gender":st.session_state.gender,
-                    "education":st.session_state.education,
-                    "residence":st.session_state.residence,
-                    #"zipcode":st.session_state.zipcode,
-                    "property":st.session_state.property,
-                   # "income":st.session_state.income,
-                    "politics":st.session_state.politics,
-                    "climate_actions":st.session_state.climate_actions,
-                    "impact_open":st.session_state.impact_open,
-                    "recycling":st.session_state.recycling,
-                    "ev":st.session_state.ev,
-                    "fossil":st.session_state.fossil,
-                    "aerosol":st.session_state.aerosol,
-                    "diet":st.session_state.diet,
-                    "inserted":st.session_state.inserted,
-                    "start_time":st.session_state.start_time,
-                    "convo_start_time":st.session_state.convo_start_time,
-                    "submission_time":submission_date,}
+        #submission_date = datetime.now() #.strftime("%Y-%m-%d %H:%M:%S")
+
+        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"] = f"p{st.query_params['p']}"
+        user_data["submission_time"] = datetime.now()
         
         from pymongo.mongo_client import MongoClient
         from pymongo.server_api import ServerApi