Update app.py

app.py

@@ -29,34 +29,17 @@ def get_pdf_text(pdf_docs):
 # 과제
 # 아래 텍스트 추출 함수를 작성
 
-def get_text_file(text_docs):
-    temp_dir = tempfile.TemporaryDirectory()  # 임시 디렉토리를 생성합니다.
-    temp_filepath = os.path.join(temp_dir.name, text_docs.name)  # 임시 파일 경로를 생성합니다.
-    with open(temp_filepath, "wb") as f:  # 임시 파일을 바이너리 쓰기 모드로 엽니다.
-        f.write(text_docs.getvalue())  # PDF 문서의 내용을 임시 파일에 씁니다.
-    text_loader = PyPDFLoader(temp_filepath)  # PyPDFLoader를 사용해 PDF를 로드합니다.
-    text_doc = text_loader.load()  # 텍스트를 추출합니다.
-    return text_doc  # 추출한 텍스트를 반환합니다.
-
-def get_csv_file(csv_docs):
-    temp_dir = tempfile.TemporaryDirectory()  # 임시 디렉토리를 생성합니다.
-    temp_filepath = os.path.join(temp_dir.name, csv_docs.name)  # 임시 파일 경로를 생성합니다.
-    with open(temp_filepath, "wb") as f:  # 임시 파일을 바이너리 쓰기 모드로 엽니다.
-        f.write(csv_docs.getvalue())  # PDF 문서의 내용을 임시 파일에 씁니다.
-    csv_loader = PyPDFLoader(temp_filepath)  # PyPDFLoader를 사용해 PDF를 로드합니다.
-    csv_doc = csv_loader.load()  # 텍스트를 추출합니다.
-    return csv_doc  # 추출한 텍스트를 반환합니다.
+def get_text_file(docs):
+    text_loader = TextLoader(docs)  # 텍스트 파일을 처리할 TextLoader 사용
+    return text_loader.load()
 
+def get_csv_file(docs):
+    csv_loader = CSVLoader(docs)  # CSV 파일을 처리할 CSVLoader 사용
+    return csv_loader.load()
 
-def get_json_file(json_docs):
-    temp_dir = tempfile.TemporaryDirectory()  # 임시 디렉토리를 생성합니다.
-    temp_filepath = os.path.join(temp_dir.name, json_docs.name)  # 임시 파일 경로를 생성합니다.
-    with open(temp_filepath, "wb") as f:  # 임시 파일을 바이너리 쓰기 모드로 엽니다.
-        f.write(json_docs.getvalue())  # PDF 문서의 내용을 임시 파일에 씁니다.
-    json_loader = PyPDFLoader(temp_filepath)  # PyPDFLoader를 사용해 PDF를 로드합니다.
-    json_doc = json_loader.load()  # 텍스트를 추출합니다.
-    return json_doc  # 추출한 텍스트를 반환합니다.
-
+def get_json_file(docs):
+    json_loader = JSONLoader(docs)  # JSON 파일을 처리할 JSONLoader 사용
+    return json_loader.load()
 
 
 # 문서들을 처리하여 텍스트 청크로 나누는 함수입니다.
@@ -137,35 +120,24 @@ def main():
         st.subheader("Your documents")
         docs = st.file_uploader(
             "Upload your PDFs here and click on 'Process'", accept_multiple_files=True)
+        # 'Process' 버튼 블록 내에서
         if st.button("Process"):
             with st.spinner("Processing"):
-                # get pdf text
                 doc_list = []
 
                 for file in docs:
-                    print('file - type : ', file.type)
                     if file.type == 'text/plain':
-                        # file is .txt
                         doc_list.extend(get_text_file(file))
                     elif file.type in ['application/octet-stream', 'application/pdf']:
-                        # file is .pdf
                         doc_list.extend(get_pdf_text(file))
                     elif file.type == 'text/csv':
-                        # file is .csv
                         doc_list.extend(get_csv_file(file))
                     elif file.type == 'application/json':
-                        # file is .json
                         doc_list.extend(get_json_file(file))
 
-                # get the text chunks
                 text_chunks = get_text_chunks(doc_list)
-
-                # create vector store
                 vectorstore = get_vectorstore(text_chunks)
-
-                # create conversation chain
-                st.session_state.conversation = get_conversation_chain(
-                    vectorstore)
+                st.session_state.conversation = get_conversation_chain(vectorstore)
 
 
 if __name__ == '__main__':