Moonjunho commited on
Commit
a0e0f09
Β·
1 Parent(s): e5b45ad

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +26 -2
app.py CHANGED
@@ -33,10 +33,34 @@ def get_text_file(docs):
33
 
34
 
35
  def get_csv_file(docs):
36
- pass
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
37
 
38
  def get_json_file(docs):
39
- pass
 
 
 
 
 
 
40
 
41
 
42
  # λ¬Έμ„œλ“€μ„ μ²˜λ¦¬ν•˜μ—¬ ν…μŠ€νŠΈ 청크둜 λ‚˜λˆ„λŠ” ν•¨μˆ˜μž…λ‹ˆλ‹€.
 
33
 
34
 
35
  def get_csv_file(docs):
36
+ temp_dir = tempfile.TemporaryDirectory() # μž„μ‹œ 디렉토리λ₯Ό μƒμ„±ν•©λ‹ˆλ‹€.
37
+ temp_filepath = os.path.join(temp_dir.name, docs.name) # μž„μ‹œ 파일 경둜λ₯Ό μƒμ„±ν•©λ‹ˆλ‹€.
38
+ with open(temp_filepath, "wb") as f: # μž„μ‹œ νŒŒμΌμ„ λ°”μ΄λ„ˆλ¦¬ μ“°κΈ° λͺ¨λ“œλ‘œ μ—½λ‹ˆλ‹€.
39
+ f.write(docs.getvalue())
40
+ loader = CSVLoader(file_path= temp_filepath, csv_args={
41
+ 'delimiter': ',',
42
+ 'quotechar': '"',
43
+ 'fieldnames': ["gameId", "blueWins", "blueWardsPlaced", "blueWardsDestroyed", "blueFirstBlood", "blueKills", "blueDeaths",
44
+ "blueAssists", "blueEliteMonsters", "blueDragons", "blueHeralds", "blueTowersDestroyed", "blueTotalGold",
45
+ "blueAvgLevel", "blueTotalExperience", "blueTotalMinionsKilled", "blueTotalJungleMinionsKilled", "blueGoldDiff",
46
+ "blueExperienceDiff", "blueCSPerMin", "blueGoldPerMin", "redWardsPlaced", "redWardsDestroyed", "redFirstBlood",
47
+ "redKills", "redDeaths", "redAssists", "redEliteMonsters", "redDragons", "redHeralds", "redTowersDestroyed",
48
+ "redTotalGold", "redAvgLevel", "redTotalExperience", "redTotalMinionsKilled", "redTotalJungleMinionsKilled",
49
+ "redGoldDiff", "redExperienceDiff", "redCSPerMin", "redGoldPerMin"]
50
+ })
51
+ data = loader.load()
52
+ return data
53
+
54
+ data = loader.load()
55
 
56
  def get_json_file(docs):
57
+ temp_dir = tempfile.TemporaryDirectory()
58
+ temp_filepath = os.path.join(temp_dir.name, docs.name)
59
+ with open(temp_filepath, "wb") as f: # μž„μ‹œ νŒŒμΌμ„ λ°”μ΄λ„ˆλ¦¬ μ“°κΈ° λͺ¨λ“œλ‘œ μ—½λ‹ˆλ‹€.
60
+ f.write(docs.getvalue())
61
+ loader = JSONLoader(file_path = temp_filepath, jq_schema ='.messages[].content', text_content=False)
62
+ data = loader.load()
63
+ return data
64
 
65
 
66
  # λ¬Έμ„œλ“€μ„ μ²˜λ¦¬ν•˜μ—¬ ν…μŠ€νŠΈ 청크둜 λ‚˜λˆ„λŠ” ν•¨μˆ˜μž…λ‹ˆλ‹€.