Update app.py

app.py

@@ -3,6 +3,7 @@ import os
 from fastapi import FastAPI, HTTPException
 from pydantic import BaseModel
 from simple_salesforce import Salesforce
+from contextlib import asynccontextmanager
 import requests
 from typing import Optional
 from dotenv import load_dotenv
@@ -11,41 +12,52 @@ from dotenv import load_dotenv
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
-# Log startup for debugging
-logger.info("Starting application initialization")
+# Global Salesforce connection
+sf = None
 
-# Load environment variables
-load_dotenv()
-SF_USERNAME = os.getenv('SF_USERNAME')
-SF_PASSWORD = os.getenv('SF_PASSWORD')
-SF_SECURITY_TOKEN = os.getenv('SF_SECURITY_TOKEN')
+@asynccontextmanager
+async def lifespan(app: FastAPI):
+    """Manage application lifecycle."""
+    global sf
+    logger.info("Starting application initialization")
+    
+    # Load environment variables
+    load_dotenv()
+    SF_USERNAME = os.getenv('SF_USERNAME')
+    SF_PASSWORD = os.getenv('SF_PASSWORD')
+    SF_SECURITY_TOKEN = os.getenv('SF_SECURITY_TOKEN')
 
-# Validate environment variables
-required_vars = {
-    'SF_USERNAME': SF_USERNAME,
-    'SF_PASSWORD': SF_PASSWORD,
-    'SF_SECURITY_TOKEN': SF_SECURITY_TOKEN
-}
-missing_vars = [var for var, value in required_vars.items() if not value]
-if missing_vars:
-    logger.error(f"Missing required environment variables: {', '.join(missing_vars)}")
-    raise ValueError(f"Missing required environment variables: {', '.join(missing_vars)}")
+    # Validate environment variables
+    required_vars = {
+        'SF_USERNAME': SF_USERNAME,
+        'SF_PASSWORD': SF_PASSWORD,
+        'SF_SECURITY_TOKEN': SF_SECURITY_TOKEN
+    }
+    missing_vars = [var for var, value in required_vars.items() if not value]
+    if missing_vars:
+        logger.error(f"Missing required environment variables: {', '.join(missing_vars)}")
+        raise ValueError(f"Missing required environment variables: {', '.join(missing_vars)}")
 
-# Initialize Salesforce connection
-try:
-    sf = Salesforce(
-        username=SF_USERNAME,
-        password=SF_PASSWORD,
-        security_token=SF_SECURITY_TOKEN,
-        instance_url='https://aicoachforsitesupervisors-dev-ed.develop.my.salesforce.com',
-        version='63.0'
-    )
-    logger.info("Successfully connected to Salesforce")
-except Exception as e:
-    logger.error(f"Failed to connect to Salesforce: {str(e)}")
-    sf = None
+    # Initialize Salesforce connection
+    try:
+        sf = Salesforce(
+            username=SF_USERNAME,
+            password=SF_PASSWORD,
+            security_token=SF_SECURITY_TOKEN,
+            instance_url='https://aicoachforsitesupervisors-dev-ed.develop.my.salesforce.com',
+            version='63.0'
+        )
+        logger.info("Successfully connected to Salesforce")
+    except Exception as e:
+        logger.error(f"Failed to connect to Salesforce: {str(e)}")
+        sf = None
+    
+    logger.info("Application initialized successfully")
+    yield
+    logger.info("Shutting down application")
 
-app = FastAPI()
+# Initialize FastAPI app with lifespan
+app = FastAPI(lifespan=lifespan)
 
 # OpenWeatherMap API key
 WEATHER_API_KEY = "60c00e1b8293d3c0482f8d6ca1a37003"
@@ -132,4 +144,11 @@ async def generate_coaching(request: ProjectRequest):
 
 @app.get("/")
 async def root():
-    return {"message": "Welcome to AI Coach API. Use POST /api/generate"}
+    logger.info("Root endpoint accessed")
+    return {"message": "Welcome to AI Coach API. Use POST /api/generate"}
+
+# Prevent direct execution in production
+if __name__ == "__main__":
+    import uvicorn
+    logger.info("Starting Uvicorn server for local development")
+    uvicorn.run(app, host="0.0.0.0", port=8000)