Update main.py

main.py

@@ -1,4 +1,5 @@
 from fastapi import FastAPI, File, UploadFile, Form
+import tempfile
 from langchain_community.document_loaders import PyMuPDFLoader
 from LoadLLM import Loadllm
 from langchain_community.embeddings import HuggingFaceEmbeddings
@@ -15,9 +16,10 @@ async def home():
 
 @app.post('/PromptBuddy')
 async def PromptLLM(file: UploadFile = File(...), query: str = Form(...)):
-    pdf_name = file.filename
-   # with open(pdf_name, 'wb') as f:
-       # f.write(file.file.read())
+    with tempfile.TemporaryDirectory() as temp_dir:
+      pdf_path = os.path.join(temp_dir, file.filename)
+      with open(pdf_path, 'wb') as f:
+        f.write(file.file.read())
 
     loader = PyMuPDFLoader(file_path=pdf_name)
     data = loader.load()