Changes

app.py

@@ -17,7 +17,7 @@ import re
 from sentence_transformers import SentenceTransformer, util
 from sklearn.feature_extraction.text import CountVectorizer
 from sklearn.metrics.pairwise import cosine_similarity
-from io import StringIO 
+import io  
 from spacy.matcher import PhraseMatcher
 from skillNer.general_params import SKILL_DB
 from skillNer.skill_extractor_class import SkillExtractor
@@ -118,24 +118,8 @@ def upload_PDF(file: UploadFile = File(...)):
 @app.post("/uploadJobDescriptionPDF2/")
 def process_pdf_file(file: bytes =  File(...)):
     # Save file locally for processing
-    contents = file.read()
-    with open(file.filename, 'wb') as f:
-        f.write(contents)
-    
+  
+    reserve_pdf_on_memory = io.BytesIO(file)
+    load_pdf = PyPDF2.PdfFileReader(reserve_pdf_on_memory)
     # Process saved file
-    return process_pdf(file.filename, is_local_file=True)
-
-def process_pdf(pdf_source, is_local_file=False):
-    # Process the PDF from URL or local file
-    file = BytesIO(requests.get(pdf_source).content) if not is_local_file else open(pdf_source, 'rb')
-
-    # Extract text from PDF
-    pdf_reader = PyPDF2.PdfFileReader(file)
-    text = ""
-    for page in range(pdf_reader.numPages):
-        text += pdf_reader.getPage(page).extractText()
-    
-    if is_local_file:
-        file.close()
-
-    return {"content": text}
+    return {"content": load_pdf}