Update app.py

app.py

@@ -380,6 +380,179 @@ if submit or (not st.session_state.precalculated_df.empty):
         
 
         
+stop = st.button("New search!", type='primary', key=500)  
+if stop:               
+    st.write("New search is launched. Please specify your preferences in the form!")
+    st.session_state.preferences_1, st.session_state.preferences_2 = [], []
+    st.session_state.restrictions = []
+    st.session_state.sort_by = ""
+    st.session_state.df = init_df
+    st.session_state.precalculated_df = pd.DataFrame()
+            
+# #TODO: implement price range as a sliding bar
+# When the user presses "New search", erase everything
+# Propose URLs
+# Show keywords instead of whole strings
+
+
+# Define custom CSS styles for the orange and blue rectangles
+css = """
+<style>
+    .orange-box {
+        background-color: orange;
+        border: 2px solid darkred;
+        border-radius: 10px;
+        display: inline-block;
+        padding: 5px 10px;
+    }
+    
+    .blue-box {
+        background-color: lightblue;
+        border: 2px solid navy;
+        border-radius: 10px;
+        display: inline-block;
+        padding: 5px 10px;
+    }
+</style>
+"""
+
+# options_disability_1 = st.multiselect(
+#      'Do you need a wheelchair?',
+#      ['Yes', 'No'], ['No'], key=101)
+
+# if options_disability_1 == 'Yes':
+#     st.session_state.restrictions.append('Wheelchair')
+    
+# price_1 = st.select_slider("Your preferred price range", options=('$', '$$', '$$$', '$$$$'), key=3)
+
+# st.session_state.preferences_1.append(ambiance_1)
+
+# Komplettes Beispiel für die Verwendung der 'with'-Notation
+# with st.form('my_form_1'):
+#     st.subheader('**User 1**')
+
+st.write("User 1")
+# Eingabe-Widgets
+food_1 = st.selectbox('Select the food type you prefer', st.session_state.food, key=1)
+if food_1 == 'Other':
+    food_1 = st.text_input(label="Your description", placeholder="What kind of food would you like to eat?", key=10)
+
+ambiance_1 = st.selectbox('What describes your occasion the best?', st.session_state.ambiance, key=2)
+if ambiance_1 == 'Other':
+    ambiance_1 = st.text_input(label="Your description", placeholder="How would you describe your meeting?", key=11)
+
+options_food_1 = st.multiselect(
+    'Do you have any dietary restrictions?',
+    ['Vegan', 'Vegetarian', 'Halal'], key=100)
+
+additional_1 = st.text_input(label="Your description", placeholder="Anything else you wanna share?", key=102)  
+    
+with_kids = st.checkbox('I will come with kids', key=200)
+    
+    # st.subheader('**User 2**')
+st.write("User 2")
+
+# Eingabe-Widgets
+food_2 = st.selectbox('Select the food type you prefer', st.session_state.food, key=3)
+if food_2 == 'Other':
+    food_2 = st.text_input(label="Your description", placeholder="What kind of food would you like to eat?", key=4)
+
+ambiance_2 = st.selectbox('What describes your occasion the best?', st.session_state.ambiance, key=5)
+if ambiance_2 == 'Other':
+    ambiance_2 = st.text_input(label="Your description", placeholder="How would you describe your meeting?", key=6)
+
+options_food_2 = st.multiselect(
+    'Do you have any dietary restrictions?',
+    ['Vegan', 'Vegetarian', 'Halal'], key=7)
+
+additional_2 = st.text_input(label="Your description", placeholder="Anything else you wanna share?", key=8) 
+     
+with_kids_2 = st.checkbox('I will come with kids', key=201)
+    
+if len(st.session_state.preferences_1) == 0:
+    st.session_state.preferences_1.append(food_1)
+    st.session_state.preferences_1.append(ambiance_1)
+    st.session_state.restrictions.extend(options_food_1)
+    if additional_1:
+        st.session_state.preferences_1.append(additional_1)
+    if with_kids:
+        st.session_state.restrictions.append('kids')
+
+if len(st.session_state.preferences_2) == 0:
+    st.session_state.preferences_2.append(food_2)
+    st.session_state.preferences_2.append(ambiance_2)
+    st.session_state.restrictions.extend(options_food_2)
+    if additional_2:
+        st.session_state.preferences_2.append(additional_2)
+    if with_kids_2:
+        st.session_state.restrictions.append('kids')
+
+submitted = st.button('Submit!')
+    
+if submitted:
+    st.markdown("Thanks, we received your preferences!")
+    st.session_state.stop_search = False
+    
+else:
+    st.write('☝️ Describe your preferences!')
+
+
+submit = st.button("Find best matches!", type='primary')
+            
+if submit or (not st.session_state.precalculated_df.empty):
+    with st.spinner("Please wait while we are finding the best solution..."):
+        if st.session_state.precalculated_df.empty:
+            query = get_combined_preferences(st.session_state.preferences_1, st.session_state.preferences_2)
+            st.write("Your query is:", query[0])
+            #sort places based on restrictions
+            st.session_state.precalculated_df = filter_places(st.session_state.restrictions)
+            #sort places by elevating preferrences
+            st.session_state.precalculated_df = promote_places(query[1])
+            st.session_state.precalculated_df = compute_cos_sim(query[0])
+        sort_by = st.selectbox(('Sort by:'), st.session_state.options, key=400, 
+                               index=st.session_state.options.index('Relevancy (default)'))
+        if sort_by:
+            st.session_state.sort_by = sort_by
+            results = generate_results(st.session_state.sort_by)
+            k = 10
+            st.write(f"Here are the best {k} matches to your preferences:")
+            i = 1
+            nums = list(range(1, 11))
+            words = ['one', 'two', 'three', 'four', 'five', 'six', 'seven', 'eight', 'nine', 'one: :zero']
+            nums_emojis = dict(zip(nums, words))
+            for name, score in results.items():
+                
+                condition = st.session_state.precalculated_df['Names'] == name
+                rating = st.session_state.precalculated_df.loc[condition, 'Rating'].values[0]
+                st.write(f":{nums_emojis[i]}: **{name}** **({str(rating)}**:star:) :", 'match score:', score)
+    
+                # Use the condition to extract the value(s)
+                # description = st.session_state.precalculated_df.loc[condition, 'Strings']
+                # st.write(description)
+                
+                type = [item for item in eval(st.session_state.precalculated_df.loc[condition, 'Category'].values[0])]
+                # Display HTML with the custom styles
+                for word in type:
+                    st.markdown(css, unsafe_allow_html=True)
+                    st.markdown(f'<div class="blue-box">{word}</div>', unsafe_allow_html=True)
+                # st.write("Restaurant type:", str(type))
+                
+                keywords = [item[0] for item in eval(st.session_state.precalculated_df.loc[condition, 'Keywords'].values[0])]
+                for pair in keywords[:3]:
+                    st.markdown(css, unsafe_allow_html=True)
+                    st.markdown(f'<div class="orange-box">{pair[0]} {pair[1]}</div>', unsafe_allow_html=True)
+                # st.write("Restaurant type:", str(type))
+                
+                
+                url = st.session_state.precalculated_df.loc[condition, 'URL'].values[0]
+                st.write("_Check on the map:_", url)
+                            
+                # st.markdown("This is a text with <span style='font-size: 20px;'>bigger</span> and <i>italic</i> text.", unsafe_allow_html=True)
+                # st.markdown("<span style='font-size: 24px;'>This is larger text</span>", unsafe_allow_html=True)
+
+        
+
+        
 stop = st.button("New search!", type='primary', key=500)  
 if stop:               
     st.write("New search is launched. Please specify your preferences in the form!")