Update app.py

app.py

@@ -25,11 +25,16 @@ st.set_page_config(page_title="Delay 🚀", layout="wide")
 
 # Salesforce connection (using environment variables)
 try:
+    sf_instance_url = os.environ.get("SF_INSTANCE_URL")
+    if not sf_instance_url:
+        raise ValueError("SF_INSTANCE_URL environment variable is not set")
+    if "lightning.force.com" in sf_instance_url:
+        logger.warning("SF_INSTANCE_URL contains lightning.force.com; consider using my.salesforce.com for reliable PDF downloads")
     sf = Salesforce(
         username=os.environ.get("SF_USERNAME"),
         password=os.environ.get("SF_PASSWORD"),
         security_token=os.environ.get("SF_SECURITY_TOKEN"),
-        instance_url=os.environ.get("SF_INSTANCE_URL")
+        instance_url=sf_instance_url
     )
 except Exception as e:
     st.error(f"Failed to connect to Salesforce: {str(e)}")
@@ -168,19 +173,23 @@ def save_to_salesforce(input_data, prediction, pdf_buffer):
         query = f"SELECT ContentDocumentId FROM ContentVersion WHERE Id = '{content_version_id}'"
         query_result = sf.query(query)
         if query_result["totalSize"] == 0:
+            logger.error(f"Failed to retrieve ContentDocumentId for ContentVersion {content_version_id}")
             return "Failed to retrieve ContentDocumentId for the ContentVersion"
         content_document_id = query_result["records"][0]["ContentDocumentId"]
 
-        # Construct the correct Salesforce URL for the ContentDocument
-        pdf_url = f"{os.environ.get('SF_INSTANCE_URL')}/sfc/servlet.shepherd/document/download/{content_document_id}"
+        # Construct the Salesforce URL for the ContentDocument
+        pdf_url = f"{sf_instance_url}/sfc/servlet.shepherd/document/download/{content_document_id}"
+        logger.info(f"Generated PDF URL: {pdf_url}")
 
         # Update the Delay_Predictor__c record with the PDF URL
         update_result = sf.Delay_Predictor__c.update(record_id, {"PDF_Report__c": pdf_url})
         if update_result != 204:  # 204 indicates success for updates
+            logger.error(f"Failed to update PDF_Report__c with URL: {pdf_url}")
             return f"Failed to update PDF_Report__c field: {update_result}"
 
         return None
     except Exception as e:
+        logger.error(f"Error saving to Salesforce: {str(e)}")
         return f"Error saving to Salesforce: {str(e)}"
 
 # Input form
@@ -260,6 +269,12 @@ if submit_button:
             chart_html = f"""
             <canvas id="{chart_id}" style="max-height: 200px; max-width: 600px;"></canvas>
             <script src="https://cdn.jsdelivr.net/npm/chart.js"></script>
+            <script>
+                try {{
+                    const ctx = document.getElementById('{chart_id}').getContext('2d');
+                    new Chart(ctx, {json.dumps(chart_config)});
+                }} catch (e) {{
+                    console.errornm/chart.js"></script>
             <script>
                 try {{
                     const ctx = document.getElementById('{chart_id}').getContext('2d');