Update app.py

app.py

@@ -12,6 +12,7 @@ from reportlab.lib.units import inch
 from io import BytesIO
 import base64
 import logging
+import retrying
 
 # ==========================
 # Configuration
@@ -68,6 +69,7 @@ model = load_model()
 # ==========================
 # Salesforce Integration
 # ==========================
+@retrying.retry(stop_max_attempt_number=3, wait_fixed=2000)
 def connect_to_salesforce():
     try:
         sf = Salesforce(**CONFIG["SF_CREDENTIALS"])
@@ -134,7 +136,11 @@ def upload_pdf_to_salesforce(sf, pdf_file, report_id):
             "VersionData": encoded_pdf,
             "FirstPublishLocationId": report_id
         }
+        logger.info(f"Creating ContentVersion for report ID: {report_id}")
         content_version = sf.ContentVersion.create(content_version_data)
+        if not content_version.get('id'):
+            logger.error("ContentVersion creation failed: No ID returned")
+            return ""
         result = sf.query(f"SELECT Id, ContentDocumentId FROM ContentVersion WHERE Id = '{content_version['id']}'")
         if not result['records']:
             logger.error("Failed to retrieve ContentVersion")
@@ -156,19 +162,24 @@ def push_report_to_salesforce(violations, score, pdf_path, pdf_file):
         pdf_url = f"{CONFIG['PUBLIC_URL_BASE']}{os.path.basename(pdf_path)}" if pdf_path else ""
 
         record_data = {
-            "Compliance Score__c": score,
-            "Violations_Found__c": len(violations),
+            "Compliance_Score__c": float(score),  # Ensure correct data type
+            "Violations_Found__c": int(len(violations)),  # Ensure integer
             "Violations_Details__c": violations_text,
             "Status__c": "Pending",
             "PDF_Report_URL__c": pdf_url
         }
+        logger.info(f"Creating Safety_Video_Report__c with data: {record_data}")
         record = sf.Safety_Video_Report__c.create(record_data)
+        if not record.get('id'):
+            logger.error("Safety_Video_Report__c creation failed: No ID returned")
+            return None, ""
         record_id = record["id"]
         logger.info(f"Salesforce record created: {record_id}")
 
         if pdf_file:
             uploaded_url = upload_pdf_to_salesforce(sf, pdf_file, record_id)
             if uploaded_url:
+                logger.info(f"Updating record {record_id} with PDF URL: {uploaded_url}")
                 sf.Safety_Video_Report__c.update(record_id, {"PDF_Report_URL__c": uploaded_url})
                 pdf_url = uploaded_url