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() # 임시 디렉토리를 생성합니다.
|