Update app.py

app.py

@@ -7,9 +7,17 @@ from PIL import Image
 import csv
 import os
 import requests
-from simple_salesforce import Salesforce
+import logging
+try:
+    from simple_salesforce import Salesforce
+except ImportError:
+    raise ImportError("The 'simple-salesforce' library is not installed. Run 'pip install simple-salesforce' to install it.")
 from dotenv import load_dotenv
 
+# Set up logging
+logging.basicConfig(level=logging.DEBUG, format='%(asctime)s - %(levelname)s - %(message)s')
+logger = logging.getLogger(__name__)
+
 # Load environment variables for Salesforce credentials
 load_dotenv()
 SF_USERNAME = os.getenv("SF_USERNAME")
@@ -19,6 +27,7 @@ SF_INSTANCE_URL = "https://login.salesforce.com"  # Use https://test.salesforce.
 
 def connect_to_salesforce():
     """Establish connection to Salesforce."""
+    logger.debug("Attempting to connect to Salesforce...")
     try:
         sf = Salesforce(
             username=SF_USERNAME,
@@ -26,8 +35,10 @@ def connect_to_salesforce():
             security_token=SF_SECURITY_TOKEN,
             instance_url=SF_INSTANCE_URL
         )
+        logger.debug("Successfully connected to Salesforce.")
         return sf
     except Exception as e:
+        logger.error(f"Error connecting to Salesforce: {str(e)}")
         return f"Error connecting to Salesforce: {str(e)}"
 
 def fetch_usd_inr_rate():
@@ -41,12 +52,16 @@ def fetch_usd_inr_rate():
     return None
 
 def predict_overrun(planned_cost, actual_spend, category, cement_index, labor_index, project_phase):
+    logger.debug("Starting prediction with inputs: planned_cost=%s, actual_spend=%s, category=%s, cement_index=%s, labor_index=%s, project_phase=%s",
+                 planned_cost, actual_spend, category, cement_index, labor_index, project_phase)
+    
     try:
         planned_cost = float(planned_cost)
         actual_spend = float(actual_spend)
         cement_index = float(cement_index)
         labor_index = float(labor_index)
     except ValueError:
+        logger.error("Invalid input: Inputs must be numeric.")
         return "**Error:** Inputs must be numeric.", None, None, None
 
     # Risk calculation logic
@@ -64,6 +79,7 @@ def predict_overrun(planned_cost, actual_spend, category, cement_index, labor_in
 
     forecast_cost = planned_cost * (1 + total_risk)
     insights = "High risk of overrun" if total_risk > 0.3 else "Risk within acceptable range"
+    logger.debug("Calculation results: total_risk=%s, forecast_cost=%s, insights=%s", total_risk, forecast_cost, insights)
 
     # Chart
     fig, ax = plt.subplots()
@@ -95,26 +111,37 @@ def predict_overrun(planned_cost, actual_spend, category, cement_index, labor_in
         writer.writerow(["Planned Cost", "Actual Spend", "Risk Score", "Forecasted Cost", "Insights"])
         writer.writerow([planned_cost, actual_spend, round(total_risk * 100, 2), round(forecast_cost, 2), insights])
 
+    # Prepare data for Salesforce
+    sf_data = {
+        "Planned_Cost__c": planned_cost,
+        "Actual_Spend__c": actual_spend,
+        "Category__c": category,
+        "Cement_Index__c": cement_index,
+        "Labor_Index__c": labor_index,
+        "Project_Phase__c": project_phase,
+        "Risk_Score__c": round(total_risk * 100, 2),
+        "Forecasted_Cost__c": round(forecast_cost, 2),
+        "Insights__c": insights
+    }
+    logger.debug("Data prepared for Salesforce: %s", sf_data)
+
     # Save to Salesforce
     sf = connect_to_salesforce()
     if isinstance(sf, str):
         salesforce_status = sf  # Error message
+        logger.error("Salesforce connection failed: %s", salesforce_status)
     else:
         try:
-            sf.Project_Budget_Risk__c.create({
-                "Planned_Cost__c": planned_cost,
-                "Actual_Spend__c": actual_spend,
-                "Category__c": category,
-                "Cement_Index__c": cement_index,
-                "Labor_Index__c": labor_index,
-                "Project_Phase__c": project_phase,
-                "Risk_Score__c": round(total_risk * 100, 2),
-                "Forecasted_Cost__c": round(forecast_cost, 2),
-                "Insights__c": insights
-            })
-            salesforce_status = "Successfully saved to Salesforce."
+            result = sf.Project_Budget_Risk__c.create(sf_data)
+            logger.debug("Salesforce create result: %s", result)
+            if result.get('success'):
+                salesforce_status = f"Successfully saved to Salesforce. Record ID: {result['id']}"
+            else:
+                salesforce_status = f"Failed to save to Salesforce: {result.get('errors')}"
+                logger.error(salesforce_status)
         except Exception as e:
-            salesforce_status = f"Error saving to Salesforce: {str(e)}"
+            salesforce_status = f"Error saving to Salesforce: {str(e)}\nData sent: {sf_data}"
+            logger.error(salesforce_status)
 
     usd_inr = fetch_usd_inr_rate()
     fx_note = f"\n**USD to INR Rate:** ₹{usd_inr:.2f}" if usd_inr else "\n(Exchange rate unavailable)"
@@ -145,7 +172,7 @@ interface = gr.Interface(
         gr.File(label="Download PDF"),
         gr.File(label="Download CSV")
     ],
-    title="Budget Overrun Risk Estimator"
+    title="🧮 Budget Overrun Risk Estimator with Salesforce Integration"
 )
 
 if __name__ == "__main__":