fix selection count

pages/1_🖼️_Gallery.py

@@ -67,7 +67,10 @@ class GalleryApp:
                             # show checkbox
                             checked = st.checkbox('Select', key=f'select_{idx + j}', value=check_init)
                             if checked:
-                                st.session_state.selected_dict[prompt_id] = st.session_state.selected_dict.get(prompt_id, []) + [modelVersion_id]
+                                if prompt_id not in st.session_state.selected_dict:
+                                    st.session_state.selected_dict[prompt_id] = []
+                                if modelVersion_id not in st.session_state.selected_dict[prompt_id]:
+                                    st.session_state.selected_dict[prompt_id].append(modelVersion_id)
                             else:
                                 try:
                                     st.session_state.selected_dict[prompt_id].remove(modelVersion_id)
@@ -337,12 +340,12 @@ if __name__ == "__main__":
             switch_page("home")
     else:
         st.write('You have already logged in as ' + st.session_state.user_id[0])
-        roster, promptBook, st.session_state["images_ds"] = load_hf_dataset()
+        roster, promptBook, images_ds = load_hf_dataset()
         # print(promptBook.columns)
 
         # initialize selected_dict
         if 'selected_dict' not in st.session_state:
             st.session_state['selected_dict'] = {}
 
-        app = GalleryApp(promptBook=promptBook, images_ds=st.session_state.images_ds)
+        app = GalleryApp(promptBook=promptBook, images_ds=images_ds)
         app.app()