Upload streamlit_app.py

src/streamlit_app.py

@@ -39,11 +39,23 @@ 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:  
+        f.write(docs.getvalue()) 
+    docs_loader = TextLoader(temp_filepath) 
+    text_doc = docs_loader.load() 
     return text_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:  
+        f.write(docs.getvalue()) 
+    csv_loader = CSVLoader(temp_filepath) 
+    csv_doc = csv_loader.load() 
     return csv_doc
 
 # def get_json_file(docs):
@@ -188,7 +200,40 @@ def main():
 
         ################## TXT, CSV 버튼 구현
         # TXT 버튼 구현 참고 : if file.type == 'text/plain':
+        if st.button("Process[TXT]"):
+            with st.spinner("Processing"):
+                doc_list = []
+                for file in docs:
+                    print('file - type : ', file.type)
+                    if file.type == 'text/plain':
+                        doc_list.extend(get_text_file(file))
+                    else:
+                        st.error("TXT 파일이 아닙니다.")
+                if not doc_list:
+                    st.error("처리 가능한 문서를 찾지 못했습니다.")
+                    st.stop()
+
+                text_chunks = get_text_chunks(doc_list)
+                vectorstore = get_vectorstore(text_chunks)
+                st.session_state.conversation = get_conversation_chain(vectorstore)
+                
         # CSV 버튼 구현 참고 : if file.type == 'text/csv':
+        if st.button("Process[CSV]"):
+            with st.spinner("Processing"):
+                doc_list = []
+                for file in docs:
+                    print('file - type : ', file.type)
+                    if file.type == 'text/csv':
+                        doc_list.extend(get_csv_file(file))
+                    else:
+                        st.error("CSV 파일이 아닙니다.")
+                if not doc_list:
+                    st.error("처리 가능한 문서를 찾지 못했습니다.")
+                    st.stop()
+
+                text_chunks = get_text_chunks(doc_list)
+                vectorstore = get_vectorstore(text_chunks)
+                st.session_state.conversation = get_conversation_chain(vectorstore)
 
         if st.button("Process[JSON]"):
             with st.spinner("Processing"):