UI/UX updates

foodnet.py

@@ -222,8 +222,9 @@ def plot_results(names, probs, n=5):
   ax.set_ylabel('Probability',fontsize='large', fontweight='bold')
   ax.set_xlabel('Ingredients', fontsize='large', fontweight='bold')
   ax.xaxis.labelpad = 10
-  ax.set_title(f'FastFood Top {n} Predictions for Leftovers = {st.session_state.leftovers}')
+  ax.set_title(f'FoodNet Top {n} Predictions = {st.session_state.leftovers}')
   # mpld3.show()
+  plt.xticks(rotation=45, ha='right')
   fig = plt.gcf()
 
   return fig
@@ -243,13 +244,16 @@ model, yum = load_model('fastfood.pth')
 if add_selectbox == "FoodNet Recommender":
     st.title("FoodNet 🍔")
     st.write("Search for similar food ingredients. Select two or more ingredients to find complementary ingredients.")
-    st.multiselect("Type or select food ingredients", list(yum.key_to_index.keys()), default=['bread', 'lettuce'], key="leftovers")
+    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")
 
     ## Slider ##
     st.slider("Number of recommendations to show", min_value=1, max_value=100, value=5, step=1, key='top_n')
 
     ## Get food recommendation ##
-    out = recommend_ingredients(yum, st.session_state.leftovers, n=st.session_state.top_n)
+    ingredients_no_space = [x.replace(' ','_') for x in st.session_state.get('leftovers')]
+    out = recommend_ingredients(yum, ingredients_no_space, n=st.session_state.top_n)
     names = [o[0] for o in out]
     probs = [o[1] for o in out]