philona commited on
Commit
0c5a65d
ยท
1 Parent(s): 7cd2cf8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +16 -4
app.py CHANGED
@@ -33,11 +33,23 @@ def get_text_file(txt_docs):
33
  txt_doc = txt_loader.load() # ํ…์ŠคํŠธ๋ฅผ ์ถ”์ถœํ•ฉ๋‹ˆ๋‹ค.
34
  return txt_doc
35
 
36
- def get_csv_file(docs):
37
- pass
 
 
 
 
 
 
38
 
39
- def get_json_file(docs):
40
- pass
 
 
 
 
 
 
41
 
42
 
43
  # ๋ฌธ์„œ๋“ค์„ ์ฒ˜๋ฆฌํ•˜์—ฌ ํ…์ŠคํŠธ ์ฒญํฌ๋กœ ๋‚˜๋ˆ„๋Š” ํ•จ์ˆ˜์ž…๋‹ˆ๋‹ค.
 
33
  txt_doc = txt_loader.load() # ํ…์ŠคํŠธ๋ฅผ ์ถ”์ถœํ•ฉ๋‹ˆ๋‹ค.
34
  return txt_doc
35
 
36
+ def get_csv_file(csv_docs):
37
+ temp_dir = tempfile.TemporaryDirectory()
38
+ temp_filepath = os.path.join(temp_dir.name, csv_docs.name)
39
+ with open(temp_filepath, "wb") as f:
40
+ f.write(csv_docs.getvalue())
41
+ csv_loader = CSVLoader(temp_filepath) # CSVLoader๋ฅผ ์‚ฌ์šฉํ•ด PDF๋ฅผ ๋กœ๋“œํ•ฉ๋‹ˆ๋‹ค.
42
+ csv_doc = csv_loader.load() # ํ…์ŠคํŠธ๋ฅผ ์ถ”์ถœํ•ฉ๋‹ˆ๋‹ค.
43
+ return csv_doc
44
 
45
+ def get_json_file(json_docs):
46
+ temp_dir = tempfile.TemporaryDirectory()
47
+ temp_filepath = os.path.join(temp_dir.name, json_docs.name)
48
+ with open(temp_filepath, "wb") as f:
49
+ f.write(json_docs.getvalue())
50
+ json_loader = JSONLLoader(temp_filepath) # JSONLoader๋ฅผ ์‚ฌ์šฉํ•ด PDF๋ฅผ ๋กœ๋“œํ•ฉ๋‹ˆ๋‹ค.
51
+ json_doc = json_loader.load() # ํ…์ŠคํŠธ๋ฅผ ์ถ”์ถœํ•ฉ๋‹ˆ๋‹ค.
52
+ return json_doc
53
 
54
 
55
  # ๋ฌธ์„œ๋“ค์„ ์ฒ˜๋ฆฌํ•˜์—ฌ ํ…์ŠคํŠธ ์ฒญํฌ๋กœ ๋‚˜๋ˆ„๋Š” ํ•จ์ˆ˜์ž…๋‹ˆ๋‹ค.