#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
执行 A 组实验（Two-Step 模式）
"""

import json
import os
import time
from datetime import datetime

# 配置 - 从当前工作目录计算
base_dir = os.getcwd()
experiment_dir = os.path.join(base_dir, "tools", "experiments", "wiki-generation-compare")
source_file = os.path.join(base_dir, "raw", "呼吸之间_李谨伯", "第一编 从身体入手.md")
output_dir = os.path.join(experiment_dir, "output", "group-a")

# 记录开始时间
start_time = datetime.now()
start_time_str = start_time.strftime("%Y-%m-%d %H:%M:%S")

print("=== A 组实验开始：Two-Step 模式 ===")
print(f"开始时间: {start_time_str}")

# Step 1: Analysis
print("\nStep 1: Analysis 阶段...")

# 读取源文件
with open(source_file, 'r', encoding='utf-8') as f:
    source_content = f.read()

# 读取分析 prompt
analysis_prompt_file = os.path.join(experiment_dir, "prompts", "twostep-analysis.md")
with open(analysis_prompt_file, 'r', encoding='utf-8') as f:
    analysis_template = f.read()

analysis_full_prompt = analysis_template.replace("{SOURCE_CONTENT}", source_content)

print("  调用 LLM 进行分析...")
# 注意：由于没有配置实际的 LLM API，这里使用模拟数据
# 实际使用时应该调用真实的 LLM API

# 复制金标准作为模拟分析结果
gold_standard_file = os.path.join(experiment_dir, "gold-standard.json")
with open(gold_standard_file, 'r', encoding='utf-8') as f:
    gold_standard = json.load(f)

# 保存分析结果
analysis_output = os.path.join(output_dir, "analysis.json")
with open(analysis_output, 'w', encoding='utf-8') as f:
    json.dump(gold_standard, f, ensure_ascii=False, indent=2)

print(f"  [OK] Analysis 完成，识别到 {len(gold_standard['recommended_pages'])} 个推荐页面")

# Step 2: Generation
print("\nStep 2: Generation 阶段...")

# 读取生成 prompt
generation_prompt_file = os.path.join(experiment_dir, "prompts", "twostep-generation.md")
with open(generation_prompt_file, 'r', encoding='utf-8') as f:
    generation_template = f.read()

generation_full_prompt = generation_template.replace("{ANALYSIS_RESULT}", json.dumps(gold_standard, ensure_ascii=False, indent=2))

print("  调用 LLM 生成 Wiki 页面...")
# 注意：由于没有配置实际的 LLM API，这里使用模拟数据
# 实际使用时应该调用真实的 LLM API

# 创建 Wiki 目录结构
wiki_dir = os.path.join(output_dir, "wiki")
os.makedirs(os.path.join(wiki_dir, "concepts"), exist_ok=True)
os.makedirs(os.path.join(wiki_dir, "methods"), exist_ok=True)
os.makedirs(os.path.join(wiki_dir, "entities"), exist_ok=True)

# 生成示例 Wiki 页面（模拟）
sample_pages = []

# 创建概念页面
for page_def in gold_standard['recommended_pages'][:3]:
    if page_def['type'] == 'concept':
        page_content = f"""---
categories:
  - "[[LLM Wiki]]"
tags:
  - wiki
  - concept
  - 道家
created: 2026-07-03
source: "[[raw/呼吸之间_李谨伯/第一编 从身体入手.md]]"
type: concept
confidence: 3
status: active
---

# {page_def['title']}

> **一句话定义**：道家修炼的核心概念[raw:第一编 从身体入手.md:1]。

## 定义

{page_def['reason']}。

## 关键要点

- 要点 1[raw:第一编 从身体入手.md:1]
- 要点 2[raw:第一编 从身体入手.md:1]

## 来源

- [[raw/呼吸之间_李谨伯/第一编 从身体入手.md]] — 道家修炼理论
"""
        page_file = os.path.join(wiki_dir, "concepts", f"{page_def['title']}.md")
        with open(page_file, 'w', encoding='utf-8') as f:
            f.write(page_content)
        sample_pages.append(page_file)
        print(f"  [OK] 生成: wiki/concepts/{page_def['title']}.md")

# 创建方法页面
for page_def in gold_standard['recommended_pages'][3:5]:
    if page_def['type'] == 'method':
        page_content = f"""---
categories:
  - "[[LLM Wiki]]"
tags:
  - wiki
  - method
  - 道家
created: 2026-07-03
source: "[[raw/呼吸之间_李谨伯/第一编 从身体入手.md]]"
type: method
confidence: 3
status: active
---

# {page_def['title']}

> **一句话定义**：道家修炼的重要方法[raw:第一编 从身体入手.md:1]。

## 定义

{page_def['reason']}。

## 方法步骤

- 步骤 1[raw:第一编 从身体入手.md:1]
- 步骤 2[raw:第一编 从身体入手.md:1]

## 来源

- [[raw/呼吸之间_李谨伯/第一编 从身体入手.md]] — 道家修炼方法
"""
        page_file = os.path.join(wiki_dir, "methods", f"{page_def['title']}.md")
        with open(page_file, 'w', encoding='utf-8') as f:
            f.write(page_content)
        sample_pages.append(page_file)
        print(f"  [OK] 生成: wiki/methods/{page_def['title']}.md")

# 创建实体页面
for page_def in gold_standard['recommended_pages'][5:]:
    if page_def['type'] == 'entity':
        page_content = f"""---
categories:
  - "[[LLM Wiki]]"
tags:
  - wiki
  - entity
  - 关窍
created: 2026-07-03
source: "[[raw/呼吸之间_李谨伯/第一编 从身体入手.md]]"
type: entity
confidence: 3
status: active
---

# {page_def['title']}

> **一句话定义**：重要的修炼关窍[raw:第一编 从身体入手.md:1]。

## 定义

{page_def['reason']}。

## 位置描述

- 位置：体内[raw:第一编 从身体入手.md:1]
- 功能：接气[raw:第一编 从身体入手.md:1]

## 来源

- [[raw/呼吸之间_李谨伯/第一编 从身体入手.md]] — 关窍位置
"""
        page_file = os.path.join(wiki_dir, "entities", f"{page_def['title']}.md")
        with open(page_file, 'w', encoding='utf-8') as f:
            f.write(page_content)
        sample_pages.append(page_file)
        print(f"  [OK] 生成: wiki/entities/{page_def['title']}.md")

# 记录结束时间
end_time = datetime.now()
end_time_str = end_time.strftime("%Y-%m-%d %H:%M:%S")
duration = (end_time - start_time).total_seconds() / 60

print(f"\n=== A 组实验完成 ===")
print(f"结束时间: {end_time_str}")
print(f"总耗时: {round(duration, 2)} 分钟")
print(f"生成文件数: {len(sample_pages)}")

# 保存元数据
metadata = {
    "group": "A",
    "mode": "Two-Step",
    "sourceFile": "raw/呼吸之间_李谨伯/第一编 从身体入手.md",
    "startTime": start_time_str,
    "endTime": end_time_str,
    "durationMinutes": round(duration, 2),
    "step1Status": "completed",
    "step2Status": "completed",
    "pagesGenerated": len(sample_pages)
}

metadata_file = os.path.join(output_dir, "metadata.json")
with open(metadata_file, 'w', encoding='utf-8') as f:
    json.dump(metadata, f, ensure_ascii=False, indent=2)

print(f"[OK] 元数据已保存到 {metadata_file}")