Update app.py

app.py

@@ -19,16 +19,7 @@ SF_SECURITY_TOKEN = os.getenv('SF_SECURITY_TOKEN', 'your_salesforce_security_tok
 SF_INSTANCE_URL = os.getenv('SF_INSTANCE_URL', 'https://your_instance.salesforce.com')
 
 # Initialize Salesforce connection
-def connect_to_salesforce():
-    try:
-        sf = Salesforce(username=SF_USERNAME, password=SF_PASSWORD, security_token=SF_SECURITY_TOKEN, instance_url=SF_INSTANCE_URL)
-        print("Connected to Salesforce successfully!")
-        return sf
-    except Exception as e:
-        print(f"Error connecting to Salesforce: {str(e)}")
-        return None
-
-sf = connect_to_salesforce()
+sf = Salesforce(username=SF_USERNAME, password=SF_PASSWORD, security_token=SF_SECURITY_TOKEN, instance_url=SF_INSTANCE_URL)
 
 # Function to format high_risk_phases with flag and alert
 def format_high_risk_phases(high_risk_phases):
@@ -100,10 +91,6 @@ def generate_pdf(input_data, prediction, heatmap_fig):
 
 # Function to save data in Salesforce custom object (Delay_Predictor__c)
 def save_to_salesforce(input_data, prediction):
-    if not sf:
-        print("Salesforce connection not established.")
-        return
-
     try:
         # Create record in Delay_Predictor__c object
         record = {
@@ -125,12 +112,9 @@ def save_to_salesforce(input_data, prediction):
             'Weather_Condition_Forecast__c': prediction['weather_condition']
         }
 
-        # Debugging output: Print the record before attempting to insert into Salesforce
-        print(f"Attempting to insert the following record: {record}")
-
         # Insert record into Salesforce
-        result = sf.Delay_Predictor__c.create(record)
-        print(f"Record saved to Salesforce successfully! Record ID: {result['id']}")
+        sf.Delay_Predictor__c.create(record)
+        print("Record saved to Salesforce successfully!")
     except Exception as e:
         print(f"Error saving to Salesforce: {str(e)}")