please

app.py

@@ -30,47 +30,40 @@ def get_pdf_text(pdf_docs):
 # 아래 텍스트 추출 함수를 작성
 
 def get_text_file(docs):
-    def get_text_file(docs):
-        temp_dir = tempfile.TemporaryDirectory()  # 임시 디렉토리를 생성합니다.
-        temp_filepath = os.path.join(temp_dir.name, "text_file.txt")  # 임시 파일 경로를 생성합니다.
-        with open(temp_filepath, "w", encoding="utf-8") as f:  # 임시 파일을 텍스트 쓰기 모드로 엽니다.
-            f.write(docs)  # 텍스트 문서의 내용을 임시 파일에 씁니다.
-        with open(temp_filepath, "r", encoding="utf-8") as f:  # 임시 파일을 텍스트 읽기 모드로 엽니다.
-            text_content = f.read()  # 텍스트 파일의 내용을 읽습니다.
-        return text_content  # 추출한 텍스트를 반환합니다.
-
+    temp_dir = tempfile.TemporaryDirectory()
+    temp_filepath = os.path.join(temp_dir.name, text_docs.name)
+    with open(temp_filepath, "wb") as f:
+        f.write(text_docs.getvalue())
+    text_loader = TextLoader(temp_filepath)
+    text_content = text_loader.load()
+    temp_dir.cleanup()
+    return text_content
 
 def get_csv_file(docs):
-    temp_dir = tempfile.TemporaryDirectory()  # 임시 디렉토리를 생성합니다.
-    temp_filepath = os.path.join(temp_dir.name, "csv_file.csv")  # 임시 파일 경로를 생성합니다.
-
-    # 텍스트 데이터를 CSV 파일에 쓰기
-    with open(temp_filepath, "w", newline="", encoding="utf-8") as csv_file:
-        csv_writer = csv.writer(csv_file)
-        for line in docs:
-            # 여기서 line은 CSV 파일의 한 행을 나타냅니다.
-            csv_writer.writerow(line.split(','))  # 예시: 쉼표로 구분된 데이터를 사용
+    temp_dir = tempfile.TemporaryDirectory()
+    temp_filepath = os.path.join(temp_dir.name, csv_docs.name)
+    with open(temp_filepath, "wb") as f:
+        f.write(csv_docs.getvalue())
+    csv_loader = CSVLoader(temp_filepath)
+    csv_content = csv_loader.load()
+    temp_dir.cleanup()
+    return csv_content
 
-    # CSV 파일에서 데이터 읽기
     with open(temp_filepath, "r", encoding="utf-8") as csv_file:
         csv_reader = csv.reader(csv_file)
         csv_data = [row for row in csv_reader]
 
-    return csv_data  # 읽어온 CSV 데이터를 반환합니다.
-
-def get_json_file(docs):
-    temp_dir = tempfile.TemporaryDirectory()  # 임시 디렉토리를 생성합니다.
-    temp_filepath = os.path.join(temp_dir.name, "json_file.json")  # 임시 파일 경로를 생성합니다.
-
-    # 텍스트 데이터를 JSON 파일에 쓰기
-    with open(temp_filepath, "w", encoding="utf-8") as json_file:
-        json.dump(docs, json_file, indent=2)  # 데이터를 JSON 파일에 쓰기
+    return csv_data
 
-    # JSON 파일에서 데이터 읽기
-    with open(temp_filepath, "r", encoding="utf-8") as json_file:
-        json_data = json.load(json_file)
+def get_json_file(json_docs):
+    with tempfile.TemporaryDirectory() as temp_dir:
+        temp_filepath = os.path.join(temp_dir, "temp_file.json")
+        with open(temp_filepath, "wb") as f:
+            f.write(json_docs.getvalue())
+        json_loader = JSONLoader(temp_filepath)
+        json_content = json_loader.load()
 
-    return json_data  # 읽어온 JSON 데이터를 반환합니다.
+    return json_content
 
 
 # 문서들을 처리하여 텍스트 청크로 나누는 함수입니다.