Update app.py

app.py

@@ -32,15 +32,14 @@ def get_text_file(docs):
     text_list = []
 
     # .txt 파일 처리 함수
-    def process_txt_file(filename, file_content):
-        file_extension = filename.split('.')[-1].lower()  # 파일 이름에서 확장자 추출
-        if file_extension == 'txt':  # 추출한 확장자가 'txt'인지 확인
-            text = file_content.decode("utf-8")
-            text_list.append(text)
+    def process_txt_file(file_content):
+        text = file_content.decode("utf-8")
+        text_list.append(text)
 
     # 주어진 파일 목록을 반복하면서 각 파일을 처리
-    for filename, doc in docs:
-        process_txt_file(filename, doc)  # .txt 파일 처리 함수 호출
+    for file_content in docs:
+        pdf_bytes = BytesIO(file_content)
+        process_txt_file(pdf_bytes.read())  # .txt 파일 처리 함수 호출
 
         # 다른 파일 유형(PDF, CSV, JSON 등)에 대한 처리
         # 필요하다면 여기에 추가적인 파일 처리 로직을 구현합니다.
@@ -49,6 +48,7 @@ def get_text_file(docs):
 
 
 
+
 def get_csv_file(docs):
     text_list = []
     for doc in docs: