from __future__ import annotations

import re
from textwrap import shorten

from openai import OpenAI

from app.config import settings
from app.schemas import RewriteRequest, RewriteResponse


SYSTEM_PROMPT = """
你是中文内容编辑与合规顾问。请把输入内容进行“原创改写”，要求：
1) 保留核心事实，但避免逐句复述；
2) 结构清晰：导语、3-5个小节、结尾行动建议；
3) 风格适合微信公众号，表达自然，避免AI腔；
4) 如果原文存在未经核实结论，请使用“可能/有待验证”等措辞；
5) 输出必须是 JSON，字段：title, summary, body_markdown。
""".strip()


class AIRewriter:
    def __init__(self) -> None:
        self._client = None
        if settings.openai_api_key:
            self._client = OpenAI(
                api_key=settings.openai_api_key,
                base_url=settings.openai_base_url,
            )

    def rewrite(self, req: RewriteRequest) -> RewriteResponse:
        if not self._client:
            return self._fallback_rewrite(req)

        user_prompt = f"""
原始内容：
{req.source_text}

改写约束：
- 标题参考：{req.title_hint or '自动生成'}
- 目标语气：{req.tone}
- 目标读者：{req.audience}
- 必须保留观点：{req.keep_points or '无'}
- 避免词汇：{req.avoid_words or '无'}
""".strip()

        completion = self._client.responses.create(
            model=settings.openai_model,
            input=[
                {"role": "system", "content": SYSTEM_PROMPT},
                {"role": "user", "content": user_prompt},
            ],
            text={"format": {"type": "json_object"}},
        )

        text = completion.output_text
        import json

        data = json.loads(text)
        return RewriteResponse(**data)

    def _fallback_rewrite(self, req: RewriteRequest) -> RewriteResponse:
        clean_text = re.sub(r"\n{2,}", "\n", req.source_text.strip())
        lines = [line.strip() for line in clean_text.split("\n") if line.strip()]
        head = lines[0] if lines else clean_text[:50]
        title = req.title_hint.strip() or f"{shorten(head, width=26, placeholder='')}：可执行解读"
        summary = shorten(clean_text, width=90, placeholder="...")
        body = (
            f"## 导语\n"
            f"这篇内容值得关注的核心在于：{summary}\n\n"
            f"## 重点拆解\n"
            f"1. 背景与问题：从原文可以看到关键矛盾已出现。\n"
            f"2. 方法与动作：建议按“目标-路径-验证”三步推进。\n"
            f"3. 风险与边界：避免绝对化表述，必要时补充数据来源。\n\n"
            f"## 公众号改写正文\n"
            f"{clean_text}\n\n"
            f"## 结尾\n"
            f"以上为原创重组版本，可继续补充案例与数据后发布。"
        )
        return RewriteResponse(title=title, summary=summary, body_markdown=body)