Update app.py

app.py

@@ -2,7 +2,6 @@
 
 
 # TODO save & restart from (if it exists) dataframe parquet
-# TODO is spaces GPU blocking? Seems significantly slower while it's generating new in background
 import torch
 
 # lol
@@ -164,14 +163,9 @@ def generate(in_im_embs):
 
 #######################
 
-# TODO add to state instead of shared across all
-glob_idx = 0
 
-# TODO
-# We can keep a df of media paths, embeddings, and user ratings. 
-#   We can drop by lowest user ratings to keep enough RAM available when we get too many rows.
-#   We can continuously update by who is most recently active in the background & server as we go, plucking using "has been seen" and similarity
-#   to user embeds
+# TODO only generate ~5 new images ahead from a specific user embedding. Do this by tracking a column of who's embedding it was and
+#    taking the intersection for unrated by that user and from that users' embedding. Then we keep styles less consistent for better variety.
 
 def get_user_emb(embs, ys):
     # handle case where every instance of calibration videos is 'Neither' or 'Like' or 'Dislike'
@@ -247,10 +241,7 @@ def background_next_image():
         time.sleep(.01)
         print('all users have 8 or more rows left to rate')
 
-
-    # TODO set this to run such that it's on a per-user-in-the-db basis; we only want to be ~5 ahead of every user in the db
     user_id_list = set(rated_rows['latest_user_to_rate'].to_list())
-
     for uid in user_id_list:
         rated_rows = prevs_df[[i[1]['user:rating'].get(uid, None) is not None for i in prevs_df.iterrows()]]
         not_rated_rows = prevs_df[[i[1]['user:rating'].get(uid, None) is None for i in prevs_df.iterrows()]]
@@ -297,8 +288,6 @@ def pluck_embs_ys(user_id):
     return embs, ys
 
 def next_image(calibrate_prompts, user_id):
-    global glob_idx
-    glob_idx = glob_idx + 1
     print(prevs_df)
     
     with torch.no_grad():
@@ -476,7 +465,7 @@ Explore the latent space without text prompts based on your preferences. Learn m
 <div style='text-align:center; font-size:14px'>Thanks to @multimodalart for their contributions to the demo, esp. the interface and @maxbittker for feedback.
 </ div>''')
 
-# TODO quite logging
+# TODO quiet logging
 log = logging.getLogger('log_here')
 log.setLevel(logging.ERROR)