Update src/streamlit_app.py

src/streamlit_app.py

@@ -41,16 +41,11 @@ def get_text_file(docs):
     #################### 내용을 추가할 부분
 
     temp_dir = tempfile.TemporaryDirectory()
-    temp_filepath = os.path.join(temp_dir.name, text_docs.name)
-
-    
+    temp_filepath = os.path.join(temp_dir.name, docs.name)
     with open(temp_filepath, "wb") as f:
-        f.write(text_docs.getvalue())
-
-    
+        f.write(docs.getvalue())
     text_loader = TextLoader(temp_filepath, encoding="utf-8")
-    text_doc = text_loader.load()  
-    
+    text_doc = text_loader.load()   
     return text_doc
 
 
@@ -58,16 +53,12 @@ def get_csv_file(docs):
     #################### 내용을 추가할 부분
 
     temp_dir = tempfile.TemporaryDirectory()
-    temp_filepath = os.path.join(temp_dir.name, csv_docs.name)
-
- 
+    temp_filepath = os.path.join(temp_dir.name, docs.name)
     with open(temp_filepath, "wb") as f:
-        f.write(csv_docs.getvalue())
-
-
+        f.write(docs.getvalue())
     csv_loader = CSVLoader(
         file_path=temp_filepath,
-        encoding="utf-8",   
+        encoding="utf-8", 
     )
     csv_doc = csv_loader.load()     
     return csv_doc
@@ -214,8 +205,24 @@ 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:
+                    if file.type == "text/plain":
+                        doc_list.extend(get_text_file(file))
+                    else:
+                        st.error("TXT 파일이 아닙니다.")
         # CSV 버튼 구현 참고 : if file.type == 'text/csv':
 
+        if st.button("Process[CSV]"):
+            with st.spinner("Processing"):
+                doc_list = []
+                for file in docs:
+                    if file.type == "text/csv":
+                        doc_list.extend(get_csv_file(file))
+                    else:
+                        st.error("CSV 파일이 아닙니다.")
         if st.button("Process[JSON]"):
             with st.spinner("Processing"):
                 # get txt text