Update app.py

app.py

@@ -97,7 +97,8 @@ st.write(f"向量库构建结束：{end_3}")
 st.write(f"向量库构建耗时：{end_3 - start_3}")
 
 if "directory_path" not in st.session_state:
-            st.session_state.directory_path = generate_random_string(20)
+    st.session_state.directory_path = generate_random_string(20)
+    os.makedirs(st.session_state.directory_path)
     
 st.session_state.new_index.storage_context.persist("st.session_state.directory_path")
 
@@ -114,16 +115,14 @@ st.write(f"向量库装载耗时：{end_4 - start_4}")
 
 if "query_engine" not in st.session_state:
     st.session_state.query_engine = st.session_state.loadedindex.as_query_engine()
-    os.makedirs(st.session_state.directory_path)
     
-if "user_question" not in st.session_state:
-    st.session_state.user_question = st.text_input("Enter your query:")
-if st.session_state.user_question !="" and not st.session_state.user_question.strip().isspace() and not st.session_state.user_question == "" and not st.session_state.user_question.strip() == "" and not st.session_state.user_question.isspace():
-    print("user question: "+st.session_state.user_question)
+user_question = st.text_input("Enter your query:")
+if user_question !="" and not user_question.strip().isspace() and not user_question == "" and not user_question.strip() == "" and not user_question.isspace():
+    print("user question: "+user_question)
     with st.spinner("AI Thinking...Please wait a while to Cheers!"):
         start_5 = timeit.default_timer() # Start timer
         st.write(f"Query Engine - AI QA开始：{start_5}") 
-        initial_response = st.session_state.query_engine.query(st.session_state.user_question)
+        initial_response = st.session_state.query_engine.query(user_question)
         temp_ai_response=str(initial_response)
         final_ai_response=temp_ai_response.partition('<|end|>')[0] 
         print("AI Response:\n"+final_ai_response)