Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -24,14 +24,30 @@ def get_pdf_text(pdf_docs):
|
|
| 24 |
|
| 25 |
# 과제
|
| 26 |
# 아래 텍스트 추출 함수를 작성
|
| 27 |
-
def get_text_file(
|
| 28 |
-
|
| 29 |
-
|
| 30 |
-
|
| 31 |
-
|
| 32 |
-
|
| 33 |
-
|
| 34 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 35 |
|
| 36 |
|
| 37 |
# 문서들을 처리하여 텍스트 청크로 나누는 함수입니다.
|
|
|
|
| 24 |
|
| 25 |
# 과제
|
| 26 |
# 아래 텍스트 추출 함수를 작성
|
| 27 |
+
def get_text_file(text_contents):
|
| 28 |
+
temp_dir = tempfile.TemporaryDirectory() # 임시 디렉토리를 생성합니다.
|
| 29 |
+
temp_filepath = os.path.join(temp_dir.name, text_contents.name) # 임시 텍스트 파일 경로를 생성합니다.
|
| 30 |
+
with open(temp_filepath, "w", encoding="utf-8") as f: # 텍스트 파일을 쓰기 모드로 엽니다.
|
| 31 |
+
f.write(text_docs.getvalue()) # 텍스트 데이터를 파일에 씁니다.
|
| 32 |
+
text_loader = TextLoader(temp_filepath) # TextLoader를 사용해 텍스트 파일을 로드합니다.
|
| 33 |
+
text_content = text_loader.load() # 텍스트를 추출합니다.
|
| 34 |
+
return text_content # 추출한 텍스트를 반환합니다.
|
| 35 |
+
|
| 36 |
+
def get_csv_file(csv_datas):
|
| 37 |
+
temp_dir = tempfile.TemporaryDirectory() # 임시 디렉토리를 생성합니다.
|
| 38 |
+
temp_filepath = os.path.join(temp_dir.name, csv_datas.name) # 임시 CSV 파일 경로를 생성합니다.
|
| 39 |
+
with open(temp_filepath, "w", newline="", encoding="utf-8") as csvfile: # CSV 파일을 쓰기 모드로 엽니다.
|
| 40 |
+
csv_loader = CSVLoader(csvfile)
|
| 41 |
+
csv_loader.load(csv_data) # CSV 데이터를 파일에 씁니다.
|
| 42 |
+
return temp_filepath # 생성된 임시 CSV 파일 경로를 반환합니다.
|
| 43 |
+
|
| 44 |
+
def get_json_file(json_data):
|
| 45 |
+
temp_dir = tempfile.TemporaryDirectory() # 임시 디렉토리를 생성합니다.
|
| 46 |
+
temp_filepath = os.path.join(temp_dir.name, json_datas.name) # 임시 JSON 파일 경로를 생성합니다.
|
| 47 |
+
with open(temp_filepath, "w", encoding="utf-8") as jsonfile: # JSON 파일을 쓰기 모드로 엽니다.
|
| 48 |
+
json_loader = JSONLoader(jsonfile)
|
| 49 |
+
json_loader.load(json_data) # JSON 데이터를 파일에 씁니다.
|
| 50 |
+
return temp_filepath # 생성된 임시 JSON 파일 경로를 반환합니다
|
| 51 |
|
| 52 |
|
| 53 |
# 문서들을 처리하여 텍스트 청크로 나누는 함수입니다.
|