Update src/streamlit_app.py

src/streamlit_app.py

@@ -213,6 +213,13 @@ def main():
                         doc_list.extend(get_text_file(file))
                     else:
                         st.error("TXT 파일이 아닙니다.")
+                if not doc_list:
+                    st.error("처리 가능한 TXT 문서를 찾지 못했습니다.")
+                    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]"):
@@ -223,6 +230,13 @@ def main():
                         doc_list.extend(get_csv_file(file))
                     else:
                         st.error("CSV 파일이 아닙니다.")
+                if not doc_list:
+                    st.error("처리 가능한 CSV 문서를 찾지 못했습니다.")
+                    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"):
                 # get txt text