Update app.py

app.py

@@ -89,6 +89,7 @@ def load_file(file_obj):
 
     return f"已加载 {len(docs)} 个文本块", None
 
+# ===== RAG 查询 =====
 def rag_query(query):
     if index is None or not docs:
         return "请先上传文件并构建知识库"
@@ -102,27 +103,32 @@ def rag_query(query):
 
 问题：{query}
 
-输出要求：
-1. 先用 2-3 句话总结所有引用片段的关键信息，形成一个完整结论
-2. 再分段详细说明，并在细节中标注引用编号
-3. 无法回答时直接说“我不知道”
+请严格按照以下格式输出：
+【结论】
+用 2-3 句话总结所有引用片段的关键信息，形成一个完整结论。
+
+【详细说明】
+整合所有引用片段的细节，分段描述，并在每个关键信息后标注引用编号。
+无法回答时直接说“我不知道”。
+
+【引用片段】
+逐条列出引用编号及对应的原文。
 """
 
-    result = generator(prompt, max_length=600, do_sample=False)
+    result = generator(prompt, max_length=800, do_sample=False)
     answer = result[0]["generated_text"]
 
-    return f"回答：\n{answer}\n\n参考片段：\n{context}"
-
+    return answer
 
 # ===== Gradio 界面 =====
 with gr.Blocks() as demo:
-    gr.Markdown("## 📚 完整性增强版 RAG（PDF/TXT/EPUB 支持 + 引用显示）")
+    gr.Markdown("## 📚 完整性增强版 RAG（PDF/TXT/EPUB 支持 + 结论 + 引用）")
     with gr.Row():
         file_input = gr.File(label="上传 PDF / TXT / EPUB 文件")
         load_btn = gr.Button("构建知识库")
     status = gr.Textbox(label="状态")
     query_input = gr.Textbox(label="输入你的问题")
-    answer_output = gr.Textbox(label="回答", lines=12)
+    answer_output = gr.Textbox(label="回答", lines=15)
     load_btn.click(load_file, inputs=file_input, outputs=status)
     query_input.submit(rag_query, inputs=query_input, outputs=answer_output)