DDingcheol commited on
Commit
7aa2764
Β·
1 Parent(s): c2a1485

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -3
app.py CHANGED
@@ -46,8 +46,14 @@ def get_csv_file(csv_docs):
46
  csv_doc = csv_loader.load()
47
  return csv_doc
48
 
49
- def get_json_file(docs):
50
- pass
 
 
 
 
 
 
51
 
52
 
53
  # λ¬Έμ„œλ“€μ„ μ²˜λ¦¬ν•˜μ—¬ ν…μŠ€νŠΈ 청크둜 λ‚˜λˆ„λŠ” ν•¨μˆ˜μž…λ‹ˆλ‹€.
@@ -159,4 +165,4 @@ def main():
159
 
160
 
161
  if __name__ == '__main__':
162
- main()
 
46
  csv_doc = csv_loader.load()
47
  return csv_doc
48
 
49
+ def get_json_file(json_docs):
50
+ temp_dir = tempfile.TemporaryDirectory()
51
+ tem_filepath=os.path.join(temp_dir.name, json_docs.name)
52
+ with open(temp_filepath, "wb") as f:
53
+ f.write(json_docs.getvalue())
54
+ json_loader = JSONLoader(file_path=temp_filepath, jq_schema='.messages[].content', text_content=False)
55
+ json_doc = json_loader.load()
56
+ return json_doc
57
 
58
 
59
  # λ¬Έμ„œλ“€μ„ μ²˜λ¦¬ν•˜μ—¬ ν…μŠ€νŠΈ 청크둜 λ‚˜λˆ„λŠ” ν•¨μˆ˜μž…λ‹ˆλ‹€.
 
165
 
166
 
167
  if __name__ == '__main__':
168
+ main()