temporal display of prompt id

pages/Gallery.py

@@ -87,11 +87,6 @@ class GalleryApp:
         edges = []
 
         for idx in items.index:
-            # if items.loc[idx, 'modelVersion_id'] in st.session_state.selected_dict.get(items.loc[idx, 'prompt_id'], 0):
-            #     opacity = 0.2
-            # else:
-            #     opacity = 1.0
-
             nodes.append(Node(id=items.loc[idx, 'image_id'],
                               # label=str(items.loc[idx, 'model_name']),
                               # title=f"model name: {items.loc[idx, 'model_name']}\nmodelVersion name: {items.loc[idx, 'modelVersion_name']}\nclip score: {items.loc[idx, 'clip_score']}\nmcos score: {items.loc[idx, 'mcos_score']}\npopularity: {items.loc[idx, 'model_download_count']}",
@@ -137,7 +132,7 @@ class GalleryApp:
                 st.caption("`Source: Parti-prompts`")
 
             # show image metadata
-            image_metadatas = ['prompt', 'negativePrompt', 'sampler', 'cfgScale', 'size', 'seed']
+            image_metadatas = ['prompt_id', 'prompt', 'negativePrompt', 'sampler', 'cfgScale', 'size', 'seed']
             for key in image_metadatas:
                 label = ' '.join(key.split('_')).capitalize()
                 st.write(f"**{label}**")
@@ -286,14 +281,6 @@ class GalleryApp:
 
                     if has_selection:
                         checkout = st.button('Check out selections ➡️', use_container_width=True, type='primary', on_click=self.switch_to_checkout, args=(tag, selected_prompt))
-                        # if checkout:
-                        #     # add focus to session state
-                        #     st.session_state.gallery_focus['tag'] = tag
-                        #     st.session_state.gallery_focus['prompt'] = selected_prompt
-                        #
-                        #     st.session_state.gallery_state = 'check out'
-                        #     # print(st.session_state.gallery_state)
-                        #     st.rerun()
                     else:
                         st.button(':orange[👇 **Select images you like below**]', disabled=True, use_container_width=True)
                 try:
@@ -326,10 +313,6 @@ class GalleryApp:
 
     def graph_mode(self, prompt_id, items):
         graph_cols = st.columns([3, 1])
-        # prompt = st.chat_input(f"Selected model version ids: {str(st.session_state.selected_dict.get(prompt_id, []))}",
-        #                        disabled=False, key=f'{prompt_id}')
-        # if prompt:
-        #     switch_page("ranking")
 
         with graph_cols[0]:
             st.caption(
@@ -378,6 +361,7 @@ class GalleryApp:
                         #     checked = True
                         # else:
                         #     checked = False
+
                     selection_control_button = st.empty()
                     report_control_button = st.empty()
 
@@ -390,7 +374,6 @@ class GalleryApp:
                             self.image_selection_control(item['tag'], item['prompt'], item['prompt_id'],
                                                          modelVersion_id,
                                                          'select')
-                            # st.toast('Image selected.', icon='👍')
 
                         if image_status == 'select' or image_status == 'reselect' or image_status is None:
                             # deselect = st.button('Deselect', key=f'select_{item["prompt_id"]}_{item["modelVersion_id"]}', use_container_width=True)
@@ -406,6 +389,7 @@ class GalleryApp:
                         report = report_control_button.form_submit_button('⚠️Report', use_container_width=True, type='secondary',
                                                        on_click=self.image_selection_control, args=(
                             item['tag'], item['prompt'], item['prompt_id'], item['modelVersion_id'], 'report'))
+
                     # st.write(item)
                     # infos = ['model_name', 'modelVersion_name', 'model_download_count', 'clip_score', 'mcos_score',
                     #          'nsfw_score']
@@ -419,15 +403,7 @@ class GalleryApp:
                 st.info('You can click on and select an image.')
 
     def image_selection_control(self, tag, prompt, prompt_id, modelVersion_id, operation:['select', 'reselect', 'deselect','report']):
-        self.remove_ranking_states(prompt_id)
-
-        # # update the status of the lastest clicktime
-        # query = "UPDATE gallery_clicks SET status= '{}' " \
-        #         "WHERE username = '{}' AND timestamp = '{}' AND prompt_id = '{}' AND modelVersion_id = {} AND clicktime = " \
-        #         "(SELECT MAX(clicktime) WHERE username = '{}' AND timestamp = '{}' AND prompt_id = '{}' AND modelVersion_id = {}" \
-        #     .format(operation,
-        #             st.session_state.user_id[0], st.session_state.user_id[1], prompt_id, modelVersion_id,
-        #             st.session_state.user_id[0], st.session_state.user_id[1], prompt_id, modelVersion_id)
+        # self.remove_ranking_states(prompt_id)
 
         if operation == 'select' or operation == 'reselect':
             st.session_state.selected_dict[prompt_id].append(modelVersion_id)
@@ -516,6 +492,7 @@ class GalleryApp:
                             proceed = st.button('Proceed ➡️', key=f'checkout_proceed_{prompt_id}', use_container_width=True,
                                                 type='primary', disabled=st.session_state.edit_state)
                             if proceed:
+                                self.remove_ranking_states(prompt_id)
                                 st.session_state.gallery_focus['tag'] = tag
                                 st.session_state.gallery_focus['prompt'] = prompt
                                 st.session_state.gallery_state = 'graph'
@@ -631,11 +608,7 @@ def load_tsne_coordinates(items):
     # load tsne coordinates
     tsne_df = pd.read_parquet('./data/feats_tsne.parquet')
 
-    # print(tsne_df['modelVersion_id'].dtype)
-
-    # print('before merge:', items)
     items = items.merge(tsne_df, on=['modelVersion_id', 'prompt_id'], how='left')
-    # print('after merge:', items)
     return items
 
 

pages/Ranking.py

@@ -420,19 +420,6 @@ class RankingApp:
         elif st.session_state.assigned_rank_mode == 'Battle':
             st.session_state.pointer[prompt_id] = {'left': 0, 'right': 1}
 
-# def connect_to_db():
-#     conn = pymysql.connect(
-#         host=os.environ.get('RANKING_DB_HOST'),
-#         port=3306,
-#         database='myRanking',
-#         user=os.environ.get('RANKING_DB_USER'),
-#         password=os.environ.get('RANKING_DB_PASSWORD'),
-#         charset='utf8mb4',
-#         cursorclass=pymysql.cursors.DictCursor
-#     )
-#
-#     return conn
-
 
 if __name__ == "__main__":
     st.set_page_config(page_title="Personal Image Ranking", page_icon="🎖️️", layout="wide")