Update app.py

app.py

@@ -28,15 +28,33 @@ def get_pdf_text(pdf_docs):
 # 과제
 # 아래 텍스트 추출 함수를 작성
 
-def get_text_file(docs):
-    pass
-
-
-def get_csv_file(docs):
-    pass
-
-def get_json_file(docs):
-    pass
+def get_text_file(txt_files):
+    temp_dir = tempfile.TemporaryDirectory() 
+    temp_filepath = os.path.join(temp_dir.name, txt_files.name) 
+    with open(temp_filepath, "wb") as f:  
+        f.write(txt_files.getvalue()) 
+    text_loader = TextLoader(temp_filepath) 
+    text_file = text_loader.load() 
+    return text_file 
+
+
+def get_csv_file(csv_files):
+    temp_dir = tempfile.TemporaryDirectory() 
+    temp_filepath = os.path.join(temp_dir.name, csv_files.name) 
+    with open(temp_filepath, "wb") as f:  
+        f.write(csv_files.getvalue())
+    csv_loader = CSVLoader(temp_filepath) 
+    csv_file = csv_loader.load() 
+    return csv_file 
+
+def get_json_file(json_files):
+    temp_dir = tempfile.TemporaryDirectory()
+    temp_filepath = os.path.join(temp_dir.name, json_files.name) 
+    with open(temp_filepath, "wb") as f: 
+        f.write(json_files.getvalue()) 
+    json_loader = JSONLoader(temp_filepath, jq_schema='.messages[].content', text_content=False) 
+    json_file = json_loader.load() 
+    return json_file 
 
     
 # 문서들을 처리하여 텍스트 청크로 나누는 함수입니다.
@@ -94,8 +112,8 @@ def handle_userinput(user_question):
 
 def main():
     load_dotenv()
-    st.set_page_config(page_title="Chat with multiple Files",
-                       page_icon=":books:")
+    st.set_page_config(page_title="Assign8: Chat with multiple Files",
+                       page_icon=":random:")
     st.write(css, unsafe_allow_html=True)
 
     if "conversation" not in st.session_state:
@@ -103,7 +121,11 @@ def main():
     if "chat_history" not in st.session_state:
         st.session_state.chat_history = None
 
+<<<<<<< HEAD
+    st.header("Assign8: Chat with multiple Files :")
+=======
     st.header("Assign 8 -Chat with multiple Files :")
+>>>>>>> eee37261f0a756aa7ad43f502da4241637d89427
     user_question = st.text_input("Ask a question about your documents:")
     if user_question:
         handle_userinput(user_question)
@@ -115,7 +137,7 @@ def main():
 
         st.subheader("Your documents")
         docs = st.file_uploader(
-            "Upload your PDFs here and click on 'Process'", accept_multiple_files=True)
+            "Upload your Files here and click on 'Process'", accept_multiple_files=True)
         if st.button("Process"):
             with st.spinner("Processing"):
                 # get pdf text