Update app.py

app.py

@@ -11,10 +11,21 @@ import io
 import time
 import asyncio
 from simple_salesforce import Salesforce
+import smtplib
+from email.mime.text import MIMEText
 
 # Configure logging
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
 
+# Email configuration
+EMAIL_CONFIG = {
+    'sender_email': 'your-email@gmail.com',  # Replace with your Gmail address
+    'app_password': 'your-app-password',    # Replace with your Gmail App Password
+    'recipient_email': 'recipient@example.com',  # Replace with recipient email
+    'smtp_server': 'smtp.gmail.com',
+    'smtp_port': 587
+}
+
 # Salesforce configuration
 try:
     sf = Salesforce(
@@ -200,6 +211,22 @@ def save_to_salesforce(df, reminders_df):
     except Exception as e:
         logging.error(f"Failed to save to Salesforce: {str(e)}")
 
+# Send email alert
+def send_email_alert(subject, body):
+    try:
+        msg = MIMEText(body)
+        msg['Subject'] = subject
+        msg['From'] = EMAIL_CONFIG['sender_email']
+        msg['To'] = EMAIL_CONFIG['recipient_email']
+
+        with smtplib.SMTP(EMAIL_CONFIG['smtp_server'], EMAIL_CONFIG['smtp_port']) as server:
+            server.starttls()
+            server.login(EMAIL_CONFIG['sender_email'], EMAIL_CONFIG['app_password'])
+            server.send_message(msg)
+        logging.info(f"Email alert sent successfully to {EMAIL_CONFIG['recipient_email']}")
+    except Exception as e:
+        logging.error(f"Failed to send email alert: {str(e)}")
+
 # Summarize logs
 def summarize_logs(df):
     try:
@@ -223,7 +250,9 @@ def detect_anomalies(df):
         anomalies = df[df["anomaly"] == -1][["device_id", "usage_hours", "downtime", "timestamp"]]
         if anomalies.empty:
             return "No anomalies detected.", anomalies
-        return "\n".join([f"- Device ID: {row['device_id']}, Usage: {row['usage_hours']}, Downtime: {row['downtime']}, Timestamp: {row['timestamp']}" for _, row in anomalies.head(5).iterrows()]), anomalies
+        anomaly_summary = "\n".join([f"- Device ID: {row['device_id']}, Usage: {row['usage_hours']}, Downtime: {row['downtime']}, Timestamp: {row['timestamp']}" for _, row in anomalies.head(5).iterrows()])
+        send_email_alert("Anomaly Detection Alert", f"Anomalies detected:\n{anomaly_summary}")
+        return anomaly_summary, anomalies
     except Exception as e:
         logging.error(f"Anomaly detection failed: {str(e)}")
         return f"Anomaly detection failed: {str(e)}", pd.DataFrame()
@@ -239,7 +268,9 @@ def check_amc_reminders(df, current_date):
         reminders = df[(df["days_to_amc"] >= 0) & (df["days_to_amc"] <= 30)][["device_id", "log_type", "status", "timestamp", "usage_hours", "downtime", "amc_date"]]
         if reminders.empty:
             return "No AMC reminders due within the next 30 days.", reminders
-        return "\n".join([f"- Device ID: {row['device_id']}, AMC Date: {row['amc_date']}" for _, row in reminders.head(5).iterrows()]), reminders
+        reminder_summary = "\n".join([f"- Device ID: {row['device_id']}, AMC Date: {row['amc_date']}" for _, row in reminders.head(5).iterrows()])
+        send_email_alert("AMC Reminder Alert", f"AMC reminders due within 30 days:\n{reminder_summary}")
+        return reminder_summary, reminders
     except Exception as e:
         logging.error(f"AMC reminder generation failed: {str(e)}")
         return f"AMC reminder generation failed: {str(e)}", pd.DataFrame()