style upgrades

app.py

@@ -62,7 +62,7 @@ def classify_email(constituent_email):
 
 
 # Function to handle saving data
-def save_data(orig_user_email, constituent_email, labels, user_response):
+def save_data(orig_user_email, constituent_email, labels, user_response, current_user):
     # save the data to the database
     # orig_user_email should have volley 0
     # constituent_email should have volley 1
@@ -81,19 +81,22 @@ def save_data(orig_user_email, constituent_email, labels, user_response):
     db_cursor = db_connection.cursor()
 
     try:
-        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', 0)",
-            (orig_user_email,),
-        )
+        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', 0)",
+                (orig_user_email,),
+            )
+            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', 1)",
-            (constituent_email,),
+            "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),
         )
 
         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', 2)",
-            (user_response,),
+            "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 a row into the message_categorys_associations table for each valid label in labels with the message_id of the constituent_email
@@ -118,7 +121,7 @@ def save_data(orig_user_email, constituent_email, labels, user_response):
 
         db_connection.commit()
 
-        return "Data successfully saved to database"
+        return "Response successfully saved to database"
 
     except Exception as e:
         print(e)
@@ -135,34 +138,45 @@ auth = [(auth_username, auth_password)]
 
 # Start building the Gradio interface
 # Start building the Gradio interface with two columns
-with gr.Blocks() as app:
+with gr.Blocks(theme=gr.themes.Soft()) as app:
     with gr.Row():
-        gr.Markdown("## Email Classification and Response Tracking")
+        gr.Markdown("## Campaign Messaging Assistant")
 
     with gr.Row():
         with gr.Column():
+            current_user = gr.Dropdown(
+                label="Current User",
+                choices=[
+                    "Sheryl Springer",
+                    "Ann E. Belyea",
+                    "Marcelo Mejia",
+                    "Rishi Vasudeva",
+                    "Diane Taylor",
+                ],
+            )
+
             email_labels_input = gr.Markdown(
-                "## Valid Email Labels\n ### " + ", ".join(potential_labels),
+                "## Message Category Library\n ### " + ", ".join(potential_labels),
             )
 
             original_email_input = gr.TextArea(
                 placeholder="Enter the original email sent by you",
-                label="Your Original Email",
+                label="Your Original Email (if any)",
             )
 
             spacer1 = gr.Label(visible=False)
 
             constituent_response_input = gr.TextArea(
-                placeholder="Enter the constituent's response",
-                label="Constituent's Response",
+                placeholder="Enter the incoming message",
+                label="Incoming Message (may be a response to original email)",
                 lines=15,
             )
 
-            classify_button = gr.Button("Classify Email")
+            classify_button = gr.Button("Process Message", variant="primary")
 
         with gr.Column():
             classification_output = gr.TextArea(
-                label="Current Email Labels",
+                label="Suggested Message Categories (modify as needed)",
                 lines=1,
                 interactive=True,
             )
@@ -171,11 +185,11 @@ with gr.Blocks() as app:
 
             user_response_input = gr.TextArea(
                 placeholder="Enter your response to the constituent",
-                label="Your Response",
+                label="Suggested Response (modify as needed)",
                 lines=25,
             )
 
-            save_button = gr.Button("Save Data")
+            save_button = gr.Button("Save Response", variant="primary")
             save_output = gr.Label(label="Backend Response")
 
     # Define button actions
@@ -192,6 +206,7 @@ with gr.Blocks() as app:
             constituent_response_input,
             classification_output,
             user_response_input,
+            current_user,
         ],
         outputs=save_output,
     )