ai

app.py

@@ -11,7 +11,6 @@ from langchain.document_loaders import PyPDFLoader, TextLoader, JSONLoader
 from langchain.document_loaders.csv_loader import CSVLoader
 import json
 from pathlib import Path
-from pprint import pprint
 import tempfile # 임시 파일을 생성하기 위한 라이브러리입니다.
 import os
 from huggingface_hub import hf_hub_download # Hugging Face Hub에서 모델을 다운로드하기 위한 함수입니다.
@@ -51,11 +50,13 @@ def get_json_file(json_docs):
     temp_filepath = os.path.join(temp_dir.name, json_docs.name)
     with open(temp_filepath, "wb") as f:  # 임시 파일을 바이너리 쓰기 모드로 엽니다.
         f.write(json_docs.getvalue()) 
-    json_loader = JSONLoader(
-                file_path = temp_filepath,
-                jq_schema='.messages[].content',
-                text_content=False)
-    json_doc = json_loader.load()
+    # json_loader = JSONLoader(
+    #             file_path = temp_filepath,
+    #             jq_schema='.messages[].content',
+    #             text_content=False)
+    
+    # json_doc = json_loader.load()
+    json_doc = json.loads(Path(temp_filepath).read_text())
     return json_doc
     
 # 문서들을 처리하여 텍스트 청크로 나누는 함수입니다.