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
 
 
 # 문서들을 처리하여 텍스트 청크로 나누는 함수입니다.