Update app.py

app.py

@@ -29,22 +29,31 @@ def get_pdf_text(pdf_docs):
 # 아래 텍스트 추출 함수를 작성
 
 def get_text_file(docs):
-    # 텍스트 파일 (.txt)에서 텍스트를 추출하는 함수
-    return [docs.getvalue().decode('utf-8')]
+    if docs.type == 'text/plain':
+        # 텍스트 파일 (.txt)에서 텍스트를 추출하는 함수
+        return [docs.getvalue().decode('utf-8')]
+    else:
+        st.warning("Unsupported file type for get_text_file")
 
 def get_csv_file(docs):
-    # CSV 파일 (.csv)에서 텍스트를 추출하는 함수
-    csv_loader = CSVLoader(docs)
-    csv_data = csv_loader.load()
-    # CSV 파일의 각 행을 문자열로 변환하여 반환
-    return [' '.join(map(str, row)) for row in csv_data]
+    if docs.type == 'text/csv':
+        # CSV 파일 (.csv)에서 텍스트를 추출하는 함수
+        csv_loader = CSVLoader(docs)
+        csv_data = csv_loader.load()
+        # CSV 파일의 각 행을 문자열로 변환하여 반환
+        return [' '.join(map(str, row)) for row in csv_data]
+    else:
+        st.warning("Unsupported file type for get_csv_file")
 
 def get_json_file(docs):
-    # JSON 파일 (.json)에서 텍스트를 추출하는 함수
-    json_loader = JSONLoader(docs)
-    json_data = json_loader.load()
-    # JSON 파일의 각 항목을 문자열로 변환하여 반환
-    return [json.dumps(item) for item in json_data]
+    if docs.type == 'application/json':
+        # JSON 파일 (.json)에서 텍스트를 추출하는 함수
+        json_loader = JSONLoader(docs)
+        json_data = json_loader.load()
+        # JSON 파일의 각 항목을 문자열로 변환하여 반환
+        return [json.dumps(item) for item in json_data]
+    else:
+        st.warning("Unsupported file type for get_json_file")
 
     
 # 문서들을 처리하여 텍스트 청크로 나누는 함수입니다.