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, and Salesforce integration
 """
 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 (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('LabOpsDashboard', SMTP_USERNAME)
-EMAIL_TO = "harishkumarr@sathkrutha.com"
-
 # Try to import reportlab
 try:
     from reportlab.lib.pagesizes import letter
@@ -129,59 +118,6 @@ 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:
@@ -343,25 +279,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 email and display)
+# AMC reminders (identify records for display)
 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.", pd.DataFrame()
+            return "AMC reminders require 'device_id' and 'amc_date' columns."
         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.", reminders
+            return "No AMC reminders due within the next 30 days."
         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), reminders
+        return "\n".join(reminder_lines)
     except Exception as e:
         logging.error(f"AMC reminder generation failed: {str(e)}")
-        return f"AMC reminder generation failed: {str(e)}", pd.DataFrame()
+        return f"AMC reminder generation failed: {str(e)}"
 
 # Dashboard insights
 def generate_dashboard_insights(df, progress=gr.Progress()):
@@ -406,7 +342,7 @@ def create_usage_chart(df, progress=gr.Progress()):
         return None
 
 # Generate PDF content
-def generate_pdf_content(summary, preview, anomalies, amc_reminders, insights, email_status):
+def generate_pdf_content(summary, preview, anomalies, amc_reminders, insights):
     if not reportlab_available:
         return None
     try:
@@ -438,10 +374,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']))
 
@@ -457,13 +389,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.", "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 = {
@@ -477,11 +409,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, 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, None, None, None, None, None, None, None
 
         with ThreadPoolExecutor() as executor:
             future_summary = executor.submit(summarize_logs, df)
@@ -493,8 +425,7 @@ 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, reminders_df = future_amc.result()  # Get both display text and DataFrame
-            amc_reminders = f"AMC Reminders\n{amc_reminders}"
+            amc_reminders = f"AMC Reminders\n{future_amc.result()}"
             insights = f"Dashboard Insights (AI)\n{future_insights.result()}"
             chart = future_chart.result()
             report_result = future_reports.result()
@@ -510,14 +441,13 @@ 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)
-        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)
 
         progress(1.0, "Done!")
-        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:
@@ -531,7 +461,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 and generate Salesforce reports.")
 
         with gr.Row():
             with gr.Column(scale=1):
@@ -566,10 +496,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()
@@ -594,8 +520,7 @@ try:
                 insights_output,
                 pdf_output,
                 salesforce_output,
-                report_output,
-                email_output
+                report_output
             ]
         )