bin20 commited on
Commit
b5a3f8b
·
1 Parent(s): 09610d9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -12
app.py CHANGED
@@ -25,12 +25,12 @@ def get_pdf_text(pdf_docs):
25
  # 과제
26
  # 아래 텍스트 추출 함수를 작성
27
  def get_text_file(text_data):
28
- temp_dir = tempfile.TemporaryDirectory() # 임시 디렉토리 생성
29
- temp_filepath = os.path.join(temp_dir.name, text_data.name) # 임시 파일 경로 생성
30
- with open(temp_filepath, "wb") as f: # 임시 파일을 바이너리 쓰기 모드로 열기
31
- f.write(text_data.getvalue()) # 텍스트 데이터를 임시 파일에 쓰기
32
 
33
- # 임시 파일을 TextLoader로 로드하여 텍스트 추출
34
  text_loader = TextLoader(temp_filepath)
35
  text = text_loader.load()
36
 
@@ -43,19 +43,18 @@ def get_csv_file(csv_file):
43
  with open(temp_filepath, "wb") as f:
44
  f.write(csv_file.getvalue())
45
 
46
- csv_loader = CSVLoader(temp_filepath) # CSV 파일을 로드합니다.
47
  csv_data = csv_loader.load()
48
 
49
- return csv_data # CSV 데이터를 그대로 반환합니다.
50
-
51
 
52
 
53
 
54
  def get_json_file(json_datas):
55
- temp_dir = tempfile.TemporaryDirectory() # 임시 디렉토리를 생성합니다.
56
- temp_filepath = os.path.join(temp_dir.name, json_datas.name) # 임시 JSON 파일 경로를 생성합니다.
57
- with open(temp_filepath, "w", encoding="utf-8") as jsonfile: # JSON 파일을 쓰기 모드로 엽니다.
58
- jsonfile.write(json_datas.getvalue().decode('utf-8')) # JSON 데이터를 파일에 씁니다.
59
  json_loader = JSONLoader(
60
  temp_filepath,
61
  jq_schema='.',
 
25
  # 과제
26
  # 아래 텍스트 추출 함수를 작성
27
  def get_text_file(text_data):
28
+ temp_dir = tempfile.TemporaryDirectory()
29
+ temp_filepath = os.path.join(temp_dir.name, text_data.name)
30
+ with open(temp_filepath, "wb") as f:
31
+ f.write(text_data.getvalue())
32
 
33
+
34
  text_loader = TextLoader(temp_filepath)
35
  text = text_loader.load()
36
 
 
43
  with open(temp_filepath, "wb") as f:
44
  f.write(csv_file.getvalue())
45
 
46
+ csv_loader = CSVLoader(temp_filepath)
47
  csv_data = csv_loader.load()
48
 
49
+ return csv_data
 
50
 
51
 
52
 
53
  def get_json_file(json_datas):
54
+ temp_dir = tempfile.TemporaryDirectory()
55
+ temp_filepath = os.path.join(temp_dir.name, json_datas.name)
56
+ with open(temp_filepath, "w", encoding="utf-8") as jsonfile:
57
+ jsonfile.write(json_datas.getvalue().decode('utf-8'))
58
  json_loader = JSONLoader(
59
  temp_filepath,
60
  jq_schema='.',