Update app.py

app.py

@@ -1,4 +1,9 @@
 import os
+from dotenv import load_dotenv
+
+# Load environment variables from .env file
+load_dotenv()
+
 os.environ["CUDA_VISIBLE_DEVICES"] = ""  # Disable GPU usage
 os.environ["TF_ENABLE_ONEDNN_OPTS"] = "0"  # Disable oneDNN optimizations
 
@@ -17,15 +22,28 @@ from datetime import datetime
 import re
 import gradio as gr
 
-from simple_salesforce import Salesforce  # Added for Salesforce integration
+from simple_salesforce import Salesforce, SalesforceAuthenticationFailed
 
-# Salesforce connection setup (replace with your credentials)
-sf = Salesforce(
-    username='YOUR_SF_USERNAME',
-    password='YOUR_SF_PASSWORD',
-    security_token='YOUR_SF_SECURITY_TOKEN',
-    domain='login'  # use 'test' if using sandbox
-)
+# Read Salesforce credentials from environment variables
+SF_USERNAME = os.getenv("SF_USERNAME")
+SF_PASSWORD = os.getenv("SF_PASSWORD")
+SF_SECURITY_TOKEN = os.getenv("SF_SECURITY_TOKEN")
+SF_DOMAIN = os.getenv("SF_DOMAIN") or "login"  # default to 'login' for production
+
+print(f"Salesforce login info: username={SF_USERNAME}, domain={SF_DOMAIN}")
+
+# Salesforce connection with error handling
+try:
+    sf = Salesforce(
+        username=SF_USERNAME,
+        password=SF_PASSWORD,
+        security_token=SF_SECURITY_TOKEN,
+        domain=SF_DOMAIN
+    )
+    print("Salesforce login successful.")
+except SalesforceAuthenticationFailed as e:
+    print(f"Salesforce authentication failed: {e}")
+    sf = None  # To avoid crashing later
 
 # Initialize Hugging Face NER pipeline (force CPU)
 ner_pipeline = pipeline("ner", model="dslim/bert-base-NER", tokenizer="dslim/bert-base-NER", device=-1)
@@ -146,21 +164,25 @@ def process_invoice(pdf_file):
     with open(output_file, "w") as f:
         json.dump([output], f, indent=2)
 
-    # Create record in Salesforce
-    try:
-        sf.Invoice_Record__c.create({
-            "Vendor_Name__c": vendor_name,
-            "Invoice_Amount__c": amount,
-            "Invoice_Date__c": str(invoice_date),
-            "Items_Listed__c": items_listed,
-            "Fraud_Score__c": fraud_score,
-            "Fraud_Reasoning__c": fraud_reasoning,
-            "Flagged__c": fraud_score > 50,
-            "Reviewed_By__c": None,
-            "Status__c": "Flagged" if fraud_score > 50 else "Cleared"
-        })
-    except Exception as e:
-        print(f"Salesforce record creation failed: {e}")
+    # Create record in Salesforce if logged in successfully
+    if sf is not None:
+        try:
+            sf.Invoice_Record__c.create({
+                "Vendor_Name__c": vendor_name,
+                "Invoice_Amount__c": amount,
+                "Invoice_Date__c": str(invoice_date),
+                "Items_Listed__c": items_listed,
+                "Fraud_Score__c": fraud_score,
+                "Fraud_Reasoning__c": fraud_reasoning,
+                "Flagged__c": fraud_score > 50,
+                "Reviewed_By__c": None,
+                "Status__c": "Flagged" if fraud_score > 50 else "Cleared"
+            })
+            print("Salesforce record created successfully.")
+        except Exception as e:
+            print(f"Salesforce record creation failed: {e}")
+    else:
+        print("Salesforce connection not available, skipping record creation.")
 
     return json.dumps(output, indent=2)