Update main.py

main.py

@@ -1,5 +1,4 @@
 from fastapi import FastAPI, File, UploadFile, Form
-import tempfile
 import os
 from langchain_community.document_loaders import PyMuPDFLoader
 from LoadLLM import Loadllm
@@ -17,12 +16,11 @@ async def home():
 
 @app.post('/PromptBuddy')
 async def PromptLLM(file: UploadFile = File(...), query: str = Form(...)):
-    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_path)
+    with open(file, 'w') as file:
+      file.write('pdf_file')
+
+    loader = PyMuPDFLoader(file_path='pdf_file')
     data = loader.load()
 
     # Create embeddings using Sentence Transformers