edit codes

app.py

@@ -30,18 +30,40 @@ def get_pdf_text(pdf_docs):
 # 과제
 # 아래 텍스트 추출 함수를 작성
 
-def get_text_file(docs):
-    temp_dir2 = tempfile.TemporaryDirectory()  # 임시 디렉토리를 생성합니다.
-    temp_filepath2 = os.path.join(temp_dir2.name, docs.name)  # 임시 파일 경로를 생성합니다.
-    with open(temp_filepath2, "wb") as f: # 임시 파일을 바이너리 쓰기 모드로 엽니다.
-        f.write(docs.getvalue())  # TXT 문서의 내용을 임시 파일에 씁니다.
-    text_loader = TextLoader(
-        file_path=temp_filepath2,
-        txt_schema='',
-        text_content=False
-    )  # Use your specific text loader here.
-    text_data = text_loader.load()  # Extract text using the loader.
-    return text_data  # 추출한 텍스트를 반환합니다.
+# def get_text_file(docs):
+#     temp_dir2 = tempfile.TemporaryDirectory()  # 임시 디렉토리를 생성합니다.
+#     temp_filepath2 = os.path.join(temp_dir2.name, docs.name)  # 임시 파일 경로를 생성합니다.
+#     with open(temp_filepath2, "wb") as f: # 임시 파일을 바이너리 쓰기 모드로 엽니다.
+#         f.write(docs.getvalue())  # TXT 문서의 내용을 임시 파일에 씁니다.
+#     text_loader = TextLoader(
+#         file_path=temp_filepath2,
+#         txt_schema='',
+#         text_content=False
+#     )  # Use your specific text loader here.
+#     text_data = text_loader.load()  # Extract text using the loader.
+#     return text_data  # 추출한 텍스트를 반환합니다.
+def get_text_file(txt_filepath):
+    with open(txt_filepath, "r", encoding="utf-8") as file:
+        txt_content = file.read()  # Read the content of the TXT file
+
+    text_chunks = get_text_chunks([txt_content])  # Split the text into chunks
+    vectorstore = get_vectorstore(text_chunks)  # Create a vector store from text chunks
+
+    conversation_chain = get_conversation_chain(vectorstore)  # Create a conversation chain
+
+    # Assuming st.session_state.conversation is used to store conversation state
+    response = st.session_state.conversation({'question': txt_content})
+
+    # Assuming st.session_state.chat_history is used to store chat history
+    st.session_state.chat_history = response['chat_history']
+
+    for i, message in enumerate(st.session_state.chat_history):
+        if i % 2 == 0:
+            st.write(user_template.replace(
+                "{{MSG}}", message.content), unsafe_allow_html=True)
+        else:
+            st.write(bot_template.replace(
+                "{{MSG}}", message.content), unsafe_allow_html=True)
 
 def get_csv_file(docs):
     temp_dir3 = tempfile.TemporaryDirectory()
@@ -54,12 +76,12 @@ def get_csv_file(docs):
     #     csv_reader = csv.reader(csv_file)
     #     for row in csv_reader:
     #         csv_data.append(row)
-    csv_loader = CSVLoader(
-        file_path=temp_filepath3,
-        cs_schema='.data[].address',
-        text_content=False
-    )
-    csv_data = csv_loader.load()
+    # csv_loader = CSVLoader(
+    #     file_path=temp_filepath3,
+    #     cs_schema='.data[].address',
+    #     text_content=False
+    # )
+    # csv_data = csv_loader.load()
     return csv_data
 
 def get_json_file(docs):