Update main.py

main.py

@@ -12,8 +12,8 @@ app = FastAPI(
 
 # Global in-memory state
 global_state = {
-    "vectorstore": None,
-    "rag_chain": None
+    "vectorstore": create_vectorstore(),
+    "rag_chain": create_rag_chain()
 }
 
 @app.get("/")
@@ -27,24 +27,4 @@ def ask(prompt: str):
         return JSONResponse(status_code=400, content={"error": "Please upload and process a PDF first."})
     
     answer = get_response(global_state["rag_chain"], global_state["vectorstore"], prompt)
-    return {"question": prompt, "answer": answer}
-
-# @app.post("/upload/")
-# async def upload_pdf(file: UploadFile = File(...)):
-#     """Endpoint to upload and process a PDF file."""
-#     # Save PDF temporarily
-#     with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as tmp:
-#         tmp_file_path = tmp.name
-#         shutil.copyfileobj(file.file, tmp)
-
-#     try:
-#         chunks = load_and_process_pdf(tmp_file_path)
-#         vectorstore = create_vectorstore(chunks)
-#         rag_chain = create_rag_chain()
-
-#         global_state["vectorstore"] = vectorstore
-#         global_state["rag_chain"] = rag_chain
-
-#         return {"message": "PDF processed successfully", "filename": file.filename}
-#     finally:
-#         os.unlink(tmp_file_path)
+    return {"question": prompt, "answer": answer}