Update app.py

app.py

@@ -10,6 +10,10 @@ from io import BytesIO
 from simple_salesforce import Salesforce
 from sendgrid import SendGridAPIClient
 from sendgrid.helpers.mail import Mail
+from dotenv import load_dotenv
+
+# Load environment variables from .env file
+load_dotenv()
 
 # Set up logging to capture errors and debug information
 logging.basicConfig(level=logging.INFO)
@@ -24,18 +28,20 @@ SF_SECURITY_TOKEN = os.getenv("SF_SECURITY_TOKEN", "NbUKcTx45azba5HEdntE9YAh")
 SF_DOMAIN = os.getenv("SF_DOMAIN", "login")
 SENDGRID_API_KEY = os.getenv("SENDGRID_API_KEY")
 
-# Validate environment variables
+# Validate required environment variables (exclude SENDGRID_API_KEY)
 required_vars = {
     'SF_USERNAME': SF_USERNAME,
     'SF_PASSWORD': SF_PASSWORD,
-    'SF_SECURITY_TOKEN': SF_SECURITY_TOKEN,
-    'SENDGRID_API_KEY': SENDGRID_API_KEY
+    'SF_SECURITY_TOKEN': SF_SECURITY_TOKEN
 }
 missing_vars = [var for var, value in required_vars.items() if not value]
 if missing_vars:
     logger.error(f"Missing required environment variables: {', '.join(missing_vars)}")
     raise ValueError(f"Missing required environment variables: {', '.join(missing_vars)}")
 
+if not SENDGRID_API_KEY:
+    logger.warning("SENDGRID_API_KEY not set. Email alerts will be disabled.")
+
 # Connect to Salesforce
 try:
     sf = Salesforce(
@@ -311,6 +317,10 @@ def generate_pdf(vendor_scores):
 
 def send_email_alert(vendor_scores, pdf_urls):
     """Send email alerts to vendors with low scores using SendGrid."""
+    if not SENDGRID_API_KEY:
+        logger.info("Skipping email alerts due to missing SENDGRID_API_KEY")
+        return
+
     sg = SendGridAPIClient(SENDGRID_API_KEY)
     for scores in vendor_scores:
         if not scores['alert_flag']:
@@ -402,7 +412,7 @@ def get_dashboard():
         vendor_data = fetch_vendor_logs()
         if not vendor_data:
             logger.warning("No vendor logs found")
-            last_updated = "12:32 PM IST on Monday, May 19, 2025"
+            last_updated = "12:43 PM IST on Monday, May 19, 2025"
             return render_template('index.html', processed_logs=[], last_updated=last_updated)
 
         vendor_scores = calculate_scores(vendor_data)
@@ -444,7 +454,7 @@ def get_dashboard():
             vendor_logs.append(log_entry)
 
         logger.info(f"Rendering dashboard with {len(processed_logs)} processed logs")
-        last_updated = "12:32 PM IST on Monday, May 19, 2025"
+        last_updated = "12:43 PM IST on Monday, May 19, 2025"
         return render_template('index.html', processed_logs=processed_logs, last_updated=last_updated)
     except Exception as e:
         logger.error(f"Error in / endpoint: {str(e)}")