Update app.py

app.py

@@ -11,6 +11,8 @@ from langchain.document_loaders import PyPDFLoader, TextLoader, JSONLoader, CSVL
 import tempfile # 임시 파일을 생성하기 위한 라이브러리입니다.
 import os
 from huggingface_hub import hf_hub_download # Hugging Face Hub에서 모델을 다운로드하기 위한 함수입니다.
+#추가로 필요한 것들
+from PyPDF2 import PdfReader
 
 
 # PDF 문서로부터 텍스트를 추출하는 함수입니다.
@@ -28,12 +30,17 @@ def get_pdf_text(pdf_docs):
 def get_text_file(docs):
     text_list = []
     for doc in docs:
-        if hasattr(doc, 'read') and callable(getattr(doc, 'read')):
-            # 파일 객체인지 확인하고 'read' 메서드가 있는지 검사합니다.
-            text = doc.read().decode("utf-8")  # 파일을 읽고 UTF-8로 디코딩하여 문자열로 변환합니다.
-            text_list.append(text)
+        if isinstance(doc, bytes):
+            # Bytes 객체를 사용하여 PdfReader를 초기화합니다.
+            pdf_reader = PdfReader(io.BytesIO(doc))
         else:
-            text_list.append(str(doc))  # 파일 객체가 아닌 경우 문자열로 변환하여 추가합니다.
+            # 파일 객체를 사용하여 PdfReader를 초기화합니다.
+            pdf_reader = PdfReader(doc)
+
+        # 각 페이지의 텍스트를 추출하여 text_list에 추가합니다.
+        for page in pdf_reader.pages:
+            text = page.extract_text()
+            text_list.append(text)
     return text_list