Daoneeee commited on
Commit
fb01e89
·
1 Parent(s): 56f757c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
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
- json_loader = JSONLoader(temp_filepath, jq_schema=jq_schema)
 
 
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