interface

.gitignore

@@ -1 +1,2 @@
-__pycache__*
+__pycache__*
+.gradio/

README.md

@@ -2,6 +2,7 @@
 license: mit
 title: The Other Tiger
 sdk: gradio
+sdk_version: 5.31.0
 emoji: 👁
 colorFrom: blue
 colorTo: purple

app.py

@@ -176,13 +176,11 @@ def background_next_image():
     
 def pluck_img(user_id):
     rated_rows = prevs_df[[i[1]['user:rating'].get(user_id, None) is not None for i in prevs_df.iterrows()]]
-    print(rated_rows)
     ems = rated_rows['embeddings'].to_list()
     ys = [i[user_id][0] for i in rated_rows['user:rating'].to_list()]
     user_emb = get_user_emb(ems, ys)
 
     not_rated_rows = prevs_df[[i[1]['user:rating'].get(user_id, 'gone') == 'gone' for i in prevs_df.iterrows()]]
-    print(not_rated_rows)
     while len(not_rated_rows) == 0:
         not_rated_rows = prevs_df[[i[1]['user:rating'].get(user_id, 'gone') == 'gone' for i in prevs_df.iterrows()]]
         time.sleep(.1)
@@ -212,11 +210,10 @@ def next_image(calibrate_prompts, user_id):
         if len(m_calibrate) - len(calibrate_prompts) < 5:
             cal_video = calibrate_prompts.pop(random.randint(0, len(calibrate_prompts)-1))
             image = prevs_df[prevs_df['paths'] == cal_video]['paths'].to_list()[0]
-            return image, calibrate_prompts
         # we switch to just getting media by similarity.
         else:
             image = pluck_img(user_id)
-            return image, calibrate_prompts
+        return image, calibrate_prompts
 
 
 
@@ -269,16 +266,13 @@ def choose(img, choice, calibrate_prompts, user_id, request: gr.Request):
         prevs_df.loc[row_mask, 'latest_user_to_rate'] = [user_id]
     else:
         print('Image apparently removed', img)
-        breakpoint()
     img, calibrate_prompts = next_image(calibrate_prompts, user_id)
     return img, calibrate_prompts
 
-css = '''.gradio-container{
-  max-width: 700px !important;
-  align-items: center;
-  justify-content: center;
-  text-align: center
-}
+css = '''.gradio-container{max-width: 700px !important}
+
+#component-0{display: block; text-align: center; margin-top: 0}
+
 #description{text-align: center}
 #description h1, #description h3{display: block}
 #description p{margin-top: 0}
@@ -330,11 +324,13 @@ document.body.addEventListener('click', function(event) {
 </script>
 '''
 
-with gr.Blocks(css=css, head=js_head) as demo:
+with gr.Blocks(head=js_head, css=css) as demo:
     gr.Markdown('''# The Other Tiger
 ### Generative Recommenders for Exporation of Possible Images
 
-Explore the latent space without text prompts based on your preferences. [rynmurdock.github.io](https://rynmurdock.github.io/).
+Explore the latent space using binary feedback.
+
+[rynmurdock.github.io](https://rynmurdock.github.io/)
     ''', elem_id="description")
     user_id = gr.State()
     # calibration videos -- this is a misnomer now :D
@@ -423,7 +419,7 @@ def encode_space(x):
 # NOTE:
 # media is moved into a random tmp folder so we need to parse filenames carefully.
 # do not have any cases where a file name is the same or could be `in` another filename
-# you also can't use jpegs lmao
+# you also maybe can't use jpegs lmao
 
 # prep our calibration videos
 m_calibrate = glob.glob('image_init/*')
@@ -436,6 +432,8 @@ for im in m_calibrate:
     tmp_df['embeddings'] = [im_emb.detach().to('cpu')]
     tmp_df['user:rating'] = [{' ': ' '}]
     tmp_df['text'] = ['']
+
+    # seems to break things...
     tmp_df['from_user_id'] = [0]
     tmp_df['latest_user_to_rate'] = [0]
     prevs_df = pd.concat((prevs_df, tmp_df))