Third iteration

foodnet.py

@@ -287,23 +287,44 @@ 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.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
 
 elif add_selectbox == "Food Donation Resources":
     st.title('Food Donation Resources')