Update app.py

app.py

@@ -1,5 +1,5 @@
 """
-LabOps Log Analyzer Dashboard with CSV file upload, PDF generation, and Salesforce integration
+LabOps Log Analyzer Dashboard with CSV file upload, PDF generation, Salesforce integration, and AMC reminder email alerts
 """
 import gradio as gr
 import pandas as pd
@@ -13,6 +13,9 @@ from concurrent.futures import ThreadPoolExecutor
 from simple_salesforce import Salesforce
 import os
 import json
+import smtplib
+from email.mime.text import MIMEText
+from email.mime.multipart import MIMEMultipart
 
 # Configure logging
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
@@ -30,6 +33,14 @@ except Exception as e:
     logging.error(f"Failed to connect to Salesforce: {str(e)}")
     sf = None
 
+# Email configuration (using environment variables)
+SMTP_SERVER = os.getenv('SMTP_SERVER', 'smtp.gmail.com')
+SMTP_PORT = int(os.getenv('SMTP_PORT', 587))
+SMTP_USERNAME = os.getenv('harishkumarr@sathkrutha.com')  # e.g., your-email@gmail.com
+SMTP_PASSWORD = os.getenv('Harish@048')  # App-specific password if using Gmail
+EMAIL_FROM = os.getenv('harishkumarr@sathkrutha.com', SMTP_USERNAME)
+EMAIL_TO = "harishkumarr@sathkrutha.com"
+
 # Try to import reportlab
 try:
     from reportlab.lib.pagesizes import letter
@@ -118,6 +129,59 @@ def get_folder_id(folder_name):
 # Cache the folder ID at startup
 LABOPS_REPORTS_FOLDER_ID = get_folder_id('LabOps Reports')
 
+# Send AMC reminder emails
+def send_amc_reminder_emails(reminders_df):
+    if reminders_df.empty:
+        logging.info("No AMC reminders to send via email.")
+        return "No AMC reminder emails sent (no reminders found)."
+
+    if not all([SMTP_USERNAME, SMTP_PASSWORD, EMAIL_FROM]):
+        logging.error("SMTP credentials not configured. Please set SMTP_USERNAME, SMTP_PASSWORD, and EMAIL_FROM environment variables.")
+        return "Failed to send emails: SMTP credentials not configured."
+
+    try:
+        # Set up the SMTP server
+        server = smtplib.SMTP(SMTP_SERVER, SMTP_PORT)
+        server.starttls()
+        server.login(SMTP_USERNAME, SMTP_PASSWORD)
+
+        email_results = []
+        for _, row in reminders_df.iterrows():
+            device_id = row['device_id']
+            amc_date = row['amc_date'].strftime('%Y-%m-%d')
+
+            # Create the email
+            msg = MIMEMultipart()
+            msg['From'] = EMAIL_FROM
+            msg['To'] = EMAIL_TO
+            msg['Subject'] = f"AMC Reminder for Device {device_id}"
+
+            body = f"""
+Dear Harish Kumar,
+
+This is a reminder that the Annual Maintenance Contract (AMC) for the following device is due:
+
+- Device ID: {device_id}
+- AMC Date: {amc_date}
+
+Please schedule the maintenance at your earliest convenience.
+
+Best regards,
+LabOps Team
+"""
+            msg.attach(MIMEText(body, 'plain'))
+
+            # Send the email
+            server.sendmail(EMAIL_FROM, EMAIL_TO, msg.as_string())
+            logging.info(f"AMC reminder email sent for Device ID {device_id} to {EMAIL_TO}")
+            email_results.append(f"Sent AMC reminder for Device ID {device_id}")
+
+        server.quit()
+        return "\n".join(email_results) if email_results else "No emails sent."
+    except Exception as e:
+        logging.error(f"Failed to send AMC reminder emails: {str(e)}")
+        return f"Failed to send AMC reminder emails: {str(e)}"
+
 # Create Salesforce reports (Usage and AMC Reminders)
 def create_salesforce_reports(df):
     if sf is None:
@@ -130,46 +194,25 @@ def create_salesforce_reports(df):
         usage_report_metadata = {
             "reportMetadata": {
                 "name": f"SmartLog_Usage_Report_{datetime.now().strftime('%Y%m%d_%H%M%S')}",
-                "developerName": f"SmartLog_Usage_Report_{datetime.now().strftime('%Y%m%d_%H%M%S')}",
-                "reportType": {
-                    "type": "CustomObject",
-                    "value": "SmartLog__c"
-                },
+                "reportType": "SmartLog__c",
                 "reportFormat": "SUMMARY",
-                "reportBooleanFilter": None,
                 "reportFilters": [
-                    {
-                        "column": "SmartLog__c.Status__c",
-                        "operator": "equals",
-                        "value": "Active"
-                    },
-                    {
-                        "column": "SmartLog__c.Timestamp__c",
-                        "operator": "greaterOrEqual",
-                        "value": "THIS_MONTH"
-                    }
+                    {"column": "Status__c", "operator": "equals", "value": "Active"},
+                    {"column": "Timestamp__c", "operator": "greaterOrEqual", "value": "THIS_MONTH"}
                 ],
-                "aggregates": ["s!SmartLog__c.Usage_Hours__c", "s!SmartLog__c.Downtime__c"],
-                "groupingsDown": [
-                    {
-                        "name": "Device_Id__c",
-                        "field": "SmartLog__c.Device_Id__c",
-                        "sortOrder": "Asc",
-                        "sortAggregate": None,
-                        "dateGranularity": "None"
-                    }
+                "reportColumns": [
+                    {"field": "Device_Id__c"},
+                    {"field": "Log_Type__c"},
+                    {"field": "Status__c"},
+                    {"field": "Timestamp__c"},
+                    {"field": "Usage_Hours__c", "aggregateTypes": ["Sum"]},
+                    {"field": "Downtime__c", "aggregateTypes": ["Sum"]},
+                    {"field": "AMC_Date__c"}
                 ],
-                "detailColumns": [
-                    "SmartLog__c.Device_Id__c",
-                    "SmartLog__c.Log_Type__c",
-                    "SmartLog__c.Status__c",
-                    "SmartLog__c.Timestamp__c",
-                    "SmartLog__c.Usage_Hours__c",
-                    "SmartLog__c.Downtime__c",
-                    "SmartLog__c.AMC_Date__c"
+                "groupingsDown": [
+                    {"field": "Device_Id__c", "sortOrder": "Asc", "dateGranularity": "None"}
                 ],
-                "folderId": LABOPS_REPORTS_FOLDER_ID,
-                "currency": None
+                "folderId": LABOPS_REPORTS_FOLDER_ID
             }
         }
         usage_result = sf.restful('analytics/reports', method='POST', json=usage_report_metadata)
@@ -180,37 +223,19 @@ def create_salesforce_reports(df):
         amc_report_metadata = {
             "reportMetadata": {
                 "name": f"SmartLog_AMC_Reminders_{datetime.now().strftime('%Y%m%d_%H%M%S')}",
-                "developerName": f"SmartLog_AMC_Reminders_{datetime.now().strftime('%Y%m%d_%H%M%S')}",
-                "reportType": {
-                    "type": "CustomObject",
-                    "value": "SmartLog__c"
-                },
+                "reportType": "SmartLog__c",
                 "reportFormat": "TABULAR",
-                "reportBooleanFilter": None,
                 "reportFilters": [
-                    {
-                        "column": "SmartLog__c.Status__c",
-                        "operator": "equals",
-                        "value": "Active"
-                    },
-                    {
-                        "column": "SmartLog__c.AMC_Date__c",
-                        "operator": "greaterOrEqual",
-                        "value": "TODAY"
-                    },
-                    {
-                        "column": "SmartLog__c.AMC_Date__c",
-                        "operator": "lessOrEqual",
-                        "value": "NEXT_N_DAYS:30"
-                    }
+                    {"column": "Status__c", "operator": "equals", "value": "Active"},
+                    {"column": "AMC_Date__c", "operator": "greaterOrEqual", "value": "TODAY"},
+                    {"column": "AMC_Date__c", "operator": "lessOrEqual", "value": "NEXT_N_DAYS:30"}
                 ],
-                "detailColumns": [
-                    "SmartLog__c.Device_Id__c",
-                    "SmartLog__c.AMC_Date__c",
-                    "SmartLog__c.Status__c"
+                "reportColumns": [
+                    {"field": "Device_Id__c"},
+                    {"field": "AMC_Date__c"},
+                    {"field": "Status__c"}
                 ],
-                "folderId": LABOPS_REPORTS_FOLDER_ID,
-                "currency": None
+                "folderId": LABOPS_REPORTS_FOLDER_ID
             }
         }
         amc_result = sf.restful('analytics/reports', method='POST', json=amc_report_metadata)
@@ -234,14 +259,14 @@ def save_to_salesforce(df, summary, anomalies, amc_reminders, insights):
             # Validate and map picklist values
             status = str(row['status'])
             log_type = str(row['log_type'])
-           
+            
             # Map Status__c
             if status not in status_values:
                 status = picklist_mapping['Status__c'].get(status.lower(), status_values[0] if status_values else None)
                 if status is None:
                     logging.warning(f"Skipping record with invalid Status__c: {row['status']}")
                     continue
-           
+            
             # Map Log_Type__c
             if log_type not in log_type_values:
                 log_type = picklist_mapping['Log_Type__c'].get(log_type.lower(), log_type_values[0] if log_type_values else None)
@@ -267,7 +292,7 @@ def save_to_salesforce(df, summary, anomalies, amc_reminders, insights):
                 'AMC_Date__c': amc_date_str
             }
             records.append(record)
-       
+        
         # Bulk insert to reduce API calls
         if records:
             sf.bulk.SmartLog__c.insert(records)
@@ -316,25 +341,25 @@ def detect_anomalies(df, progress=gr.Progress()):
         logging.error(f"Anomaly detection failed: {str(e)}")
         return f"Anomaly detection failed: {str(e)}"
 
-# AMC reminders (identify records for display)
+# AMC reminders (identify records for display and email)
 def check_amc_reminders(df, current_date, progress=gr.Progress()):
     progress(0.6, "Checking AMC reminders...")
     try:
         if "device_id" not in df.columns or "amc_date" not in df.columns:
-            return "AMC reminders require 'device_id' and 'amc_date' columns."
+            return "AMC reminders require 'device_id' and 'amc_date' columns.", pd.DataFrame()
         df["amc_date"] = pd.to_datetime(df["amc_date"], errors='coerce')
         current_date = pd.to_datetime(current_date)
         df["days_to_amc"] = (df["amc_date"] - current_date).dt.days
         reminders = df[(df["days_to_amc"] >= 0) & (df["days_to_amc"] <= 30)][["device_id", "amc_date"]]
         if reminders.empty:
-            return "No AMC reminders due within the next 30 days."
+            return "No AMC reminders due within the next 30 days.", reminders
         reminder_lines = ["Upcoming AMC Reminders:"]
         for _, row in reminders.head(5).iterrows():
             reminder_lines.append(f"- Device ID: {row['device_id']}, AMC Date: {row['amc_date']}")
-        return "\n".join(reminder_lines)
+        return "\n".join(reminder_lines), reminders
     except Exception as e:
         logging.error(f"AMC reminder generation failed: {str(e)}")
-        return f"AMC reminder generation failed: {str(e)}"
+        return f"AMC reminder generation failed: {str(e)}", pd.DataFrame()
 
 # Dashboard insights
 def generate_dashboard_insights(df, progress=gr.Progress()):
@@ -379,7 +404,7 @@ def create_usage_chart(df, progress=gr.Progress()):
         return None
 
 # Generate PDF content
-def generate_pdf_content(summary, preview, anomalies, amc_reminders, insights):
+def generate_pdf_content(summary, preview, anomalies, amc_reminders, insights, email_status):
     if not reportlab_available:
         return None
     try:
@@ -411,6 +436,10 @@ def generate_pdf_content(summary, preview, anomalies, amc_reminders, insights):
         story.append(safe_paragraph(amc_reminders or "No AMC reminders.", styles['Normal']))
         story.append(Spacer(1, 12))
 
+        story.append(Paragraph("Email Notification Status", styles['Heading2']))
+        story.append(safe_paragraph(email_status or "No emails sent.", styles['Normal']))
+        story.append(Spacer(1, 12))
+
         story.append(Paragraph("Dashboard Insights", styles['Heading2']))
         story.append(safe_paragraph(insights or "No insights generated.", styles['Normal']))
 
@@ -426,13 +455,13 @@ async def process_logs(file_obj, progress=gr.Progress()):
     try:
         progress(0, "Starting file processing...")
         if not file_obj:
-            return "No file uploaded.", "No data to preview.", None, "No anomalies detected.", "No AMC reminders.", "No insights generated.", None, "No Salesforce data saved.", "No report created."
+            return "No file uploaded.", "No data to preview.", None, "No anomalies detected.", "No AMC reminders.", "No insights generated.", None, "No Salesforce data saved.", "No report created.", "No emails sent."
 
         file_name = file_obj.name
         logging.info(f"Processing file: {file_name}")
-       
+        
         if not file_name.endswith(".csv"):
-            return "Please upload a CSV file.", "", None, "", "", "", None, "", ""
+            return "Please upload a CSV file.", "", None, "", "", "", None, "", "", ""
 
         required_columns = ["device_id", "log_type", "status", "timestamp", "usage_hours", "downtime", "amc_date"]
         dtypes = {
@@ -446,11 +475,11 @@ async def process_logs(file_obj, progress=gr.Progress()):
         df = pd.read_csv(file_obj, dtype=dtypes)
         missing_columns = [col for col in required_columns if col not in df.columns]
         if missing_columns:
-            return f"Missing columns: {missing_columns}", None, None, None, None, None, None, None, None
+            return f"Missing columns: {missing_columns}", None, None, None, None, None, None, None, None, None
         df["timestamp"] = pd.to_datetime(df["timestamp"], errors='coerce')
         df["amc_date"] = pd.to_datetime(df["amc_date"], errors='coerce')
         if df.empty:
-            return "No data available.", None, None, None, None, None, None, None, None
+            return "No data available.", None, None, None, None, None, None, None, None, None
 
         with ThreadPoolExecutor() as executor:
             future_summary = executor.submit(summarize_logs, df)
@@ -462,7 +491,8 @@ async def process_logs(file_obj, progress=gr.Progress()):
 
             summary = f"Step 1: Summary Report\n{future_summary.result()}"
             anomalies = f"Anomaly Detection\n{future_anomalies.result()}"
-            amc_reminders = f"AMC Reminders\n{future_amc.result()}"
+            amc_reminders, reminders_df = future_amc.result()  # Get both display text and DataFrame
+            amc_reminders = f"AMC Reminders\n{amc_reminders}"
             insights = f"Dashboard Insights (AI)\n{future_insights.result()}"
             chart = future_chart.result()
             report_result = future_reports.result()
@@ -478,13 +508,14 @@ async def process_logs(file_obj, progress=gr.Progress()):
         preview = "\n".join(preview_lines)
 
         salesforce_result = save_to_salesforce(df, summary, anomalies, amc_reminders, insights)
-        pdf_file = generate_pdf_content(summary, preview, anomalies, amc_reminders, insights)
+        email_status = send_amc_reminder_emails(reminders_df)
+        pdf_file = generate_pdf_content(summary, preview, anomalies, amc_reminders, insights, email_status)
 
         progress(1.0, "Done!")
-        return summary, preview, chart, anomalies, amc_reminders, insights, pdf_file, salesforce_result, report_result
+        return summary, preview, chart, anomalies, amc_reminders, insights, pdf_file, salesforce_result, report_result, email_status
     except Exception as e:
         logging.error(f"Failed to process file: {str(e)}")
-        return f"Error: {str(e)}", None, None, None, None, None, None, None, None
+        return f"Error: {str(e)}", None, None, None, None, None, None, None, None, None
 
 # Gradio Interface
 try:
@@ -498,7 +529,7 @@ try:
         .dashboard-section ul {margin: 2px 0; padding-left: 20px;}
     """) as iface:
         gr.Markdown("<h1>LabOps Log Analyzer Dashboard (Hugging Face AI)</h1>")
-        gr.Markdown("Upload a CSV file to analyze and generate Salesforce reports.")
+        gr.Markdown("Upload a CSV file to analyze, generate Salesforce reports, and send AMC reminder emails.")
 
         with gr.Row():
             with gr.Column(scale=1):
@@ -508,39 +539,43 @@ try:
             with gr.Column(scale=2):
                 with gr.Group(elem_classes="dashboard-container"):
                     gr.Markdown("<div class='dashboard-title'>Analysis Results</div>")
-                   
+                    
                     with gr.Group(elem_classes="dashboard-section"):
                         gr.Markdown("### Step 1: Summary Report")
                         summary_output = gr.Markdown()
-                   
+                    
                     with gr.Group(elem_classes="dashboard-section"):
                         gr.Markdown("### Step 2: Log Preview")
                         preview_output = gr.Markdown()
-                       
+                        
                     with gr.Group(elem_classes="dashboard-section"):
                         gr.Markdown("### Step 3: Usage Chart")
                         chart_output = gr.Plot()
-                       
+                        
                     with gr.Group(elem_classes="dashboard-section"):
                         gr.Markdown("### Step 4: Anomaly Detection")
                         anomaly_output = gr.Markdown()
-                       
+                        
                     with gr.Group(elem_classes="dashboard-section"):
                         gr.Markdown("### Step 5: AMC Reminders")
                         amc_output = gr.Markdown()
-                       
+                        
                     with gr.Group(elem_classes="dashboard-section"):
                         gr.Markdown("### Step 6: Insights (AI)")
                         insights_output = gr.Markdown()
-                       
+                        
+                    with gr.Group(elem_classes="dashboard-section"):
+                        gr.Markdown("### Step 7: Email Notification Status")
+                        email_output = gr.Markdown()
+                        
                     with gr.Group(elem_classes="dashboard-section"):
                         gr.Markdown("### Salesforce Integration")
                         salesforce_output = gr.Markdown()
-                       
+                        
                     with gr.Group(elem_classes="dashboard-section"):
                         gr.Markdown("### Salesforce Reports")
                         report_output = gr.Markdown()
-                       
+                        
                     with gr.Group(elem_classes="dashboard-section"):
                         gr.Markdown("### Download Report")
                         pdf_output = gr.File(label="Download Analysis Report as PDF")
@@ -557,7 +592,8 @@ try:
                 insights_output,
                 pdf_output,
                 salesforce_output,
-                report_output
+                report_output,
+                email_output
             ]
         )
 
@@ -574,4 +610,4 @@ if __name__ == "__main__":
     except Exception as e:
         logging.error(f"Failed to launch Gradio interface: {str(e)}")
         print(f"Error launching app: {str(e)}")
-        raise e
+        raise e