8042750d41
* stage academic conversation * stage document conversation * fix buggy gradio version * file dynamic load * merge more academic plugins * accelerate nltk * feat: 为predict函数添加文件和URL读取功能 - 添加URL检测和网页内容提取功能，支持自动提取网页文本 - 添加文件路径识别和文件内容读取功能，支持private_upload路径格式 - 集成WebTextExtractor处理网页内容提取 - 集成TextContentLoader处理本地文件读取 - 支持文件路径与问题组合的智能处理 * back * block unstable --------- Co-authored-by: XiaoBoAI <liuboyin2019@ia.ac.cn>
48 lines
1.5 KiB
Python
48 lines
1.5 KiB
Python
class MarkdownFormatter:
|
|
"""Markdown格式文档生成器 - 用于生成对话记录的markdown文档"""
|
|
|
|
def __init__(self):
|
|
self.content = []
|
|
|
|
def _add_content(self, text: str):
|
|
"""添加正文内容"""
|
|
if text:
|
|
self.content.append(f"\n{text}\n")
|
|
|
|
def create_document(self, question: str, answer: str, ranked_papers: list = None) -> str:
|
|
"""创建完整的Markdown文档
|
|
Args:
|
|
question: str, 用户问题
|
|
answer: str, AI回答
|
|
ranked_papers: list, 排序后的论文列表
|
|
Returns:
|
|
str: 生成的Markdown文本
|
|
"""
|
|
content = []
|
|
|
|
# 添加问答部分
|
|
content.append("## 问题")
|
|
content.append(question)
|
|
content.append("\n## 回答")
|
|
content.append(answer)
|
|
|
|
# 添加参考文献
|
|
if ranked_papers:
|
|
content.append("\n## 参考文献")
|
|
for idx, paper in enumerate(ranked_papers, 1):
|
|
authors = ', '.join(paper.authors[:3])
|
|
if len(paper.authors) > 3:
|
|
authors += ' et al.'
|
|
|
|
ref = f"[{idx}] {authors}. *{paper.title}*"
|
|
if paper.venue_name:
|
|
ref += f". {paper.venue_name}"
|
|
if paper.year:
|
|
ref += f", {paper.year}"
|
|
if paper.doi:
|
|
ref += f". [DOI: {paper.doi}](https://doi.org/{paper.doi})"
|
|
|
|
content.append(ref)
|
|
|
|
return "\n\n".join(content)
|