coding

app.py

@@ -88,11 +88,13 @@ def answer_question(repo_id, temperature, max_length, question):
         with st.spinner("正在筛选本地数据集..."):
             question_embedding = st.session_state.embeddings.embed_query(question)
             question_embedding_str = " ".join(map(str, question_embedding))
-            # print('question_embedding: ' + question_embedding_str)
             docs_and_scores = st.session_state.db.similarity_search_with_score(question_embedding_str)
 
-            context = "\n".join([doc.page_content for doc, _ in docs_and_scores])
-            print('context: ' + context)
+            context_list = []
+            for doc, score in docs_and_scores:
+                print(str(score) + ' : ' + doc.page_content)
+                context_list.append(doc.page_content)
+            context = "\n".join(context_list)
 
             prompt = f"请根据以下知识库回答问题：\n{context}\n问题：{question}"
             print('prompt: ' + prompt)
@@ -100,10 +102,10 @@ def answer_question(repo_id, temperature, max_length, question):
             st.success("本地数据集筛选完成！")
             print("本地数据集筛选完成！")
 
-        with st.spinner("正在生成答案..."):
+        with st.spinner("正在生成答案(基于本地数据集)..."):
             answer = get_answer(prompt)
-            st.success("答案生成完毕！")
-            print("答案生成完毕！")
+            st.success("答案生成完毕(基于本地数据集)！")
+            print("答案生成完毕(基于本地数据集)！")
         return {"prompt": prompt, "answer": answer, "pure_answer": pure_answer}
     except Exception as e:
         st.error(f"问答过程出错：{e}")
@@ -149,7 +151,7 @@ with col3:
         generate_answer(gemma, float(temperature), int(max_length), random_question)
 
 with col4:
-    question = st.text_area("请输入问题", "谁是潜水员？")
+    question = st.text_area("请输入问题", "太阳为什么是绿色的？")
     if st.button("提交输入的问题"):
         if not question:
             st.warning("请输入问题！")