Changed how matching keywords in results show + removed full description from results

app.py

@@ -59,7 +59,8 @@ def compute_cos_sim(input):
     # for el in st.session_state.preferences_2:
     #     query += el
     
-    st.write("Your query for cos sim calculation is", query)
+    # st.write("Your query is", query)
+    # st.write("Your restrictions are", st.session_state.restrictions)
     embedded_query = get_bert_embeddings(query, model, tokenizer)
     embedded_query = embedded_query.numpy()
     top_similar = np.array([])
@@ -180,8 +181,8 @@ def promote_places():
     a function that takes most common words, checks if descriptions fit them, increases their weight if they do
     '''
     #punish the weight of places that don't fit restrictions
-    st.write("Here are the most common preferences you provided:")
-    st.write(st.session_state.fixed_preferences)
+    # st.write("Here are the most common preferences you provided:")
+    # st.write(st.session_state.fixed_preferences)
     
     preferences = st.session_state.fixed_preferences
 
@@ -259,6 +260,9 @@ if 'precalculated_df' not in st.session_state:
     
 if 'results' not in st.session_state:
     st.session_state.results = {}
+    
+if 'fixed_restrictions' not in st.session_state:
+    st.session_state.fixed_restrictions = []
 
 # Configure Streamlit page and state
 st.title("GoTogether!")
@@ -417,6 +421,7 @@ if submit or (not st.session_state.precalculated_df.empty):
             query = get_combined_preferences(st.session_state.preferences_1, st.session_state.preferences_2)
             #sort places based on restrictions
             st.session_state.precalculated_df = filter_places(st.session_state.restrictions)
+            st.session_state.fixed_restrictions = st.session_state.restrictions
             #sort places by elevating preferrences
             
             st.session_state.precalculated_df = promote_places()
@@ -452,8 +457,13 @@ if submit or (not st.session_state.precalculated_df.empty):
                         
                         descr = st.session_state.precalculated_df.loc[condition, 'Strings'].values[0]
                         for word in set([word.lower() for word in descr.split()]):
-                            if word in st.session_state.fixed_preferences:
+                            if word in [el.lower() for el in st.session_state.fixed_preferences]:
                                 st.markdown(f'✅{word.capitalize()}')
+                            if word in [el.lower() for el in st.session_state.fixed_restrictions]:
+                                if word == 'kids':
+                                    st.markdown(f'✅Good for kids')
+                                else: 
+                                    st.markdown(f'✅{word.capitalize()}')
                                
                         
                         #Restaurant category
@@ -478,7 +488,7 @@ if submit or (not st.session_state.precalculated_df.empty):
                         url = st.session_state.precalculated_df.loc[condition, 'URL'].values[0]
                         st.write(f"_Check on the_ [_map_]({url})")
                          
-                        st.write(descr)
+                        # st.write(descr)
                                         
                     i+=1
                             
@@ -487,7 +497,7 @@ if submit or (not st.session_state.precalculated_df.empty):
 
         
             st.session_state.preferences_1, st.session_state.preferences_2 = [], []
-            st.session_state.restrictions = []
+            # st.session_state.restrictions = []
         
 stop = st.button("New search!", type='primary', key=500)  
 if stop:               
@@ -500,4 +510,5 @@ if stop:
     st.session_state.precalculated_df = pd.DataFrame()
     st.session_state.results = {}
     st.session_state.fixed_preferences = []
+