some db updates

app.py

@@ -44,9 +44,6 @@ potential_labels = get_potential_labels()
 
 # Function to handle the classification
 def classify_email(constituent_email):
-    global potential_labels
-    potential_labels = get_potential_labels()
-
     print("classifying email")
     model_out = classifier_model(constituent_email, potential_labels, multi_label=True)
     print("classification complete")
@@ -100,8 +97,8 @@ def save_data(orig_user_email, constituent_email, labels, user_response):
             label_exists = db_cursor.fetchall()
             if label_exists:
                 db_cursor.execute(
-                    "INSERT INTO message_category_associations (message_id, message_category_id) VALUES ((SELECT id FROM messages WHERE body = %s), (SELECT id FROM message_categorys WHERE message_category_name = %s))",
-                    (constituent_email, label),
+                    "INSERT INTO message_category_associations (message_id, message_category_id) VALUES ((SELECT id FROM messages WHERE body = %s), %i)",
+                    (constituent_email, label_exists[0][0]),
                 )
 
         db_connection.commit()