Update app.py

app.py

@@ -218,9 +218,9 @@ def get_user_emb(embs, ys):
 
 def pluck_img(user_id, user_emb):
     print(user_id, 'user_id')
-    not_rated_rows = prevs_df[[i[1]['user:rating'].get(user_id, None) == None for i in prevs_df.iterrows()]]
+    not_rated_rows = prevs_df[[i[1]['user:rating'].get(user_id, 'gone') == 'gone' for i in prevs_df.iterrows()]]
     while len(not_rated_rows) == 0:
-        not_rated_rows = prevs_df[[i[1]['user:rating'].get(user_id, None) == None for i in prevs_df.iterrows()]]
+        not_rated_rows = prevs_df[[i[1]['user:rating'].get(user_id, 'gone') == 'gone' for i in prevs_df.iterrows()]]
         time.sleep(.01)
         print(f'user {user_id} has rated all rows')
     # TODO optimize this lol
@@ -274,8 +274,8 @@ def background_next_image():
             else:
                 # If it fails, inform the user.
                 print("Error: %s file not found" % oldest_path)
-            # only keep 50 images & embeddings & ips, then remove oldest
-            prevs_df = prevs_df.iloc[1:]
+            # only keep 50 images & embeddings & ips, then remove oldest besides calibrating
+            prevs_df = pd.concat((prevs_df.iloc[:6], prevs_df.iloc[7:]))
     
 
 def pluck_embs_ys(user_id):