fix sql writing bug, add auto image complete

auto image complete will based on model download count

pages/Gallery.py

@@ -101,10 +101,6 @@ class GalleryApp:
                       config=config,
                       )
 
-
-
-
-
     def selection_panel(self, items):
         # temperal function
 
@@ -304,8 +300,7 @@ class GalleryApp:
             graph_space = st.empty()
 
             with graph_space.container():
-                with st.spinner('Loading graph...'):
-                    return_value = self.gallery_graph(items)
+                return_value = self.gallery_graph(items)
 
         with graph_cols[1]:
             if return_value:
@@ -344,7 +339,12 @@ class GalleryApp:
                     # st.write(item)
                     infos = ['model_name', 'modelVersion_name', 'model_download_count', 'clip_score', 'mcos_score',
                              'nsfw_score']
-                    st.table(item[infos])
+
+                    infos_df = item[infos]
+                    # rename columns
+                    infos_df = infos_df.rename(index={'model_name': 'Model', 'modelVersion_name': 'Version', 'model_download_count': 'Downloads', 'clip_score': 'Clip Score', 'mcos_score': 'mcos Score', 'nsfw_score': 'NSFW Score'})
+                    st.table(infos_df)
+
                     # for info in infos:
                     #     st.write(f"**{info}**:")
                     #     st.write(item[info])

pages/Ranking.py

@@ -60,6 +60,7 @@ class RankingApp:
                 st.session_state.ranking[prompt_id][st.session_state.counter[prompt_id]][str(items['image_id'][i])] = i
         else:
             # set the index of items to the corresponding ranking value of the image_id
+            print(items['image_id'])
             items.index = items['image_id'].apply(lambda x: st.session_state.ranking[prompt_id][st.session_state.counter[prompt_id]][str(x)])
 
         with elements('dashboard'):
@@ -184,6 +185,10 @@ class RankingApp:
             # print({'image_id': image_id, 'modelVersion_id': modelVersion_id, 'ranking': ranking, "user_name": st.session_state.user_id[0], "timestamp": st.session_state.user_id[1]})
             # ranking_dataset = ranking_dataset.add_item({'image_id': image_id, 'modelVersion_id': modelVersion_id, 'ranking': ranking, "user_name": st.session_state.user_id[0], "timestamp": st.session_state.user_id[1]})
 
+            # remove the old ranking if exists
+            query = "DELETE FROM rankings WHERE image_id = %s AND user_name = %s AND timestamp = %s"
+            curser.execute(query, (image_id, st.session_state.user_id[0], st.session_state.user_id[1]))
+
             query = "INSERT INTO rankings (image_id, modelVersion_id, ranking, user_name, timestamp) VALUES (%s, %s, %s, %s, %s)"
             curser.execute(query, (image_id, modelVersion_id, ranking, st.session_state.user_id[0], st.session_state.user_id[1]))
 
@@ -250,7 +255,16 @@ if __name__ == "__main__":
             for key, value in st.session_state.selected_dict.items():
                 # promptBook_selected = promptBook_selected.append(promptBook[(promptBook['prompt_id'] == key) & (promptBook['modelVersion_id'].isin(value))])
                 # replace append with pd.concat
-                promptBook_selected = pd.concat([promptBook_selected, promptBook[(promptBook['prompt_id'] == key) & (promptBook['modelVersion_id'].isin(value))]])
+                user_selections = promptBook[(promptBook['prompt_id'] == key) & (promptBook['modelVersion_id'].isin(value))]
+
+                # auto complete the selection with random images
+                residual = len(user_selections) % 4
+                if residual != 0:
+                    # select 4-residual items from the promptbook outside the user_selections
+                    npc = promptBook[(promptBook['prompt_id'] == key) & (~promptBook['modelVersion_id'].isin(value))].sort_values(by=['model_download_count'], ascending=False).reset_index(drop=True).iloc[:4-residual]
+                    user_selections = pd.concat([user_selections, npc])
+
+                promptBook_selected = pd.concat([promptBook_selected, user_selections])
             promptBook_selected = promptBook_selected.reset_index(drop=True)
             # st.write(promptBook_selected)
             images_endpoint = "https://modelcofferbucket.s3-accelerate.amazonaws.com/"