Update app.py

app.py

@@ -224,6 +224,7 @@ def pluck_img(user_id, user_emb):
         not_rated_rows = prevs_df[[i[1]['user:rating'].get(user_id, None) == None for i in prevs_df.iterrows()]]
         rated_rows = prevs_df[[i[1]['user:rating'].get(user_id, None) != None for i in prevs_df.iterrows()]]
         time.sleep(.01)
+        print('current user has rated all rows')
     # TODO optimize this lol
     best_sim = -100000
     for i in not_rated_rows.iterrows():
@@ -246,12 +247,14 @@ def background_next_image():
         not_rated_rows = prevs_df[[i[1]['user:rating'] == {' ': ' '} for i in prevs_df.iterrows()]]
         rated_rows = prevs_df[[i[1]['user:rating'] != {' ': ' '} for i in prevs_df.iterrows()]]
         time.sleep(.01)
+        print('all users have 8 or more rows left to rate')
     
     latest_user_id = rated_rows.iloc[-1]['latest_user_to_rate']
     rated_rows = prevs_df[[i[1]['user:rating'].get(latest_user_id, None) is not None for i in prevs_df.iterrows()]]
     while len(rated_rows) < 4:
         rated_rows = prevs_df[[i[1]['user:rating'].get(latest_user_id, None) is not None for i in prevs_df.iterrows()]]
         time.sleep(.01)
+        print('latest user has < 4 rated_rows')
     
     print(latest_user_id)
     embs, ys = pluck_embs_ys(latest_user_id)
@@ -284,6 +287,7 @@ def pluck_embs_ys(user_id):
         not_rated_rows = prevs_df[[i[1]['user:rating'].get(user_id, None) == None for i in prevs_df.iterrows()]]
         rated_rows = prevs_df[[i[1]['user:rating'].get(user_id, None) != None for i in prevs_df.iterrows()]]
         time.sleep(.01)
+        print('current user has 0 not_rated_rows')
     
     embs = rated_rows['embeddings'].to_list()
     ys = [i[user_id] for i in rated_rows['user:rating'].to_list()]
@@ -348,7 +352,7 @@ def choose(img, choice, calibrate_prompts, user_id, request: gr.Request):
         choice = 0
     
     row_mask = [p.split('/')[-1] in img for p in prevs_df['paths'].to_list()]
-    old_d = prevs_df.loc[row_mask, 'user:rating'][0][user_id] = choice
+    prevs_df.loc[row_mask, 'user:rating'][0][user_id] = choice
     prevs_df.loc[row_mask, 'latest_user_to_rate'] = [user_id]
     img, calibrate_prompts = next_image(calibrate_prompts, user_id)
     return img, calibrate_prompts