MyeongGyun commited on
Commit
556c289
β€’
1 Parent(s): 897ec15

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +17 -4
app.py CHANGED
@@ -29,14 +29,27 @@ def get_pdf_text(pdf_docs):
29
  # μ•„λž˜ ν…μŠ€νŠΈ μΆ”μΆœ ν•¨μˆ˜λ₯Ό μž‘μ„±
30
 
31
  def get_text_file(docs):
32
- pass
33
-
 
 
 
34
 
35
  def get_csv_file(docs):
36
- pass
 
 
 
 
 
37
 
38
  def get_json_file(docs):
39
- pass
 
 
 
 
 
40
 
41
 
42
  # λ¬Έμ„œλ“€μ„ μ²˜λ¦¬ν•˜μ—¬ ν…μŠ€νŠΈ 청크둜 λ‚˜λˆ„λŠ” ν•¨μˆ˜μž…λ‹ˆλ‹€.
 
29
  # μ•„λž˜ ν…μŠ€νŠΈ μΆ”μΆœ ν•¨μˆ˜λ₯Ό μž‘μ„±
30
 
31
  def get_text_file(docs):
32
+ if docs.type == 'text/plain':
33
+ text = docs.getvalue().decode("utf-8")
34
+ return [text]
35
+ else:
36
+ return []
37
 
38
  def get_csv_file(docs):
39
+ if docs.type == 'text/csv':
40
+ csv_text = docs.getvalue().decode("utf-8")
41
+
42
+ return csv_text.splitlines()
43
+ else:
44
+ return []
45
 
46
  def get_json_file(docs):
47
+ if docs.type == 'application/json':
48
+ json_text = docs.getvalue().decode("utf-8")
49
+
50
+ return [json_text]
51
+ else:
52
+ return []
53
 
54
 
55
  # λ¬Έμ„œλ“€μ„ μ²˜λ¦¬ν•˜μ—¬ ν…μŠ€νŠΈ 청크둜 λ‚˜λˆ„λŠ” ν•¨μˆ˜μž…λ‹ˆλ‹€.