Update app.py

app.py

@@ -1,5 +1,5 @@
 """
-LabOps Log Analyzer Dashboard with CSV file upload, PDF generation, Salesforce integration, and AMC reminder email alerts
+LabOps Log Analyzer Dashboard with CSV file upload, PDF generation, Salesforce integration, and AMC reminder via Salesforce trigger
 """
 import gradio as gr
 import pandas as pd
@@ -13,9 +13,6 @@ 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')
@@ -33,14 +30,6 @@ except Exception as e:
     logging.error(f"Failed to connect to Salesforce: {str(e)}")
     sf = None
 
-# Email configuration
-SMTP_SERVER = 'smtp.gmail.com'
-SMTP_PORT = 587
-SMTP_USERNAME = 'harishkumarr@sathkrutha.com'
-SMTP_PASSWORD = 'Harish@048'  # Ensure this is an app-specific password
-EMAIL_FROM = "harishkumarr@sathkrutha.com"
-EMAIL_TO = "sanjaybhargavneela@sathkrutha.com"
-
 # Try to import reportlab
 try:
     from reportlab.lib.pagesizes import letter
@@ -129,56 +118,7 @@ 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. Dataframe is empty.")
-        return "No AMC reminder emails sent (no reminders found)."
-
-    try:
-        # Log the reminders for debugging
-        logging.info(f"Found {len(reminders_df)} AMC reminders to send: {reminders_df.to_dict()}")
-
-        # 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') if pd.notna(row['amc_date']) else "N/A"
-
-            # 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 Sanjay Bhargav Neela,
-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,
-Harish Kumar
-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)}"
-
-# Updated Salesforce report creation with correct field API names
+# Updated Salesforce report creation with minimal metadata
 def create_salesforce_reports(df):
     if sf is None:
         return "Salesforce connection not available."
@@ -186,19 +126,16 @@ def create_salesforce_reports(df):
         return "Cannot create reports: 'LabOps Reports' folder not found in Salesforce."
 
     try:
-        # Usage Report (Summary Report)
+        # Usage Report (Tabular Report for simplicity)
         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": "CustomEntity$SmartLog__c",
-                "reportFormat": "SUMMARY",
+                "reportFormat": "TABULAR",
                 "columns": [
                     {"field": "SmartLog__c.Device_Id__c"},
-                    {"field": "SmartLog__c.Usage_Hours__c", "aggregateTypes": ["Sum"]}
-                ],
-                "groupingsDown": [
-                    {"field": "SmartLog__c.Device_Id__c", "sortOrder": "Asc", "dateGranularity": "None"}
+                    {"field": "SmartLog__c.Usage_Hours__c"}
                 ],
                 "folderId": LABOPS_REPORTS_FOLDER_ID
             }
@@ -329,7 +266,7 @@ def detect_anomalies(df):
         logging.error(f"Anomaly detection failed: {str(e)}")
         return f"Anomaly detection failed: {str(e)}"
 
-# AMC reminders
+# AMC reminders (for display purposes only, email handled by Salesforce trigger)
 def check_amc_reminders(df, current_date):
     try:
         if "device_id" not in df.columns or "amc_date" not in df.columns:
@@ -390,8 +327,8 @@ def create_usage_chart(df):
         logging.error(f"Failed to create usage chart: {str(e)}")
         return None
 
-# Generate PDF content
-def generate_pdf_content(summary, preview, anomalies, amc_reminders, insights, email_status):
+# Generate PDF content (removed email_status since email is handled by Salesforce)
+def generate_pdf_content(summary, preview, anomalies, amc_reminders, insights):
     if not reportlab_available:
         return None
     try:
@@ -423,10 +360,6 @@ def generate_pdf_content(summary, preview, anomalies, amc_reminders, insights, e
         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']))
 
@@ -442,13 +375,13 @@ async def process_logs(file_obj):
     try:
         start_time = datetime.now()
         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.", "No emails sent."
+            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."
 
         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 = {
@@ -462,11 +395,13 @@ async def process_logs(file_obj):
         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, None
+            return f"Missing columns: {missing_columns}", 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, None
+            return "No data available.", None, No
+```python
+ne, None, None, None, None, None, None
 
         # Run tasks concurrently
         with ThreadPoolExecutor(max_workers=4) as executor:
@@ -496,15 +431,14 @@ async def process_logs(file_obj):
         preview = "\n".join(preview_lines)
 
         salesforce_result = save_to_salesforce(df, summary, 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)
+        pdf_file = generate_pdf_content(summary, preview, anomalies, amc_reminders, insights)
 
         elapsed_time = (datetime.now() - start_time).total_seconds()
         logging.info(f"Processing completed in {elapsed_time:.2f} seconds")
-        return summary, preview, chart, anomalies, amc_reminders, insights, pdf_file, salesforce_result, report_result, email_status
+        return summary, preview, chart, anomalies, amc_reminders, insights, pdf_file, salesforce_result, report_result
     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, None
+        return f"Error: {str(e)}", None, None, None, None, None, None, None, None
 
 # Gradio Interface
 try:
@@ -518,7 +452,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, generate Salesforce reports, and send AMC reminder emails.")
+        gr.Markdown("Upload a CSV file to analyze, generate Salesforce reports, and trigger AMC reminder emails via Salesforce.")
 
         with gr.Row():
             with gr.Column(scale=1):
@@ -553,10 +487,6 @@ try:
                         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()
@@ -581,8 +511,7 @@ try:
                 insights_output,
                 pdf_output,
                 salesforce_output,
-                report_output,
-                email_output
+                report_output
             ]
         )