first commit
Browse files
app.py
CHANGED
|
@@ -13,8 +13,6 @@ from langchain.llms import HuggingFaceHub, LlamaCpp, CTransformers # For loadin
|
|
| 13 |
from langchain.document_loaders import PyPDFLoader, TextLoader, JSONLoader, CSVLoader
|
| 14 |
import tempfile # 임시 파일을 생성하기 위한 라이브러리입니다.
|
| 15 |
import os
|
| 16 |
-
import pandas as pd
|
| 17 |
-
import json
|
| 18 |
|
| 19 |
|
| 20 |
# PDF 문서로부터 텍스트를 추출하는 함수입니다.
|
|
@@ -49,9 +47,9 @@ def get_csv_file(csv_docs):
|
|
| 49 |
with open(temp_filepath, "wb") as f: # 임시 파일을 바이너리 쓰기 모드로 엽니다.
|
| 50 |
f.write(csv_docs.getvalue())
|
| 51 |
|
| 52 |
-
|
| 53 |
-
|
| 54 |
-
return
|
| 55 |
|
| 56 |
def get_json_file(json_docs):
|
| 57 |
temp_dir = tempfile.TemporaryDirectory() # 임시 디렉토리를 생성합니다.
|
|
|
|
| 13 |
from langchain.document_loaders import PyPDFLoader, TextLoader, JSONLoader, CSVLoader
|
| 14 |
import tempfile # 임시 파일을 생성하기 위한 라이브러리입니다.
|
| 15 |
import os
|
|
|
|
|
|
|
| 16 |
|
| 17 |
|
| 18 |
# PDF 문서로부터 텍스트를 추출하는 함수입니다.
|
|
|
|
| 47 |
with open(temp_filepath, "wb") as f: # 임시 파일을 바이너리 쓰기 모드로 엽니다.
|
| 48 |
f.write(csv_docs.getvalue())
|
| 49 |
|
| 50 |
+
csv_loader = CSVLoader(temp_filepath)
|
| 51 |
+
csv_doc = csv_loader.load()
|
| 52 |
+
return csv_doc
|
| 53 |
|
| 54 |
def get_json_file(json_docs):
|
| 55 |
temp_dir = tempfile.TemporaryDirectory() # 임시 디렉토리를 생성합니다.
|