[email protected] commited on
Commit
5f4b8cc
Β·
1 Parent(s): 333bd63

edit codes

Browse files
Files changed (1) hide show
  1. app.py +7 -3
app.py CHANGED
@@ -35,8 +35,7 @@ def get_text_file(docs):
35
  temp_filepath2 = os.path.join(temp_dir2.name, docs.name) # μž„μ‹œ 파일 경둜λ₯Ό μƒμ„±ν•©λ‹ˆλ‹€.
36
  with open(temp_filepath2, "wb") as f: # μž„μ‹œ νŒŒμΌμ„ λ°”μ΄λ„ˆλ¦¬ μ“°κΈ° λͺ¨λ“œλ‘œ μ—½λ‹ˆλ‹€.
37
  f.write(docs.getvalue()) # TXT λ¬Έμ„œμ˜ λ‚΄μš©μ„ μž„μ‹œ νŒŒμΌμ— μ”λ‹ˆλ‹€.
38
- with open(temp_filepath2, "r") as txt_file: # TXT νŒŒμΌμ„ λ°”μ΄λ„ˆλ¦¬ μ“°κΈ° λͺ¨λ“œλ‘œ μ—½λ‹ˆλ‹€.
39
- txt_doc = txt_file.read() # ν…μŠ€νŠΈλ₯Ό μΆ”μΆœν•©λ‹ˆλ‹€.
40
  return txt_doc # μΆ”μΆœν•œ ν…μŠ€νŠΈλ₯Ό λ°˜ν™˜ν•©λ‹ˆλ‹€.
41
 
42
  def get_csv_file(docs):
@@ -50,7 +49,12 @@ def get_csv_file(docs):
50
  csv_reader = csv.reader(csv_file)
51
  for row in csv_reader:
52
  csv_data.append(row)
53
-
 
 
 
 
 
54
  return csv_data
55
 
56
  def get_json_file(docs):
 
35
  temp_filepath2 = os.path.join(temp_dir2.name, docs.name) # μž„μ‹œ 파일 경둜λ₯Ό μƒμ„±ν•©λ‹ˆλ‹€.
36
  with open(temp_filepath2, "wb") as f: # μž„μ‹œ νŒŒμΌμ„ λ°”μ΄λ„ˆλ¦¬ μ“°κΈ° λͺ¨λ“œλ‘œ μ—½λ‹ˆλ‹€.
37
  f.write(docs.getvalue()) # TXT λ¬Έμ„œμ˜ λ‚΄μš©μ„ μž„μ‹œ νŒŒμΌμ— μ”λ‹ˆλ‹€.
38
+
 
39
  return txt_doc # μΆ”μΆœν•œ ν…μŠ€νŠΈλ₯Ό λ°˜ν™˜ν•©λ‹ˆλ‹€.
40
 
41
  def get_csv_file(docs):
 
49
  csv_reader = csv.reader(csv_file)
50
  for row in csv_reader:
51
  csv_data.append(row)
52
+ # csv_loader = CSVLoader(
53
+ # file_path=temp_filepath3,
54
+ # cs_schema='.messages[].content',
55
+ # text_content=False
56
+ # )
57
+ # csv_data = csv_loader.load()
58
  return csv_data
59
 
60
  def get_json_file(docs):