coding

app.py

@@ -51,7 +51,6 @@ if not st.session_state.vector_created:
         st.stop()
     st.session_state.vector_created = True
 
-# 问答函数
 if "repo_id" not in st.session_state:
     st.session_state.repo_id = ''
 if "temperature" not in st.session_state:
@@ -64,10 +63,9 @@ def get_answer(prompt):
     # 去掉 prompt 的内容
     answer = answer.replace(prompt, "").strip()
     print(answer)
-    st.success("答案已经生成！")
-    print("答案已经生成！")
     return answer
 
+# 问答函数
 def answer_question(repo_id, temperature, max_length, question):
     # 初始化 Gemma 模型
     if repo_id != st.session_state.repo_id or temperature != st.session_state.temperature or max_length != st.session_state.max_length:
@@ -85,6 +83,10 @@ def answer_question(repo_id, temperature, max_length, question):
 
     # 获取答案
     try:
+        with st.spinner("正在生成答案(基于模型自身)..."):
+            pure_answer = get_answer(question)
+            st.success("答案生成完毕(基于模型自身)！")
+            print("答案生成完毕(基于模型自身)！")
         with st.spinner("正在筛选本地数据集..."):
             question_embedding = st.session_state.embeddings.embed_query(question)
             question_embedding_str = " ".join(map(str, question_embedding))
@@ -101,8 +103,9 @@ def answer_question(repo_id, temperature, max_length, question):
             print("本地数据集筛选完成！")
 
         with st.spinner("正在生成答案..."):
-            pure_answer = get_answer(question)
             answer = get_answer(prompt)
+            st.success("答案生成完毕！")
+            print("答案生成完毕！")
         return {"prompt": prompt, "answer": answer, "pure_answer": pure_answer}
     except Exception as e:
         st.error(f"问答过程出错：{e}")