Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -12,10 +12,12 @@ from htmlTemplates import css, bot_template, user_template
|
|
12 |
from langchain.llms import HuggingFaceHub, LlamaCpp, CTransformers # For loading transformer models.
|
13 |
from langchain.document_loaders import PyPDFLoader, TextLoader, JSONLoader, CSVLoader
|
14 |
import tempfile # 임시 파일을 생성하기 위한 라이브러리입니다.
|
|
|
15 |
import os
|
16 |
|
17 |
|
18 |
|
|
|
19 |
# PDF 문서로부터 텍스트를 추출하는 함수입니다.
|
20 |
def get_pdf_text(pdf_docs):
|
21 |
temp_dir = tempfile.TemporaryDirectory() # 임시 디렉토리를 생성합니다.
|
@@ -57,7 +59,9 @@ def get_json_file(json_docs):
|
|
57 |
|
58 |
# JSONLoader 초기화 시 jq_schema 지정
|
59 |
jq_schema = {} # JSONQuery 스키마, 필요에 따라 적절한 스키마를 지정해야 합니다.
|
60 |
-
|
|
|
|
|
61 |
json_doc = json_loader.load()
|
62 |
return json_doc
|
63 |
|
|
|
12 |
from langchain.llms import HuggingFaceHub, LlamaCpp, CTransformers # For loading transformer models.
|
13 |
from langchain.document_loaders import PyPDFLoader, TextLoader, JSONLoader, CSVLoader
|
14 |
import tempfile # 임시 파일을 생성하기 위한 라이브러리입니다.
|
15 |
+
import json
|
16 |
import os
|
17 |
|
18 |
|
19 |
|
20 |
+
|
21 |
# PDF 문서로부터 텍스트를 추출하는 함수입니다.
|
22 |
def get_pdf_text(pdf_docs):
|
23 |
temp_dir = tempfile.TemporaryDirectory() # 임시 디렉토리를 생성합니다.
|
|
|
59 |
|
60 |
# JSONLoader 초기화 시 jq_schema 지정
|
61 |
jq_schema = {} # JSONQuery 스키마, 필요에 따라 적절한 스키마를 지정해야 합니다.
|
62 |
+
# JQ 스키마를 JSON 문자열로 변환
|
63 |
+
jq_schema_str = json.dumps(jq_schema)
|
64 |
+
json_loader = JSONLoader(temp_filepath, jq_schema=jq_schema_str)
|
65 |
json_doc = json_loader.load()
|
66 |
return json_doc
|
67 |
|