Update app.py

app.py

@@ -107,9 +107,13 @@ def forecast_labour(csv_file):
             "expected_headcount": predictions[0]['headcount'],
             "actual_headcount": int(df['Attendance'].iloc[-1]),
             "Forecast_Next_3_Days__c": predictions,
-            "Project__c": "a01WU00000wklH4YAI"  # Replace with valid Project ID
+            "Project__c": "a01WU00000wklH4YAI"  # Replace with valid Project ID (Check if this field exists or correct it)
         }
 
+        # Check if 'Project__c' or the correct project field exists in Salesforce
+        if 'Project__c' not in result_data:
+            return {"error": "Invalid or missing field for 'Project__c'. Please ensure this field exists in Salesforce."}
+
         salesforce_record = {
             'Trade__c': result_data['Trade'],
             'Shortage_Risk__c': result_data['shortage_risk'],
@@ -117,12 +121,13 @@ def forecast_labour(csv_file):
             'Expected_Headcount__c': result_data['expected_headcount'],
             'Actual_Headcount__c': result_data['actual_headcount'],
             'Forecast_Next_3_Days__c': str(result_data['Forecast_Next_3_Days__c']),
-            'Project_ID__c': result_data['Project__c'],
+            'Project_ID__c': result_data['Project__c'],  # Replace with correct field name if needed
             'Alert_Status__c': 'Normal',
             'Dashboard_Display__c': True,
             'Date__c': df['Date'].max().date().isoformat()
         }
 
+        # Attempt to save the record to Salesforce
         sf_result = save_to_salesforce(salesforce_record)
         result_data.update(sf_result)
 
@@ -146,4 +151,4 @@ def gradio_interface():
     ).launch()
 
 if __name__ == '__main__':
-    gradio_interface()
+    gradio_interface()