Commit Β·
4647bc2
1
Parent(s): 14e7cb3
Update app.py
Browse files
app.py
CHANGED
|
@@ -38,11 +38,27 @@ def get_text_file(text_docs):
|
|
| 38 |
return text_doc
|
| 39 |
|
| 40 |
|
| 41 |
-
def get_csv_file(
|
| 42 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 43 |
|
| 44 |
def get_json_file(docs):
|
| 45 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 46 |
|
| 47 |
|
| 48 |
# λ¬Έμλ€μ μ²λ¦¬νμ¬ ν
μ€νΈ μ²ν¬λ‘ λλλ ν¨μμ
λλ€.
|
|
|
|
| 38 |
return text_doc
|
| 39 |
|
| 40 |
|
| 41 |
+
def get_csv_file(csv_docs):
|
| 42 |
+
temp_dir = tempfile.TemporaryDirectory() # μμ λλ ν 리λ₯Ό μμ±ν©λλ€.
|
| 43 |
+
temp_filepath = os.path.join(temp_dir.name, csv_docs.name) # μμ νμΌ κ²½λ‘λ₯Ό μμ±ν©λλ€.
|
| 44 |
+
with open(temp_filepath, "wb") as f: # μμ νμΌμ λ°μ΄λ리 μ°κΈ° λͺ¨λλ‘ μ½λλ€.
|
| 45 |
+
f.write(csv_docs.getvalue()) # csv λ¬Έμμ λ΄μ©μ μμ νμΌμ μλλ€.
|
| 46 |
+
csv_loader = CSVLoader(temp_filepath)
|
| 47 |
+
csv_doc = csv_loader.load()
|
| 48 |
+
return csv_doc
|
| 49 |
+
|
| 50 |
|
| 51 |
def get_json_file(docs):
|
| 52 |
+
temp_dir = tempfile.TemporaryDirectory() # μμ λλ ν 리λ₯Ό μμ±ν©λλ€.
|
| 53 |
+
temp_filepath = os.path.join(temp_dir.name, json_docs.name) # μμ νμΌ κ²½λ‘λ₯Ό μμ±ν©λλ€.
|
| 54 |
+
with open(temp_filepath, "wb") as f: # μμ νμΌμ λ°μ΄λ리 μ°κΈ° λͺ¨λλ‘ μ½λλ€.
|
| 55 |
+
f.write(json_docs.getvalue()) # json λ¬Έμμ λ΄μ©μ μμ νμΌμ μλλ€.
|
| 56 |
+
json_loader = JSONLoader(temp_filepath,
|
| 57 |
+
jq_schema='.messages[].content',
|
| 58 |
+
text_content=False)
|
| 59 |
+
json_doc = json_loader.load()
|
| 60 |
+
return json_doc
|
| 61 |
+
|
| 62 |
|
| 63 |
|
| 64 |
# λ¬Έμλ€μ μ²λ¦¬νμ¬ ν
μ€νΈ μ²ν¬λ‘ λλλ ν¨μμ
λλ€.
|