Parcourir la source

feat: 添加人设图谱可视化到构建流程

- 新增步骤7: visualization/build.py 生成人设图谱可视化HTML
- 流程从6步增加到7步

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude <noreply@anthropic.com>
yangxiaohui il y a 6 heures
Parent
commit
516fcf5d21
1 fichiers modifiés avec 15 ajouts et 4 suppressions
  1. 15 4
      script/data_processing/run_graph_pipeline.sh

+ 15 - 4
script/data_processing/run_graph_pipeline.sh

@@ -13,7 +13,8 @@
 #   7. build_persona_graph.py     - 构建人设图谱
 #   8. build_match_graph.py       - 构建匹配图谱
 #   9. build_post_graph.py        - 构建帖子图谱
-#  10. visualize_match_graph.py   - 生成可视化HTML
+#  10. visualize_match_graph.py   - 生成匹配图谱可视化HTML
+#  11. visualization/build.py     - 生成人设图谱可视化HTML
 #
 # 使用方式:
 #   ./run_graph_pipeline.sh              # 使用默认账号
@@ -46,7 +47,7 @@ run_step() {
     local step_name=$2
     local script_name=$3
 
-    print_step "$step_num/6" "$step_name"
+    print_step "$step_num/7" "$step_name"
 
     if python "script/data_processing/$script_name"; then
         print_success "$step_name 完成"
@@ -88,8 +89,18 @@ process_account() {
     # 步骤9: 构建帖子图谱
     run_step 5 "构建帖子图谱" "build_post_graph.py" || return 1
 
-    # 步骤10: 生成可视化HTML
-    run_step 6 "生成可视化HTML" "visualize_match_graph.py" || return 1
+    # 步骤10: 生成匹配图谱可视化HTML
+    run_step 6 "生成匹配图谱可视化" "visualize_match_graph.py" || return 1
+
+    # 步骤11: 生成人设图谱可视化HTML
+    print_step "7/7" "生成人设图谱可视化"
+    if python "script/visualization/build.py"; then
+        print_success "生成人设图谱可视化 完成"
+        echo ""
+    else
+        print_error "生成人设图谱可视化 失败"
+        return 1
+    fi
 
     echo "=========================================="
     print_success "图谱构建与可视化流程完成!"