database fix

pages/Gallery.py

@@ -459,22 +459,24 @@ class GalleryApp:
                                 st.session_state.gallery_state = 'graph'
 
                                 print('selected_dict: ', st.session_state.selected_dict)
-
-                                # # save the user selection to database
                                 cursor = GALLERY_CONN.cursor()
-                                st.session_state.epoch['gallery'] += 1
-                                checkouttime = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-                                # for modelVersion_id in st.session_state.selected_dict[prompt_id]:
-                                for key, values in st.session_state.selected_dict.items():
-                                    # print('key: ', key, 'values: ', values)
-                                    key_tag = self.promptBook[self.promptBook['prompt_id'] == key]['tag'].unique()[0]
-                                    for value in values:
-                                        query = "INSERT INTO gallery_selections (username, timestamp, tag, prompt_id, modelVersion_id, checkouttime, epoch) VALUES ('{}', '{}', '{}', '{}', {}, '{}', {})".format(st.session_state.user_id[0], st.session_state.user_id[1], key_tag, key, value, checkouttime, st.session_state.epoch['gallery'])
-                                        print(query)
-                                        cursor.execute(query)
-                                GALLERY_CONN.commit()
                                 cursor.close()
 
+                                # # save the user selection to database
+                                # cursor = GALLERY_CONN.cursor()
+                                # st.session_state.epoch['gallery'] += 1
+                                # checkouttime = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+                                # # for modelVersion_id in st.session_state.selected_dict[prompt_id]:
+                                # for key, values in st.session_state.selected_dict.items():
+                                #     # print('key: ', key, 'values: ', values)
+                                #     key_tag = self.promptBook[self.promptBook['prompt_id'] == key]['tag'].unique()[0]
+                                #     for value in values:
+                                #         query = "INSERT INTO gallery_selections (username, timestamp, tag, prompt_id, modelVersion_id, checkouttime, epoch) VALUES ('{}', '{}', '{}', '{}', {}, '{}', {})".format(st.session_state.user_id[0], st.session_state.user_id[1], key_tag, key, value, checkouttime, st.session_state.epoch['gallery'])
+                                #         print(query)
+                                #         cursor.execute(query)
+                                # GALLERY_CONN.commit()
+                                # cursor.close()
+
                                 # get the largest epoch number of this user and prompt
                                 cursor = GALLERY_CONN.cursor()
                                 db_table = 'battle_results' if st.session_state.assigned_rank_mode=='Battle' else 'sort_results'