QOL improvements

app.py

@@ -25,6 +25,8 @@ db_connection = mysql.connector.connect(
 
 db_cursor = db_connection.cursor()
 
+ORG_ID = 731
+
 
 def get_potential_labels():
     # get potential labels from db
@@ -61,6 +63,12 @@ def classify_email(constituent_email):
     return ", ".join(top_labels)
 
 
+def remove_spaces_after_comma(s):
+    parts = s.split(",")
+    parts = [part.strip() for part in parts]
+    return ",".join(parts)
+
+
 # Function to handle saving data
 def save_data(orig_user_email, constituent_email, labels, user_response, current_user):
     # save the data to the database
@@ -80,27 +88,41 @@ def save_data(orig_user_email, constituent_email, labels, user_response, current
 
     db_cursor = db_connection.cursor()
 
+    if current_user == "Sheryl Springer":
+        person_id = 11021
+    elif current_user == "Diane Taylor":
+        person_id = 11023
+    elif current_user == "Ann E. Belyea":
+        person_id = 11025
+    elif current_user == "Marcelo Mejia":
+        person_id = 11027
+    elif current_user == "Rishi Vasudeva":
+        person_id = 11029
+
     try:
         volley = 1
         if orig_user_email != "":
             db_cursor.execute(
-                "INSERT INTO radmap_frog12.messages (app_id, org_id, person_id, communication_method_id, status_id, subject, body, send_date, message_type, volley) VALUES (0, 0, 0, 1, 1, 'Email Classification and Response Tracking', %s, NOW(), 'Email Classification and Response Tracking', %s)",
-                (orig_user_email, volley),
+                "INSERT INTO radmap_frog12.messages (app_id, org_id, person_id, communication_method_id, status_id, subject, body, send_date, message_type, volley) VALUES (345678, %s, %s, 1, 1, 'Email Classification and Response Tracking', %s, NOW(), 'Email Classification and Response Tracking', %s)",
+                (ORG_ID, person_id, orig_user_email, volley),
             )
             volley = 2
 
         db_cursor.execute(
-            "INSERT INTO radmap_frog12.messages (app_id, org_id, person_id, communication_method_id, status_id, subject, body, send_date, message_type, volley) VALUES (0, 0, 0, 1, 1, 'Email Classification and Response Tracking', %s, NOW(), 'Email Classification and Response Tracking', %s)",
-            (constituent_email, volley),
+            "INSERT INTO radmap_frog12.messages (app_id, org_id, person_id, communication_method_id, status_id, subject, body, send_date, message_type, volley) VALUES (345678, %s, 0, 1, 1, 'Email Classification and Response Tracking', %s, NOW(), 'Email Classification and Response Tracking', %s)",
+            (ORG_ID, constituent_email, volley),
         )
 
         db_cursor.execute(
-            "INSERT INTO radmap_frog12.messages (app_id, org_id, person_id, communication_method_id, status_id, subject, body, send_date, message_type, volley) VALUES (0, 0, 0, 1, 1, 'Email Classification and Response Tracking', %s, NOW(), 'Email Classification and Response Tracking', %s)",
-            (user_response, volley + 1),
+            "INSERT INTO radmap_frog12.messages (app_id, org_id, person_id, communication_method_id, status_id, subject, body, send_date, message_type, volley) VALUES (345678, %s, %s, 1, 1, 'Email Classification and Response Tracking', %s, NOW(), 'Email Classification and Response Tracking', %s)",
+            (ORG_ID, person_id, user_response, volley + 1),
         )
 
         # insert a row into the message_categorys_associations table for each valid label in labels with the message_id of the constituent_email
-        labels = labels.split(", ")
+        # if there is a comma, remove all spaces after the comma
+
+        labels = remove_spaces_after_comma(labels)
+        labels = labels.split(",")
         for label in labels:
             label_exists = db_cursor.execute(
                 "SELECT * FROM radmap_frog12.message_categorys WHERE message_category_name = %s",
@@ -176,7 +198,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as app:
 
         with gr.Column():
             classification_output = gr.TextArea(
-                label="Suggested Message Categories (modify as needed)",
+                label="Suggested Message Categories (modify as needed). Separate categories with commas",
                 lines=1,
                 interactive=True,
             )