lds204 commited on
Commit
f78d864
β€’
1 Parent(s): fcb2ac7
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -40,8 +40,8 @@ def get_csv_file(csv_docs):
40
  temp_dir = tempfile.TemporaryDirectory()
41
  temp_filepath = os.path.join(temp_dir.name, csv_docs.name)
42
  with open(temp_filepath, "wb") as f: # μž„μ‹œ νŒŒμΌμ„ λ°”μ΄λ„ˆλ¦¬ μ“°κΈ° λͺ¨λ“œλ‘œ μ—½λ‹ˆλ‹€.
43
- f.write(csv_docs.getvalue().decode("utf-8"))
44
- csv_loader = CSVLoader(temp_filepath)
45
  csv_doc = csv_loader.load()
46
  return csv_doc
47
 
@@ -50,8 +50,8 @@ def get_json_file(json_docs):
50
  temp_filepath = os.path.join(temp_dir.name, json_docs.name)
51
 
52
  with open(temp_filepath, "wb") as f: # μž„μ‹œ νŒŒμΌμ„ λ°”μ΄λ„ˆλ¦¬ μ“°κΈ° λͺ¨λ“œλ‘œ μ—½λ‹ˆλ‹€.
53
- f.write(json_docs.getvalue().decode("utf-8"))
54
- json_loader = JSONLoader(file_path=temp_filepath, jq_schema='.messages[].content')
55
  json_doc = json_loader.load()
56
  return json_doc
57
 
 
40
  temp_dir = tempfile.TemporaryDirectory()
41
  temp_filepath = os.path.join(temp_dir.name, csv_docs.name)
42
  with open(temp_filepath, "wb") as f: # μž„μ‹œ νŒŒμΌμ„ λ°”μ΄λ„ˆλ¦¬ μ“°κΈ° λͺ¨λ“œλ‘œ μ—½λ‹ˆλ‹€.
43
+ f.write(csv_docs.getvalue())
44
+ csv_loader = TextLoader(temp_filepath)
45
  csv_doc = csv_loader.load()
46
  return csv_doc
47
 
 
50
  temp_filepath = os.path.join(temp_dir.name, json_docs.name)
51
 
52
  with open(temp_filepath, "wb") as f: # μž„μ‹œ νŒŒμΌμ„ λ°”μ΄λ„ˆλ¦¬ μ“°κΈ° λͺ¨λ“œλ‘œ μ—½λ‹ˆλ‹€.
53
+ f.write(json_docs.getvalue())
54
+ json_loader = TextLoader(temp_filepath)
55
  json_doc = json_loader.load()
56
  return json_doc
57