Daoneeee commited on
Commit
660085e
·
1 Parent(s): da8893e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -6
app.py CHANGED
@@ -12,7 +12,6 @@ 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 json
16
  import os
17
 
18
 
@@ -57,11 +56,13 @@ def get_json_file(json_docs):
57
  with open(temp_filepath, "wb") as f:
58
  f.write(json_docs.getvalue())
59
 
60
- # JSONLoader 초기화 시 jq_schema 지정
61
- jq_schema = '.'
62
- json_loader = JSONLoader(temp_filepath, jq_schema=jq_schema)
63
- json_doc = json_loader.load()
64
- return json_doc
 
 
65
 
66
 
67
  # 문서들을 처리하여 텍스트 청크로 나누는 함수입니다.
 
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
 
 
56
  with open(temp_filepath, "wb") as f:
57
  f.write(json_docs.getvalue())
58
 
59
+ # JSONLoader 초기화
60
+ json_loader = JSONLoader(temp_filepath, text_content=True)
61
+ json_content = json_loader.load()
62
+ # JSON 데이터를 문자열로 변환
63
+ json_string = json.dumps(json_content)
64
+ # 변환된 문자열을 리스트에 담아 반환
65
+ return [json_string]
66
 
67
 
68
  # 문서들을 처리하여 텍스트 청크로 나누는 함수입니다.