rynmurdock commited on
Commit
c0c2a96
·
1 Parent(s): 9278d5e

pluck change

Browse files
Files changed (1) hide show
  1. app.py +6 -0
app.py CHANGED
@@ -179,6 +179,12 @@ def pluck_img(user_id):
179
  not_rated_rows = prevs_df[[i[1]['user:rating'].get(user_id, 'gone') == 'gone' for i in prevs_df.iterrows()]]
180
  time.sleep(.1)
181
  # TODO optimize this lol
 
 
 
 
 
 
182
  best_sim = -10000000
183
  for i in not_rated_rows.iterrows():
184
  # TODO sloppy .to but it is 3am.
 
179
  not_rated_rows = prevs_df[[i[1]['user:rating'].get(user_id, 'gone') == 'gone' for i in prevs_df.iterrows()]]
180
  time.sleep(.1)
181
  # TODO optimize this lol
182
+
183
+ unrated_from_user = not_rated_rows[[i[1]['from_user_id'] == user_id for i in not_rated_rows.iterrows()]]
184
+ if len(unrated_from_user) > 0:
185
+ # NOTE the way I've setup pandas here is so gdm horrible. TODO overhaul
186
+ img = unrated_from_user['paths'].to_list()[0]
187
+
188
  best_sim = -10000000
189
  for i in not_rated_rows.iterrows():
190
  # TODO sloppy .to but it is 3am.