Update app.py

app.py

@@ -1,10 +1,6 @@
 import streamlit as st
 from dotenv import load_dotenv
 from PyPDF2 import PdfReader
-from langchain.document_loaders import DirectoryLoader
-from langchain.document_loaders import TextLoader
-from langchain.document_loaders.csv_loader import CSVLoader
-from langchain.document_loaders import JSONLoader
 from langchain.text_splitter import CharacterTextSplitter, RecursiveCharacterTextSplitter
 from langchain.embeddings import OpenAIEmbeddings, HuggingFaceInstructEmbeddings
 from langchain.vectorstores import FAISS, Chroma
@@ -18,6 +14,8 @@ from langchain.document_loaders import PyPDFLoader, TextLoader, JSONLoader, CSVL
 import tempfile # 임시 파일을 생성하기 위한 라이브러리입니다.
 import os
 
+from langchain.document_loaders import DirectoryLoader
+
 
 # PDF 문서로부터 텍스트를 추출하는 함수입니다.
 def get_pdf_text(pdf_docs):
@@ -36,7 +34,7 @@ def get_text_file(docs):
     temp_dir = tempfile.TemporaryDirectory() # 임시 디렉토리를 생성합니다.
     temp_filepath = os.path.join(temp_dir.name, docs.name) # 임시 파일 경로를 생성합니다.
     with open(temp_filepath, "wb") as f:  # 임시 파일을 바이너리 쓰기 모드로 엽니다.
-        f.write(docs.getvalue()) # PDF 문서의 내용을 임시 파일에 씁니다.
+        f.write(docs.getvalue()) # text 파일의 내용을 임시 파일에 씁니다.
     loader = DirectoryLoader(temp_dir.name, glob=docs.name, loader_cls=TextLoader)
     data = loader.load()
     return data
@@ -45,8 +43,9 @@ def get_csv_file(docs):
     temp_dir = tempfile.TemporaryDirectory() # 임시 디렉토리를 생성합니다.
     temp_filepath = os.path.join(temp_dir.name, docs.name) # 임시 파일 경로를 생성합니다.
     with open(temp_filepath, "wb") as f:  # 임시 파일을 바이너리 쓰기 모드로 엽니다.
-        f.write(docs.getvalue()) # PDF 문서의 내용을 임시 파일에 씁니다.
-    loader = CSVLoader(temp_filepath)
+        f.write(docs.getvalue()) # csv 파일의 내용을 임시 파일에 씁니다.
+    #loader = CSVLoader(file_path=temp_filepath)
+    loader = DirectoryLoader(temp_dir.name, glob=docs.name, loader_cls=CSVLoader)
     data = loader.load()
     return data
 
@@ -54,7 +53,7 @@ def get_json_file(docs):
     temp_dir = tempfile.TemporaryDirectory() # 임시 디렉토리를 생성합니다.
     temp_filepath = os.path.join(temp_dir.name, docs.name) # 임시 파일 경로를 생성합니다.
     with open(temp_filepath, "wb") as f:  # 임시 파일을 바이너리 쓰기 모드로 엽니다.
-        f.write(docs.getvalue()) # PDF 문서의 내용을 임시 파일에 씁니다.
+        f.write(docs.getvalue()) # json 내용을 임시 파일에 씁니다.
     loader = JSONLoader(
         file_path=temp_filepath, 
         jq_schema='.messages[].content',