add email stats

app.py

@@ -8,7 +8,7 @@ import re
 import random
 import string
 from transformers import AutoTokenizer
-from tools import toolsInfo
+from tools import toolsInfo, showEmailStats
 
 from dotenv import load_dotenv
 load_dotenv()
@@ -85,9 +85,9 @@ Check with them what they would like to change in the variation.
 
 # Once the mail is finalized. You ask the user for all the missing placeholder values to write the final mail.
 
-# Once the placeholder values are available in the final email, you ask for the recipient email ID.
+# After the placeholder values are available in the final email, you ask for the recipient email ID.
 
-# Once you have the final mail with placeholder values and recipient exact email ID, you send the email to this id.
+# Once you have the final mail with placeholder values and recipient exact email ID, you send this final well formatted mail to the email ID.
 Dont send email until you have received a valid email from user.
 
 # You congratulate the user and ask if he would like to save this email as a template. If he agrees, save this template in Google Sheet.
@@ -155,6 +155,10 @@ st.markdown(
         font-family: 'Courier New', Courier, monospace; /* Monospace font */
     }
 
+    div[aria-label="dialog"] {
+        width: 90vw;
+    }
+
     </style>
     """,
     unsafe_allow_html=True
@@ -195,6 +199,9 @@ if "buttonValue" not in st.session_state:
 if "startMsg" not in st.session_state:
     st.session_state.startMsg = ""
 
+if "emailSent" not in st.session_state:
+    st.session_state.emailSent = False
+
 st.session_state.toolResponseDisplay = {}
 
 
@@ -235,15 +242,6 @@ def __showToolResponse(toolResponseDisplay: dict):
             st.markdown(msg)
 
 
-def __process_stream_chunk(chunk):
-    delta = chunk.choices[0].delta
-    if delta.content:
-        return delta.content
-    elif delta.tool_calls:
-        return delta.tool_calls[0]
-    return None
-
-
 def __addToolCallToMsgs(toolCall: dict):
     st.session_state.messages.append(
         {
@@ -442,3 +440,13 @@ if prompt := (st.chat_input() or st.session_state["buttonValue"] or st.session_s
             "role": "assistant",
             "content": response,
         })
+
+
+@st.dialog("Email Delivery Stats", width="large")
+def emailStats():
+    showEmailStats()
+
+
+with st.sidebar:
+    if st.session_state.emailSent:
+        st.button("Email Delivery Stats", on_click=emailStats)

requirements.txt

@@ -5,3 +5,5 @@ gradio_client
 oauth2client
 gspread
 sendgrid
+plotly
+streamlit_extras

tools.py

@@ -3,16 +3,22 @@ import gspread
 import os
 import json
 from sendgrid import SendGridAPIClient
-from sendgrid.helpers.mail import Mail
+import streamlit as st
+import pandas as pd
+import plotly.express as px
 
 from dotenv import load_dotenv
 load_dotenv()
 
 GCP_JSON_KEY = os.environ.get("GCP_JSON_KEY")
+SENDGRID_API_KEY = os.environ.get("SENDGRID_API_KEY")
+
 MY_EMAIL = "[email protected]"
+MY_NAME = "Ashhar Akhlaque"
 TEMPLATE_COL_NAME = "Template"
 
-sg = SendGridAPIClient(os.environ.get('SENDGRID_API_KEY'))
+print(SENDGRID_API_KEY)
+sg = SendGridAPIClient(SENDGRID_API_KEY)
 
 
 def saveProfileDetailsInGSheet(
@@ -94,21 +100,40 @@ def sendEmail(
     subject: str,
     htmlContent: str
 ):
-    message = Mail(
-        from_email=MY_EMAIL,
-        to_emails=toEmail,
-        subject=subject,
-        html_content=htmlContent,
-    )
+    data = {
+        "personalizations": [{
+            "to": [
+                {
+                    "email": toEmail,
+                }
+            ],
+            "subject": subject
+        }],
+        "from": {
+            "email": MY_EMAIL,
+            "name": MY_NAME
+        },
+        "reply_to": {
+            "email": MY_EMAIL,
+            "name": MY_NAME
+        },
+        "content": [{
+            "type": "text/html",
+            "value": htmlContent
+        }]
+    }
+
     try:
-        sg.send(message)
+        response = sg.client.mail.send.post(request_body=data)
+        print(f"{response.status_code=}")
+        st.session_state["emailSent"] = True
         return {
             "response": "Email sent",
             "display": {
                 "text": f"`Email sent to {toEmail}`",
                 "icon": "icons/mail.png",
             }
-        }
+        }    
     except Exception as e:
         print(e.message)
         return {
@@ -210,3 +235,91 @@ toolsInfo = {
     },
 
 }
+
+
+def showEmailStats():
+    params = {'start_date': '2024-09-07'}
+
+    response = sg.client.stats.get(
+        query_params=params
+    )
+
+    stats = json.loads(response.body)
+
+    # Convert stats to a pandas DataFrame
+    df = pd.DataFrame([
+        {
+            'date': item['date'],
+            **item['stats'][0]['metrics']
+        } for item in stats
+    ])
+    df['date'] = pd.to_datetime(df['date'])
+
+    # Display overall stats
+    st.subheader("Email Campaign Overview")
+
+    col1, col2, col3, col4, col5 = st.columns([1, 1, 1, 1, 4])
+    col1.metric("Total Sent", df['requests'].sum())
+    col2.metric("Delivered", df['delivered'].sum())
+    col3.metric("Opens", df['opens'].sum())
+    col4.metric("Bounces", df['bounces'].sum())
+
+    col1, col2, col3 = st.columns([3, 1, 3])
+
+    # Create line chart for key metrics
+    figLine = px.line(
+        df,
+        x='date',
+        y=['delivered', 'opens', 'unique_opens', 'bounces'],
+        title='Key Metrics Over Time',
+        color_discrete_map={  # Define custom colors
+            'delivered': 'green',
+            'opens': 'blue',
+            'unique_opens': 'purple',
+            'bounces': 'red'
+        },
+        labels={'date': 'Date', 'value': 'Count'},
+        markers=True,
+        height=400,
+    )
+    figLine.update_layout(
+        legend=dict(
+            orientation="h",  # Horizontal orientation
+            yanchor="bottom",
+            y=1.02,  # Position above the chart
+            xanchor="right",
+            x=1,
+            title=None,  # Remove legend title
+        )
+    )
+    col1.plotly_chart(figLine)
+
+    # Create bar chart for email status
+    dfStatus = df[['delivered', 'bounces', 'invalid_emails']].sum().reset_index()
+    dfStatus.columns = ['status', 'count']
+    figBar = px.bar(
+        dfStatus,
+        x='status',
+        y='count',
+        title='Email Status Distribution',
+        color='status',  # Use status for color
+        color_discrete_map={  # Define custom colors
+            'delivered': 'green',
+            'bounces': 'red',
+            'invalid_emails': 'orange'
+        },
+        labels={'status': 'Status', 'count': 'Count'},
+        text='count',
+        height=400,
+    )
+    figBar.update_layout(
+        legend=dict(
+            orientation="h",  # Horizontal orientation
+            yanchor="bottom",
+            y=1.02,  # Position above the chart
+            xanchor="right",
+            x=1,
+            title=None,  # Remove legend title
+        )
+    )
+    col3.plotly_chart(figBar)