updated app.py

app.py

@@ -3,8 +3,9 @@ import os
 import copy
 import sqlite3
 import operator
-from fastapi import FastAPI, Query
 from math import ceil
+from fastapi import FastAPI, Query
+from contextlib import asynccontextmanager
 from datetime import datetime, timedelta
 from collections import defaultdict
 from langchain_openai import ChatOpenAI
@@ -16,14 +17,6 @@ from langgraph.graph import StateGraph, START, END
 from langgraph.constants import Send
 from fastapi.responses import HTMLResponse
 
-
-# Initialize FastAPI app
-app = FastAPI(
-    title="JEE Roadmap Planner API",
-    description="API for managing and analyzing JEE Roadmaps",
-    version="1.0.0"
-)
-
 # Session State replacement
 session_state = {
     "data": None,
@@ -35,7 +28,8 @@ session_state = {
 
 
 # AGENT 1
-def load_initial_data():
+@asynccontextmanager
+async def lifespan(app: FastAPI):
     try:
         with open('fourdayRoadmap.json', 'r') as file:
             data = json.load(file)
@@ -47,6 +41,15 @@ def load_initial_data():
     except Exception as e:
         return False
 
+# Initialize FastAPI app
+app = FastAPI(
+    title="JEE Roadmap Planner API",
+    description="API for managing and analyzing JEE Roadmaps",
+    version="1.0.0",
+    lifespan=lifespan
+)
+
+
 # Function to mark tasks as incomplete
 def process_task_completion_data():
     data = session_state.data