Update app.py

app.py

@@ -9,7 +9,6 @@ import os
 import logging
 from tenacity import retry, stop_after_attempt, wait_fixed, retry_if_exception_type
 from dotenv import load_dotenv
-from contextlib import asynccontextmanager
 
 # Load environment variables from .env file (for local development)
 load_dotenv()
@@ -48,7 +47,7 @@ def init_db():
         logger.error(f"Error initializing SQLite database: {e}")
         raise
 
-# Run database initialization
+# Run database initialization on startup
 init_db()
 
 # Salesforce connection (using environment variables)
@@ -113,23 +112,17 @@ async def sync_to_salesforce():
 
 # Set up scheduler
 scheduler = AsyncIOScheduler()
+scheduler.add_job(sync_to_salesforce, "interval", hours=4)
 
-# Define lifespan handler for startup and shutdown
-@asynccontextmanager
-async def lifespan(app: FastAPI):
-    # Startup
-    scheduler.add_job(sync_to_salesforce, "interval", hours=4)
+@app.on_event("startup")
+async def startup_event():
     scheduler.start()
     logger.debug("Scheduler started successfully.")
-    try:
-        yield
-    finally:
-        # Shutdown
-        scheduler.shutdown()
-        logger.debug("Scheduler shut down successfully.")
-
-# Attach the lifespan handler to the app
-app.lifespan = lifespan
+
+@app.on_event("shutdown")
+async def shutdown_event():
+    scheduler.shutdown()
+    logger.debug("Scheduler shut down successfully.")
 
 @app.get("/", response_class=HTMLResponse)
 async def index(request: Request):