remove duplication

app.py

@@ -3,8 +3,6 @@ import requests
 from pymongo.mongo_client import MongoClient
 from pymongo.server_api import ServerApi
 from datetime import datetime
-import time
-#import webbrowser
 import random
 
 st.set_page_config(
@@ -75,9 +73,9 @@ with st.sidebar:
 
     ❗ **Do not share any personal information (e.g., name or address).**
 
-    ⚠️ You must click on **at least 5 links** before you can submit the search. An *End Search* button will appear then. You can continue before submitting, but **you must submit the conversation to receive compensation**.
+    ⚠️ You must click on **at least 5 links** before you can submit the search. A *Finish* button will appear then. You can continue before submitting, but **you must finish and enter your completion code into the survey to recieve compensation**.
 
-    {"☑" if st.session_state.inserted > 1 else "☐"} **Step 3. Use the *End Search* button to  submit your response.** 
+    {"☑" if st.session_state.inserted > 1 else "☐"} **Step 3. Use the *Finish* button to  submit your response.** 
 
     **Do not forget to copy & paste your completion code!**
 
@@ -120,7 +118,7 @@ def display_results(results):
 
 @st.dialog('Form', width="large")
 def form():
-    st.markdown("**❗ Please answer every question to proceed.**")
+    st.markdown("**Please answer every question to proceed.**")
     st.session_state.age = st.text_input("How old are you in years?")
     st.session_state.gender = st.radio("Do you describe yourself as a man, a woman, or in some other way?", 
                 ['','Man', 'Woman', 'Other'])
@@ -167,50 +165,19 @@ def form():
         st.session_state.user_data["inserted"] = st.session_state.inserted
         st.session_state.user_data["submission_time"] = datetime.now()
         
-        with MongoClient(st.secrets["mongo"],server_api=ServerApi('1')) as client:
-                db = client.chat
-                collection = db.app
-                user_data = st.session_state.user_data
-                collection.insert_one(user_data)  
-                st.session_state.inserted += 1
+        #with MongoClient(st.secrets["mongo"],server_api=ServerApi('1')) as client:
+                #db = client.chat
+                #collection = db.app
+                #user_data = st.session_state.user_data
+                #collection.insert_one(user_data)  
+        
+        st.session_state.inserted += 1
         st.session_state.submitted = True
         st.rerun()
 
 if st.session_state.gotit and st.session_state.submitted == False:
     form()
 
-@st.dialog('Submit', width="large")
-def submit():
-    st.markdown("You must answer all questions marked with a ❗ to submit.")
-    if st.query_params["p"] != "n":
-        st.slider('❗ How would you rate the interaction on a scale from *Terrible* to *Perfect*?', 0, 100, format="", key="score", value=50)
-        st.slider('❗ How personalized did the interaction feel, on a scale from *Not at all* to *Extremely personalized*?', 0, 100, format="", key="personalization_score", value=50)
-        st.slider('❗ How much do you feel you learned from this interaction, on a scale from *Nothing* to *Very much*?', 0, 100, format="", key="knowledge_score", value=50)
-    else:
-        st.session_state.score = 0
-        st.session_state.personalization_score = 0
-        st.session_state.knowledge_score = 0
-
-    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 = ["score", "personalization_score", "knowledge_score", 
-                "feedback", 
-                "inserted", "messages", "convo_start_time", 'clicked_links']
-
-        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()
-
-        with MongoClient(st.secrets["mongo"],server_api=ServerApi('1')) as client:
-                db = client.chat
-                collection = db.app
-                user_data = st.session_state.user_data
-                del user_data['_id']
-
-                collection.insert_one(user_data)  
-                st.session_state.inserted += 1
-
-                st.rerun()
 
 if len(st.session_state.messages) >= st.session_state.max_messages:
     st.info(
@@ -226,7 +193,7 @@ elif st.session_state.inserted > 1:
     st.markdown(f'## {st.session_state.user_id}')
     st.markdown('**Please copy the code and enter it into the survey field below.**')
 
-elif query := st.text_input("Enter search query and press Enter ⏎", placeholder="Search Google..."):
+elif query := st.text_input("Enter search query and press Enter", placeholder="Search Google..."):
 
     if query != st.session_state.messages[-1]["query"] or st.session_state.page != st.session_state.messages[-1]["page"]:
         results = google_search(query, st.session_state.page)   
@@ -250,6 +217,21 @@ elif query := st.text_input("Enter search query and press Enter ⏎", placeholde
 if (len(st.session_state.clicked_links) > 4 or st.session_state.max_messages == len(st.session_state.messages)) and not st.session_state.inserted > 1:
     columns = st.columns((1,1,1))
     with columns[2]:
-        if st.button("End Search",use_container_width=True):
-            submit()
+        if st.button("Finish",use_container_width=True):
+            keys = ["inserted", "messages", "convo_start_time", 'clicked_links']
+
+            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()
+
+            with MongoClient(st.secrets["mongo"],server_api=ServerApi('1')) as client:
+                    db = client.chat
+                    collection = db.app
+                    user_data = st.session_state.user_data
+                    #del user_data['_id']
+
+                    collection.insert_one(user_data)  
+                    st.session_state.inserted += 1
+
+                    st.rerun()