Update app.py

app.py

@@ -29,14 +29,22 @@ def get_pdf_text(pdf_docs):
 # 아래 텍스트 추출 함수를 작성
 
 def get_text_file(docs):
-    pass
-
+    # 텍스트 파일 (.txt)에서 텍스트를 추출하는 함수
+    return [docs.getvalue().decode('utf-8')]
 
 def get_csv_file(docs):
-    pass
+    # CSV 파일 (.csv)에서 텍스트를 추출하는 함수
+    csv_loader = CSVLoader(docs)
+    csv_data = csv_loader.load()
+    # CSV 파일의 각 행을 문자열로 변환하여 반환
+    return [' '.join(map(str, row)) for row in csv_data]
 
 def get_json_file(docs):
-    pass
+    # JSON 파일 (.json)에서 텍스트를 추출하는 함수
+    json_loader = JSONLoader(docs)
+    json_data = json_loader.load()
+    # JSON 파일의 각 항목을 문자열로 변환하여 반환
+    return [json.dumps(item) for item in json_data]
 
     
 # 문서들을 처리하여 텍스트 청크로 나누는 함수입니다.
@@ -117,35 +125,34 @@ def main():
 
         st.subheader("Your documents")
         docs = st.file_uploader(
-            "Upload your PDFs here and click on 'Process'", accept_multiple_files=True)
+            "Upload your documents here and click on 'Process'", accept_multiple_files=True)
         if st.button("Process"):
             with st.spinner("Processing"):
-                # get pdf text
+                # 문서에서 추출한 텍스트를 담을 리스트
                 doc_list = []
 
                 for file in docs:
                     if file.type == 'text/plain':
-                        # file is .txt
+                        # .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
+                        # .csv 파일의 경우
                         doc_list.extend(get_csv_file(file))
                     elif file.type == 'application/json':
-                        # file is .json
+                        # .json 파일의 경우
                         doc_list.extend(get_json_file(file))
+                    elif file.type in ['application/octet-stream', 'application/pdf']:
+                        # .pdf 파일의 경우
+                        doc_list.extend(get_pdf_text(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__':