Update main.py

main.py

@@ -1,5 +1,6 @@
 from fastapi import FastAPI, File, UploadFile, Form
 import os
+import tempfile
 from langchain_community.document_loaders import PyMuPDFLoader
 from LoadLLM import Loadllm
 from langchain_community.embeddings import HuggingFaceEmbeddings
@@ -17,12 +18,13 @@ async def home():
 @app.post('/PromptBuddy')
 async def PromptLLM(file: UploadFile = File(...), query: str = Form(...)):
 
-    contents = await file.read()
-    with open(file.filename, 'wb') as f:
-        f.write(contents)
+    with tempfile.NamedTemporaryFile(delete=False) as temp_file:  # Create temporary file
+      temp_file_path = temp_file.name
+      with open(temp_file_path, 'wb') as f:
+        shutil.copyfileobj(file.file, f)
     
 
-    loader = PyMuPDFLoader(file_path= file.filename)
+    loader = PyMuPDFLoader(file_path= temp_file_path)
     data = loader.load()
 
     # Create embeddings using Sentence Transformers