yeonmorae commited on
Commit
03e4143
ยท
1 Parent(s): 2528860
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -45,7 +45,7 @@ def get_text_file(docs):
45
  def get_csv_file(docs):
46
  temp_dir = tempfile.TemporaryDirectory() # ์ž„์‹œ ๋””๋ ‰ํ† ๋ฆฌ๋ฅผ ์ƒ์„ฑํ•ฉ๋‹ˆ๋‹ค.
47
  temp_filepath = os.path.join(temp_dir.name, docs.name) # ์ž„์‹œ ํŒŒ์ผ ๊ฒฝ๋กœ๋ฅผ ์ƒ์„ฑํ•ฉ๋‹ˆ๋‹ค.
48
- with open(temp_filepath, "w") as f: # ์ž„์‹œ ํŒŒ์ผ์„ ํ…์ŠคํŠธ ์“ฐ๊ธฐ ๋ชจ๋“œ๋กœ ์—ฝ๋‹ˆ๋‹ค.
49
  f.write(docs.getvalue()) # csv ๋ฌธ์„œ์˜ ๋‚ด์šฉ์„ ์ž„์‹œ ํŒŒ์ผ์— ์”๋‹ˆ๋‹ค.
50
  csv_loader = CSVLoader(temp_filepath) # CSVLoader๋ฅผ ์‚ฌ์šฉํ•ด csv์„ ๋กœ๋“œํ•ฉ๋‹ˆ๋‹ค.
51
  csv_doc = csv_loader.load() # ํ…์ŠคํŠธ๋ฅผ ์ถ”์ถœํ•ฉ๋‹ˆ๋‹ค.
@@ -55,9 +55,8 @@ def get_csv_file(docs):
55
  def get_json_file(docs):
56
  temp_dir = tempfile.TemporaryDirectory()
57
  temp_filepath = os.path.join(temp_dir.name, docs.name)
58
- with open(temp_filepath, "w") as f:
59
  f.write(docs.getvalue())
60
-
61
  json_loader = JSONLoader(temp_filepath)
62
  json_data = json_loader.load()
63
 
 
45
  def get_csv_file(docs):
46
  temp_dir = tempfile.TemporaryDirectory() # ์ž„์‹œ ๋””๋ ‰ํ† ๋ฆฌ๋ฅผ ์ƒ์„ฑํ•ฉ๋‹ˆ๋‹ค.
47
  temp_filepath = os.path.join(temp_dir.name, docs.name) # ์ž„์‹œ ํŒŒ์ผ ๊ฒฝ๋กœ๋ฅผ ์ƒ์„ฑํ•ฉ๋‹ˆ๋‹ค.
48
+ with open(temp_filepath, "wb") as f: # ์ž„์‹œ ํŒŒ์ผ์„ ํ…์ŠคํŠธ ์“ฐ๊ธฐ ๋ชจ๋“œ๋กœ ์—ฝ๋‹ˆ๋‹ค.
49
  f.write(docs.getvalue()) # csv ๋ฌธ์„œ์˜ ๋‚ด์šฉ์„ ์ž„์‹œ ํŒŒ์ผ์— ์”๋‹ˆ๋‹ค.
50
  csv_loader = CSVLoader(temp_filepath) # CSVLoader๋ฅผ ์‚ฌ์šฉํ•ด csv์„ ๋กœ๋“œํ•ฉ๋‹ˆ๋‹ค.
51
  csv_doc = csv_loader.load() # ํ…์ŠคํŠธ๋ฅผ ์ถ”์ถœํ•ฉ๋‹ˆ๋‹ค.
 
55
  def get_json_file(docs):
56
  temp_dir = tempfile.TemporaryDirectory()
57
  temp_filepath = os.path.join(temp_dir.name, docs.name)
58
+ with open(temp_filepath, "wb") as f:
59
  f.write(docs.getvalue())
 
60
  json_loader = JSONLoader(temp_filepath)
61
  json_data = json_loader.load()
62