edit codes

app.py

@@ -31,22 +31,22 @@ def get_pdf_text(pdf_docs):
 # 아래 텍스트 추출 함수를 작성
 
 def get_text_file(docs):
-    temp_dir = tempfile.TemporaryDirectory()  # 임시 디렉토리를 생성합니다.
-    temp_filepath = os.path.join(temp_dir.name, docs.name)  # 임시 파일 경로를 생성합니다.
-    with open(temp_filepath, "wb") as f: # 임시 파일을 바이너리 쓰기 모드로 엽니다.
+    temp_dir2 = tempfile.TemporaryDirectory()  # 임시 디렉토리를 생성합니다.
+    temp_filepath2 = os.path.join(temp_dir2.name, docs.name)  # 임시 파일 경로를 생성합니다.
+    with open(temp_filepath2, "wb") as f: # 임시 파일을 바이너리 쓰기 모드로 엽니다.
         f.write(docs.getvalue())  # TXT 문서의 내용을 임시 파일에 씁니다.
-    with open(temp_filepath, "r") as txt_file: # TXT 파일을 바이너리 쓰기 모드로 엽니다.
+    with open(temp_filepath2, "r") as txt_file: # TXT 파일을 바이너리 쓰기 모드로 엽니다.
         txt_doc = txt_file.read() # 텍스트를 추출합니다.
     return txt_doc  # 추출한 텍스트를 반환합니다.
 
 def get_csv_file(docs):
-    temp_dir = tempfile.TemporaryDirectory()
-    temp_filepath = os.path.join(temp_dir.name, docs.name)
-    with open(temp_filepath, "wb") as f:
+    temp_dir3 = tempfile.TemporaryDirectory()
+    temp_filepath3 = os.path.join(temp_dir3.name, docs.name)
+    with open(temp_filepath3, "wb") as f:
         f.write(docs.getvalue())
 
     csv_data = []
-    with open(temp_filepath, "r") as csv_file:
+    with open(temp_filepath3, "r") as csv_file:
         csv_reader = csv.reader(csv_file)
         for row in csv_reader:
             csv_data.append(row)
@@ -54,14 +54,16 @@ def get_csv_file(docs):
     return csv_data
 
 def get_json_file(docs):
-    temp_dir = tempfile.TemporaryDirectory()
-    temp_filepath = os.path.join(temp_dir.name, docs.name)
-    with open(temp_filepath, "wb") as f:
+    temp_dir4 = tempfile.TemporaryDirectory()
+    temp_filepath4 = os.path.join(temp_dir4.name, docs.name)
+    with open(temp_filepath4, "wb") as f:
         f.write(docs.getvalue())
-
-    with open(temp_filepath, "r") as json_file:
-        json_data = json.load(json_file)
-
+    json_loader = JSONLoader(
+        file_path=temp_filepath4,
+        jq_schema='.messages[].content',
+        text_content=False
+    )
+    json_data = json_loader.load()
     return json_data
 
     
@@ -72,17 +74,9 @@ def get_text_chunks(documents):
         chunk_overlap=200, # 청크 사이의 중복을 지정합니다.
         length_function=len # 텍스트의 길이를 측정하는 함수를 지정합니다.
     )
-    texts = []
-    for doc in documents:
-        if isinstance(doc, str):
-            # doc이 문자열인 경우 직접 texts에 추가
-            texts.append(doc)
-        else:
-            # doc이 'page_content' 속성을 갖춘 객체인 경우
-            texts.append(doc.page_content)
 
-    chunks = text_splitter.split_documents(texts)
-    return chunks
+    documents = text_splitter.split_documents(documents) # 문서들을 청크로 나눕니다
+    return documents # 나눈 청크를 반환합니다.
 
 
 # 텍스트 청크들로부터 벡터 스토어를 생성하는 함수입니다.