Update app/main.py

app/main.py

@@ -12,8 +12,15 @@ app.mount("/static", StaticFiles(directory="static"), name="static")
 templates = Jinja2Templates(directory="templates")
 
 # Load your RAG data once
-documents, embeddings= load_data()
+# documents, embeddings= load_data()
 
+documents = None
+embeddings = None
+
+@app.on_event("startup")
+def startup_event():
+    global documents, embeddings
+    documents, embeddings = load_data()
 
 @app.get("/", response_class=HTMLResponse)
 async def index(request: Request):
@@ -36,6 +43,8 @@ async def index(request: Request):
 
 
 
+
+
 @app.post("/ask")
 async def ask(query: str = Form(...)):
     retrieved = retrieve_chunks(query, documents, embeddings)