First iteration

foodnet.py

@@ -240,7 +240,7 @@ def plot_results(names, probs, n=5):
   ax.xaxis.labelpad = 10
   ax.set_title(f'FoodNet Top {n} Predictions = {st.session_state.leftovers}')
   # mpld3.show()
-  plt.xticks(rotation=45, ha='right')
+  # plt.xticks(rotation=45, ha='right')
   fig = plt.gcf()
 
   return fig
@@ -287,44 +287,23 @@ if add_selectbox == "FoodNet Recommender":
     names = [o[0] for o in out]
     probs = [o[1] for o in out]
 
-    # if 'probs' not in st.session_state:
-    #     st.session_state['probs'] = False
+    if 'probs' not in st.session_state:
+        st.session_state['probs'] = False
     
-    # if st.session_state.probs:
-    #     st.table(data=out)
-    # else:
-    #     st.table(data=names)
+    if st.session_state.probs:
+        st.table(data=out)
+    else:
+        st.table(data=names)
         
-    # st.checkbox(label="Show model scores", value=False, key="probs")
-    # ## Plot Results ##
-    # st.checkbox(label="Show results bar chart", value=False, key="plot")
-    # if st.session_state.plot:
-    #     fig = plot_results(names, probs, st.session_state.top_n)
-
-    #     ## Show Plot ##
-    #     st.pyplot(fig)
-    st.selectbox(label="Select dietary restriction", options=('None', 'Kosher', 'Vegetarian'), key="diet")
-    if st.session_state.diet != 'None':
-      if st.session_state.diet == 'Vegetarian':
-        out = [o for o in out if not any(ignore in o[0] for ignore in blacklists.vegitarian)]
-      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)
-
-    for i, name in enumerate(names): 
-      search = py_un.search(type_='photos', query=name)
-      for photo in search.entries:
-        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)
-        break
+    st.checkbox(label="Show model scores", value=False, key="probs")
+    ## Plot Results ##
+    st.checkbox(label="Show results bar chart", value=False, key="plot")
+    if st.session_state.plot:
+        fig = plot_results(names, probs, st.session_state.top_n)
+
+        ## Show Plot ##
+        st.pyplot(fig)
+
 
 elif add_selectbox == "Food Donation Resources":
     st.title('Food Donation Resources')