Another minor UI update

foodnet.py

@@ -62,6 +62,7 @@ def recommend_ingredients(yum, leftovers, n=10):
   leftovers_embedding = leftovers_embedding_sum / len(leftovers) # Embedding for leftovers
   top_matches = yum.similar_by_vector(leftovers_embedding, topn=100)
   top_matches = [(x[0].replace('_',' '), x[1]) for x in top_matches]
+  leftovers = [x.replace('_',' ') for x in leftovers]
   output = [x for x in top_matches if not any(ignore in x[0] for ignore in leftovers)] # Remove boring same item matches, e.g. "romaine lettuce" if leftovers already contain "lettuce".
   return output[:n]
 
@@ -248,7 +249,7 @@ def load_image(image_file):
 	img = Image.open(image_file)
 	return img
 
-st.set_page_config(page_title="FoodNet", page_icon = "🍔", layout = "centered", initial_sidebar_state = "expanded")
+st.set_page_config(page_title="FoodNet", page_icon = "🍔", layout = "centered", initial_sidebar_state = "collapsed")
 
 ##### UI/UX #####
 ## Sidebar ##
@@ -261,10 +262,10 @@ if add_selectbox == "FoodNet Recommender":
     st.write("Search for similar food ingredients. Select two or more ingredients to find complementary ingredients.")
     ingredients = list(yum.key_to_index.keys())
     ingredients = [x.replace('_',' ') for x in ingredients]
-    st.multiselect("Type or select food ingredients", ingredients, default=['bread', 'lettuce'], key="leftovers")
+    st.multiselect("Type or select food ingredients", ingredients, default=['tomato', 'parsley'], key="leftovers")
 
     ## Slider ##
-    st.slider("Number of recommendations to show", min_value=1, max_value=100, value=5, step=1, key='top_n')
+    st.slider("Select number of recommendations to show", min_value=1, max_value=100, value=5, step=1, key='top_n')
 
     ## Show Images ## 
     # search = py_un.search(type_="photos", query="cookie")
@@ -302,45 +303,29 @@ if add_selectbox == "FoodNet Recommender":
 
     #     ## Show Plot ##
     #     st.pyplot(fig)
-    st.selectbox(label="Dietary Resriction", options=('None', 'Kosher', 'Vegitarian'), key="diet")
+    st.selectbox(label="Select dietary restriction", options=('None', 'Kosher', 'Vegitarian'), key="diet")
     if st.session_state.diet != 'None':
       if st.session_state.diet == 'Vegitarian':
         out = [o for o in out if not any(ignore in o[0] for ignore in blacklists.vegitarian)]
-      if st.session_state.diet == "Kosher": 
+      if st.session_state.diet == 'Kosher': 
         out = [o for o in out if not any(ignore in o[0] for ignore in blacklists.kosher)]
       names = [o[0] for o in out]
       probs = [o[1] for o in out]
 
     col1, col2, col3 = st.columns(3)
 
-    # st.checkbox(label="Show model score", value=False, key="probs")
-    # if st.session_state.probs:
-    #   col1.table(data=out)
-    # else:
-    #   col1.table(data=names)
-    
     for i, name in enumerate(names): 
       search = py_un.search(type_='photos', query=name)
       for photo in search.entries:
-        col_id = i%3
+        col_id = i % 3
         if col_id == 0:
           col1.image(photo.link_download, caption=name, use_column_width=True)
         elif col_id == 1:
           col2.image(photo.link_download, caption=name, use_column_width=True)
         elif col_id == 2: 
           col3.image(photo.link_download, caption=name, use_column_width=True)
-          # print(photo.id, photo.link_download)
-        
         break
 
-    ## Plot Results ##
-    # st.checkbox(label="Show model bar chart", value=False, key="plot")
-    # if st.session_state.plot:
-    #   fig = plot_results(names, probs, st.session_state.top_n)
-      
-    #   ## Show Plot ## 
-    #   col2.pyplot(fig)
-
 elif add_selectbox == "Food Donation Resources":
     st.title('Food Donation Resources')
     st.subheader('Pittsburgh Food Bank:')