please

app.py

@@ -30,19 +30,47 @@ def get_pdf_text(pdf_docs):
 # 아래 텍스트 추출 함수를 작성
 
 def get_text_file(docs):
- text_loader = TextLoader(docs) # TextLoader를 사용해 텍스트 파일을 로드합니다.
- text_doc = text_loader.load() # 텍스트를 추출합니다.
- return text_doc
+ 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 # 추출한 텍스트를 반환합니다.
+
 
 def get_csv_file(docs):
- csv_loader = CSVLoader(docs) # CSVLoader를 사용해 CSV 파일을 로드합니다.
- csv_doc = csv_loader.load() # 텍스트를 추출합니다.
- return csv_doc
+ 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(',')) # 예시: 쉼표로 구분된 데이터를 사용
+
+ # 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):
- json_loader = JSONLoader(docs) # JSONLoader를 사용해 JSON 파일을 로드합니다.
- json_doc = json_loader.load() # 텍스트를 추출합니다.
- return json_doc
+ 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 파일에 쓰기
+
+ # JSON 파일에서 데이터 읽기
+ with open(temp_filepath, "r", encoding="utf-8") as json_file:
+ json_data = json.load(json_file)
+
+ return json_data # 읽어온 JSON 데이터를 반환합니다.
 
 
 # 문서들을 처리하여 텍스트 청크로 나누는 함수입니다.