Update app.py

app.py

@@ -339,13 +339,9 @@ def choose(img, choice, calibrate_prompts, user_id, request: gr.Request):
         print('NSFW -- choice is disliked')
         choice = 0
     
-    # TODO clean up
-    old_d = prevs_df.loc[[p.split('/')[-1] in img for p in prevs_df['paths'].to_list()], 'user:rating'][0]
-    old_d[user_id] = choice
-    prevs_df.loc[[p.split('/')[-1] in img for p in prevs_df['paths'].to_list()], 'user:rating'][0] = old_d
-    prevs_df.loc[[p.split('/')[-1] in img for p in prevs_df['paths'].to_list()], 'latest_user_to_rate'] = [user_id]
-    print('full_df, prevs_df', prevs_df, prevs_df['latest_user_to_rate'])
-    
+    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, 'latest_user_to_rate'] = [user_id]
     img, calibrate_prompts = next_image(calibrate_prompts, user_id)
     return img, calibrate_prompts