Upload app.py

app.py

@@ -134,6 +134,12 @@ def handle_userinput(user_question):
         else:
             st.write(bot_template.replace(
                 "{{MSG}}", message.content), unsafe_allow_html=True)
+text_chunks = []
+def initialize_conversation_chain():
+    # Add the necessary code to initialize the conversation_chain
+    # This may include loading the LlamaCpp model and creating the conversation_chain
+    vectorstore = get_vectorstore(text_chunks)  # Replace this with the appropriate code
+    return get_conversation_chain(vectorstore)
 
 
 def main():
@@ -142,14 +148,24 @@ def main():
                        page_icon=":books:")
     st.write(css, unsafe_allow_html=True)
 
-    if "conversation" not in st.session_state:
-        st.session_state.conversation = None
-    if "chat_history" not in st.session_state:
-        st.session_state.chat_history = None
+    # 대화 체인이 세션 상태에 없거나 None인 경우 초기화합니다.
+    if "conversation" not in st.session_state or st.session_state.conversation is None:
+        # 적절한 데이터로 text_chunks를 정의해야 합니다.
+        st.session_state.conversation = initialize_conversation_chain(text_chunks)
+    # if "conversation" not in st.session_state:
+    #     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:")
     user_question = st.text_input("Ask a question about your documents:")
+    # if user_question:
+    #     handle_userinput(user_question)
     if user_question:
+        # Ensure that conversation_chain is initialized before calling handle_userinput
+        if st.session_state.conversation is None:
+            st.session_state.conversation = initialize_conversation_chain()
+
         handle_userinput(user_question)
 
     with st.sidebar: