Update app.py

app.py

@@ -62,19 +62,21 @@ def get_vectorstore(text_chunks):
 
 
 def get_conversation_chain(vectorstore):
-    gpt_model_name = 'gpt-3.5-turbo'
-    llm = ChatOpenAI(model_name = gpt_model_name) #gpt-3.5 모델 로드
-    
-    # 대화 기록을 저장하기 위한 메모리를 생성합니다.
-    memory = ConversationBufferMemory(
-        memory_key='chat_history', return_messages=True)
-    # 대화 검색 체인을 생성합니다.
-    conversation_chain = ConversationalRetrievalChain.from_llm(
-        llm=llm,
-        retriever=vectorstore.as_retriever(),
-        memory=memory
-    )
-    st.session_state.conversation = conversation_chain
+    if st.session_state.conversation is None:
+        gpt_model_name = 'gpt-3.5-turbo'
+        llm = ChatOpenAI(model_name=gpt_model_name)  # gpt-3.5 모델 로드
+
+        # 대화 기록을 저장하기 위한 메모리를 생성합니다.
+        memory = ConversationBufferMemory(
+            memory_key='chat_history', return_messages=True)
+        # 대화 검색 체인을 생성합니다.
+        conversation_chain = ConversationalRetrievalChain.from_llm(
+            llm=llm,
+            retriever=vectorstore.as_retriever(),
+            memory=memory
+        )
+        st.session_state.conversation = conversation_chain
+
     return st.session_state.conversation
 
 # 사용자 입력을 처리하는 함수입니다.
@@ -99,9 +101,8 @@ def main():
                        page_icon=":books:")
     st.write(css, unsafe_allow_html=True)
 
-    if "conversation" not in st.session_state:
+    if "conversation" not in st.session_state or st.session_state.conversation is None:
         st.session_state.conversation = None
-    if "chat_history" not in st.session_state:
         st.session_state.chat_history = None
 
     st.header("Chat with multiple Files :")
@@ -123,7 +124,6 @@ def main():
                 doc_list = []
 
                 for file in docs:
-                    print('file - type : ', file.type)
                     if file.type == 'text/plain':
                         # file is .txt
                         doc_list.extend(get_text_file(file))