Update app.py

app.py

@@ -14,11 +14,15 @@ load_dotenv()
 generator = pipeline("text-generation", model="distilgpt2")
 
 # Initialize Salesforce connection using environment variables
-sf = Salesforce(
-    username=os.getenv("SF_USERNAME"),
-    password=os.getenv("SF_PASSWORD"),
-    security_token=os.getenv("SF_SECURITY_TOKEN")
-)
+try:
+    sf = Salesforce(
+        username=os.getenv("SF_USERNAME"),
+        password=os.getenv("SF_PASSWORD"),
+        security_token=os.getenv("SF_SECURITY_TOKEN")
+    )
+except Exception as e:
+    print(f"Error connecting to Salesforce: {str(e)}")
+    sf = None
 
 @app.route('/generate-ai-data', methods=['POST'])
 def generate_ai_data():
@@ -26,6 +30,12 @@ def generate_ai_data():
     Generate AI coaching data and reports based on supervisor and project data.
     This endpoint is called by Salesforce when a new project is created or during daily refresh.
     """
+    if sf is None:
+        return jsonify({
+            "status": "error",
+            "message": "Salesforce connection failed. Check credentials in .env file."
+        }), 500
+
     try:
         data = request.get_json()
         supervisor_id = data['supervisor_id']
@@ -113,4 +123,4 @@ def generate_ai_data():
         }), 500
 
 if __name__ == "__main__":
-    app.run(host="0.0.0.0", port=8080)
+    app.run(host="0.0.0.0", port=7860)