Update app.py

app.py

@@ -23,6 +23,8 @@ import shutil
 import os
 from fastapi import FastAPI, File, UploadFile
 
+vector_index = None
+
 app = FastAPI()
 
 @app.get("/")
@@ -43,7 +45,7 @@ def retrieve(query_text):
     if vector_index is None:
         return "Vector index not found. Please upload a file first."
     else:
-        retriever = vector_index.as_retriever(similarity_top_k=3)
+        retriever = app.state.vector_index.as_retriever(similarity_top_k=3)
         result = retriever.retrieve(query_text)
         if result:
             return "\n\n".join([node.node.text for node in result])
@@ -51,7 +53,6 @@ def retrieve(query_text):
 
 tavily_search = TavilySearchResults(max_results=4)
 
-vector_index = None
 
 @app.post("/uploadpdfs")
 async def upload_file(file: UploadFile = File(...)):
@@ -74,10 +75,10 @@ async def upload_file(file: UploadFile = File(...)):
     # Create or update vector index
     embed_model = HuggingFaceEmbedding(model_name="WhereIsAI/UAE-Large-V1")
     if vector_index is None:
-        vector_index = VectorStoreIndex(nodes, embed_model=embed_model)
+        app.state.vector_index = VectorStoreIndex(nodes, embed_model=embed_model)
         message = "New vector index created and file stored."
     else:
-        vector_index.insert_nodes(nodes)
+        app.state.vector_index.insert_nodes(nodes)
         message = "File stored and vector index updated."
 
     return {"message": message, "filename": file.filename}