Upload app.py with huggingface_hub

app.py

@@ -116,8 +116,40 @@ def format_report(result: dict) -> str:
     if categories:
         md += f"| **分类** | {', '.join(categories)} |\n"
 
-    # ===== 第二部分：研究方向全景 =====
-    md += "\n---\n\n## 二、研究方向全景\n\n"
+    # ===== 论文自身代码评估 =====
+    own_repo = None
+    other_repos = []
+    for r in repos:
+        if r.get("method_family") == "本文代码":
+            own_repo = r
+        else:
+            other_repos.append(r)
+
+    if own_repo is not None:
+        ev = own_repo.get("evaluation", {})
+        md += "\n---\n\n## 二、本文代码复现评估\n\n"
+        md += "> 该论文是否提供了官方代码？代码质量如何？能否直接复现论文实验结果？\n\n"
+        md += f"| 仓库 | Stars | 综合评分 | 可复现性 | 对比实验适配度 |\n"
+        md += f"|------|:----:|:----:|:----:|:----:|\n"
+        verdict_icon = {"reproducible": "✅", "partially": "⚠️", "not_reproducible": "❌", "error": "💥"}.get(ev.get("verdict", ""), "❓")
+        readiness_icon = {"ready": "🟢", "partial": "🟡", "not_ready": "🔴"}.get(ev.get("benchmark_readiness", ""), "⚪")
+        md += f"| [{own_repo['full_name']}]({own_repo.get('html_url', '#')}) | {own_repo.get('stars', 0)} | **{ev.get('overall_score', 'N/A')}/100** | {verdict_icon} | {readiness_icon} |\n\n"
+        md += f"### 📋 详细评估\n\n{ev.get('reasoning', 'N/A')}\n\n"
+
+        risks = ev.get("risks", [])
+        if risks:
+            md += "### ⚠️ 风险点\n\n"
+            for r_ in risks:
+                md += f"- {r_}\n"
+            md += "\n"
+        md += f"### 💡 使用建议\n\n{ev.get('suggested_use', 'N/A')}\n\n"
+    else:
+        md += "\n---\n\n## 二、本文代码复现评估\n\n"
+        md += "> 🔴 **未找到该论文的官方开源代码。**\n>\n"
+        md += "> 这意味着复现该论文的实验结果需要从头实现，难度较高。建议在对比实验中使用下面列出的同一方法族的开源实现作为替代。\n\n"
+
+    # ===== 第三部分：研究方向全景 =====
+    md += "\n---\n\n## 三、研究方向全景\n\n"
     md += f"### 🎯 子领域定位\n\n**{direction.get('subfield', '未知')}**\n\n"
     
     trend = direction.get('subfield_trend', '暂无分析')
@@ -149,15 +181,18 @@ def format_report(result: dict) -> str:
 
     md += "\n"
 
-    # ===== 第三部分：仓库评估排名 =====
-    md += "\n---\n\n## 三、开源实现评估排名\n\n"
+    # ===== 第四部分：仓库评估排名 =====
+    md += "\n---\n\n## 四、开源实现评估排名\n\n"
     md += "> 评分逻辑：可复现性（80分）评估代码能否被他人成功跑通；对比实验适配度（20分）评估该仓库能否直接用于论文 Experiment section。\n\n"
 
-    if not repos:
+    if not other_repos and not own_repo:
         md += "**未找到相关开源仓库。**\n\n"
         return md
 
-    for i, repo in enumerate(repos):
+    if not other_repos:
+        md += "**未找到其他开源仓库（仅找到论文自身代码）。**\n\n"
+
+    for i, repo in enumerate(other_repos):
         full_name = repo.get("full_name", f"仓库 {i+1}")
         html_url = repo.get("html_url", "")
         ev = repo.get("evaluation", {})
@@ -219,13 +254,14 @@ def format_report(result: dict) -> str:
             md += f" | 匹配词: `{repo.get('match_keyword', '')}`"
         md += "\n\n---\n\n"
 
-    # ===== 第四部分：对比实验推荐总结 =====
-    md += "## 四、对比实验推荐总结\n\n"
+    # ===== 第五部分：对比实验推荐总结 =====
+    md += "## 五、对比实验推荐总结\n\n"
 
-    ready = [r for r in repos if r["evaluation"].get("benchmark_readiness") == "ready"]
-    partial = [r for r in repos if r["evaluation"].get("benchmark_readiness") == "partial"]
+    all_evaluated = other_repos + ([own_repo] if own_repo else [])
+    ready = [r for r in all_evaluated if r["evaluation"].get("benchmark_readiness") == "ready"]
+    partial = [r for r in all_evaluated if r["evaluation"].get("benchmark_readiness") == "partial"]
     not_ready = [
-        r for r in repos
+        r for r in all_evaluated
         if r["evaluation"].get("benchmark_readiness") not in ("ready", "partial")
     ]