lds204 commited on
Commit
fcb2ac7
β€’
1 Parent(s): 35017a4
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -40,7 +40,7 @@ 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.read())
44
  csv_loader = CSVLoader(temp_filepath)
45
  csv_doc = csv_loader.load()
46
  return csv_doc
@@ -50,7 +50,7 @@ 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.read())
54
  json_loader = JSONLoader(file_path=temp_filepath, jq_schema='.messages[].content')
55
  json_doc = json_loader.load()
56
  return json_doc
 
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
 
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