finish phase 2

.gitignore

@@ -1,2 +1,3 @@
 __pycache__
 examples
+.env

app.py

@@ -31,6 +31,7 @@ db_connection = mysql.connector.connect(
 db_cursor = db_connection.cursor()
 
 ORG_ID = 731
+AI_PERSON_ID = 11056
 
 potential_labels = []
 
@@ -42,6 +43,8 @@ examples_prompt = PromptTemplate(
     input_variables=["example"], template="Example:\n\n {example}"
 )
 
+ai_response = ""
+
 
 def get_potential_labels():
     # get potential labels from db
@@ -119,10 +122,14 @@ def classify_email_and_generate_response(representative_email, constituent_email
     import time
 
     start = time.time()
-    out = "GPT4:\n\n" + llm.invoke(formatted_prompt).content
+    ai_out = llm.invoke(formatted_prompt).content
+
     print("GPT4 response generated in", time.time() - start, "seconds")
 
-    return ", ".join(top_labels), out
+    global ai_response
+    ai_response = ai_out
+
+    return ", ".join(top_labels), ai_out
 
 
 def remove_spaces_after_comma(s):
@@ -141,7 +148,8 @@ def get_similar_messages(constituent_email):
     db_cursor = db_connection.cursor()
 
     messages_for_category = db_cursor.execute(
-        "SELECT id, person_id, body FROM radmap_frog12.messages WHERE id IN (SELECT message_id FROM radmap_frog12.message_category_associations)"
+        "SELECT id, person_id, body FROM radmap_frog12.messages WHERE person_id <> %s AND id IN (SELECT message_id FROM radmap_frog12.message_category_associations)",
+        (AI_PERSON_ID,),
     )
 
     messages_for_category = db_cursor.fetchall()
@@ -164,8 +172,8 @@ def get_similar_messages(constituent_email):
 
         while next_message_id:
             next_message = db_cursor.execute(
-                "SELECT id, body FROM radmap_frog12.messages WHERE previous_message_id = %s",
-                (next_message_id,),
+                "SELECT id, body FROM radmap_frog12.messages WHERE previous_message_id = %s AND person_id <> %s",
+                (next_message_id, AI_PERSON_ID),
             )
             next_message = db_cursor.fetchall()
             if not next_message:
@@ -233,28 +241,59 @@ def save_data(orig_user_email, constituent_email, labels, user_response, current
         person_id = 11027
     elif current_user == "Rishi Vasudeva":
         person_id = 11029
+    else:
+        return "You need to select a user to save data"
 
     try:
-        message_id = 0
+        first_message_id = 0
         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, previous_message_id) 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, message_id),
             )
 
-            #
-            message_id = db_cursor.lastrowid
+            first_message_id = db_cursor.lastrowid
+
+            db_cursor.execute(
+                "INSERT INTO radmap_frog12.messages_log (datetime, message_id, app_id, org_id, person_id, communication_method_id, status_id, subject, body, send_date, message_type, previous_message_id) VALUES (NOW(), %s, 345678, %s, %s, 1, 1, 'Email Classification and Response Tracking', %s, NOW(), 'Email Classification and Response Tracking', %s)",
+                (first_message_id, ORG_ID, person_id, orig_user_email, 0),
+            )
 
         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, previous_message_id) VALUES (345678, %s, 0, 1, 1, 'Email Classification and Response Tracking', %s, NOW(), 'Email Classification and Response Tracking', %s)",
-            (ORG_ID, constituent_email, message_id),
+            (ORG_ID, constituent_email, first_message_id),
+        )
+
+        second_message_id = db_cursor.lastrowid
+
+        db_cursor.execute(
+            "INSERT INTO radmap_frog12.messages_log (datetime, message_id, app_id, org_id, person_id, communication_method_id, status_id, subject, body, send_date, message_type, previous_message_id) VALUES (NOW(), %s, 345678, %s, 0, 1, 1, 'Email Classification and Response Tracking', %s, NOW(), 'Email Classification and Response Tracking', %s)",
+            (second_message_id, ORG_ID, constituent_email, first_message_id),
+        )
+
+        # add the ai response to the database
+        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, previous_message_id) VALUES (345678, %s, %s, 1, 1, 'Email Classification and Response Tracking', %s, NOW(), 'Email Classification and Response Tracking', %s)",
+            (ORG_ID, AI_PERSON_ID, ai_response, second_message_id),
         )
 
-        message_id = db_cursor.lastrowid
+        ai_message_id = db_cursor.lastrowid
+
+        db_cursor.execute(
+            "INSERT INTO radmap_frog12.messages_log (datetime, message_id, app_id, org_id, person_id, communication_method_id, status_id, subject, body, send_date, message_type, previous_message_id) VALUES (NOW(), %s, 345678, %s, %s, 1, 1, 'Email Classification and Response Tracking', %s, NOW(), 'Email Classification and Response Tracking', %s)",
+            (ai_message_id, ORG_ID, AI_PERSON_ID, ai_response, second_message_id),
+        )
 
         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, previous_message_id) 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, message_id),
+            (ORG_ID, person_id, user_response, second_message_id),
+        )
+
+        third_message_id = db_cursor.lastrowid
+
+        db_cursor.execute(
+            "INSERT INTO radmap_frog12.messages_log (datetime, message_id, app_id, org_id, person_id, communication_method_id, status_id, subject, body, send_date, message_type, previous_message_id) VALUES (NOW(), %s, 345678, %s, %s, 1, 1, 'Email Classification and Response Tracking', %s, NOW(), 'Email Classification and Response Tracking', %s)",
+            (third_message_id, ORG_ID, person_id, user_response, second_message_id),
         )
 
         # insert a row into the message_categorys_associations table for each valid label in labels with the message_id of the constituent_email
@@ -280,6 +319,18 @@ def save_data(orig_user_email, constituent_email, labels, user_response, current
                     (message_id[0][0], label_exists[0][0]),
                 )
 
+                message_catergory_association_id = db_cursor.lastrowid
+
+                # save to logs
+                db_cursor.execute(
+                    "INSERT INTO radmap_frog12.message_category_associations_log (datetime, message_category_associations_id, message_id, message_category_id) VALUES (NOW(), %s, %s, %s)",
+                    (
+                        message_catergory_association_id,
+                        message_id[0][0],
+                        label_exists[0][0],
+                    ),
+                )
+
         db_connection.commit()
 
         return "Response successfully saved to database"