OsanAB commited on
Commit
55ad768
Β·
1 Parent(s): da90579
Files changed (1) hide show
  1. app.py +21 -3
app.py CHANGED
@@ -29,14 +29,32 @@ def get_pdf_text(pdf_docs):
29
  # μ•„λž˜ ν…μŠ€νŠΈ μΆ”μΆœ ν•¨μˆ˜λ₯Ό μž‘μ„±
30
 
31
  def get_text_file(docs):
32
- return get_file(TextLoader, text_docs)
 
 
 
 
 
 
33
 
34
 
35
  def get_csv_file(docs):
36
- return get_file(CSVLoader, csv_docs)
 
 
 
 
 
 
37
 
38
  def get_json_file(docs):
39
- return get_file(JSONLoader, json_docs)
 
 
 
 
 
 
40
 
41
 
42
  # λ¬Έμ„œλ“€μ„ μ²˜λ¦¬ν•˜μ—¬ ν…μŠ€νŠΈ 청크둜 λ‚˜λˆ„λŠ” ν•¨μˆ˜μž…λ‹ˆλ‹€.
 
29
  # μ•„λž˜ ν…μŠ€νŠΈ μΆ”μΆœ ν•¨μˆ˜λ₯Ό μž‘μ„±
30
 
31
  def get_text_file(docs):
32
+ temp_dir = tempfile.TemporaryDirectory()
33
+ temp_path = os.path.join(temp_dir.name, docs.name)
34
+ with open(temp_path, "wb") as f:
35
+ f.write(docs.getvalue())
36
+ loader = TextLoader(temp_path)
37
+ doc = loader.load()
38
+ return doc
39
 
40
 
41
  def get_csv_file(docs):
42
+ temp_dir = tempfile.TemporaryDirectory()
43
+ temp_path = os.path.join(temp_dir.name, docs.name)
44
+ with open(temp_path, "wb") as f:
45
+ f.write(docs.getvalue())
46
+ loader = CSVLoader(temp_path)
47
+ doc = loader.load()
48
+ return doc
49
 
50
  def get_json_file(docs):
51
+ temp_dir = tempfile.TemporaryDirectory()
52
+ temp_path = os.path.join(temp_dir.name, docs.name)
53
+ with open(temp_path, "wb") as f:
54
+ f.write(docs.getvalue())
55
+ loader = JSONLoader(temp_path)
56
+ doc = loader.load()
57
+ return doc
58
 
59
 
60
  # λ¬Έμ„œλ“€μ„ μ²˜λ¦¬ν•˜μ—¬ ν…μŠ€νŠΈ 청크둜 λ‚˜λˆ„λŠ” ν•¨μˆ˜μž…λ‹ˆλ‹€.