adrianpierce commited on
Commit
70c529a
·
1 Parent(s): b3c4c85

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -8
app.py CHANGED
@@ -89,14 +89,16 @@ with st.expander("Search and filter recipes"):
89
  with st.expander("Home bar"):
90
  # input home bar
91
  st.session_state.bar_df_edited = st.data_editor(st.session_state.bar_df, disabled=["ingredients"], hide_index=True)
92
-
93
- ingredients_joined = ingredients.join(st.session_state.bar_df_edited.set_index('ingredients'), on='ingredients')
94
- ingredients_joined.replace({'have': {True: 1, False: 0}}, inplace=True)
95
- pivot = ingredients_joined.groupby('name').agg(sum_needs_ingredients=('have', 'count'), sum_has_ingredients=('have', 'sum')).reset_index()
96
- filter_1 = (pivot['sum_has_ingredients'] > 0) & (pivot['sum_needs_ingredients'] > 0)
97
- filter_2 = pivot['sum_needs_ingredients'] == pivot['sum_has_ingredients']
98
- filter_all = filter_1 & filter_2
99
- st.session_state.filter_bar = pivot.loc[filter_all, 'name'].to_list()
 
 
100
 
101
  # st.button(label='Submit Bar', on_click=submit_bar)
102
  # st.button(label='Reset', on_click=reset_bar)
 
89
  with st.expander("Home bar"):
90
  # input home bar
91
  st.session_state.bar_df_edited = st.data_editor(st.session_state.bar_df, disabled=["ingredients"], hide_index=True)
92
+ if st.session_state.bar_df_edited['have'].sum() == 0:
93
+ st.session_state.filter_bar = recipes['name'].to_list()
94
+ else:
95
+ ingredients_joined = ingredients.join(st.session_state.bar_df_edited.set_index('ingredients'), on='ingredients')
96
+ ingredients_joined.replace({'have': {True: 1, False: 0}}, inplace=True)
97
+ pivot = ingredients_joined.groupby('name').agg(sum_needs_ingredients=('have', 'count'), sum_has_ingredients=('have', 'sum')).reset_index()
98
+ filter_1 = (pivot['sum_has_ingredients'] > 0) & (pivot['sum_needs_ingredients'] > 0)
99
+ filter_2 = pivot['sum_needs_ingredients'] == pivot['sum_has_ingredients']
100
+ filter_all = filter_1 & filter_2
101
+ st.session_state.filter_bar = pivot.loc[filter_all, 'name'].to_list()
102
 
103
  # st.button(label='Submit Bar', on_click=submit_bar)
104
  # st.button(label='Reset', on_click=reset_bar)