database fully connected

pages/Gallery.py

@@ -459,24 +459,24 @@ class GalleryApp:
                                 st.session_state.gallery_state = 'graph'
 
                                 print('selected_dict: ', st.session_state.selected_dict)
-                                cursor = GALLERY_CONN.cursor()
-                                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()
 
+                                # 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'