Update app.py

app.py

@@ -10,7 +10,7 @@ from reportlab.pdfgen import canvas
 from reportlab.lib.utils import ImageReader
 import base64
 from PIL import Image
-from simple_salesforce import Salesforce
+from simple_salesforce import Salesforce, SalesforceMalformedRequest
 
 # ==========================
 # Configuration
@@ -51,13 +51,18 @@ except Exception as e:
 # Salesforce connection (hardcoded creds for testing)
 # ==========================
 def connect_to_salesforce():
-    sf = Salesforce(
-        username="prashanth1ai@safety.com",
-        password="SaiPrash461",
-        security_token="AP4AQnPoidIKPvSvNEfAHyoK",
-        domain="login"
-    )
-    return sf
+    try:
+        sf = Salesforce(
+            username="prashanth1ai@safety.com",
+            password="SaiPrash461",
+            security_token="AP4AQnPoidIKPvSvNEfAHyoK",
+            domain="login"
+        )
+        print("✅ Salesforce connection successful.")
+        return sf
+    except Exception as e:
+        print(f"❌ Salesforce connection failed: {e}")
+        raise
 
 # ==========================
 # Violation Format Update
@@ -70,17 +75,17 @@ def violations_to_text(violations):
     return "\n".join(lines)
 
 def push_report_to_salesforce(score, violations, pdf_filename):
-    sf = connect_to_salesforce()
+    try:
+        sf = connect_to_salesforce()
 
-    violations_count = len(violations)
-    violations_details = violations_to_text(violations)
+        violations_count = len(violations)
+        violations_details = violations_to_text(violations)
 
-    # Upload the PDF to Salesforce Files
-    try:
+        # Upload the PDF to Salesforce Files
         with open(pdf_filename, "rb") as f:
             pdf_data = f.read()
         file_name = os.path.basename(pdf_filename)
-        
+
         # Upload to ContentVersion (Salesforce Files)
         file_upload = sf.ContentVersion.create({
             'Title': file_name,
@@ -89,7 +94,6 @@ def push_report_to_salesforce(score, violations, pdf_filename):
         })
 
         file_id = file_upload.get('id')
-
         # Get the public URL for the uploaded file (direct download link)
         public_url = f"https://{sf.sf_instance}.salesforce.com/sfc/servlet.shepherd/version/download/{file_id}"
 
@@ -103,6 +107,11 @@ def push_report_to_salesforce(score, violations, pdf_filename):
         })
         print(f"✅ Salesforce record created with Id: {record.get('id')}")
         return record.get("id")
+
+    except SalesforceMalformedRequest as e:
+        print(f"❌ Salesforce malformed request: {e}")
+        print(f"Error details: {e.content}")
+        return None
     except Exception as e:
         print(f"❌ Salesforce record creation failed: {e}")
         return None