Update app.py

app.py

@@ -89,12 +89,11 @@ def load_file(file_obj):
 
     return f"已加载 {len(docs)} 个文本块", None
 
-# ===== RAG 查询 =====
 def rag_query(query):
     if index is None or not docs:
         return "请先上传文件并构建知识库"
     q_emb = embed_text(query).reshape(1, -1)
-    D, I = index.search(q_emb, k=8)  # Top-K=8
+    D, I = index.search(q_emb, k=8)
     retrieved = [docs[i]["text"] for i in I[0]]
     context = "\n".join([f"[{idx+1}] {txt}" for idx, txt in enumerate(retrieved)])
 
@@ -103,17 +102,18 @@ def rag_query(query):
 
 问题：{query}
 
-要求：
-1. 整合所有引用片段的信息回答
-2. 无法回答时直接说“我不知道”
-3. 在回答中标注引用的片段编号
+输出要求：
+1. 先用 2-3 句话总结所有引用片段的关键信息，形成一个完整结论
+2. 再分段详细说明，并在细节中标注引用编号
+3. 无法回答时直接说“我不知道”
 """
 
-    result = generator(prompt, max_length=500, do_sample=False)
+    result = generator(prompt, max_length=600, do_sample=False)
     answer = result[0]["generated_text"]
 
     return f"回答：\n{answer}\n\n参考片段：\n{context}"
 
+
 # ===== Gradio 界面 =====
 with gr.Blocks() as demo:
     gr.Markdown("## 📚 完整性增强版 RAG（PDF/TXT/EPUB 支持 + 引用显示）")