from openai import OpenAI

from Config.Config import ALY_LLM_API_KEY, ALY_LLM_MODEL_NAME, ALY_LLM_BASE_URL

if __name__ == '__main__':
    # 初始化OpenAI客户端
    client = OpenAI(
        api_key=ALY_LLM_API_KEY,
        base_url=ALY_LLM_BASE_URL
    )

    reasoning_content = ""  # 定义完整思考过程
    answer_content = ""  # 定义完整回复
    is_answering = False  # 判断是否结束思考过程并开始回复


    # 3、 分块追加：在上一步代码基础上，加一条动态垂直线段，连接圆上的动点与 x 轴，并显示线段长度数值。
    prompt = """ 
        你是一位资深 Manim 动画工程师，专注于教育场景。请遵循以下规范：
        1. 仅输出 Python 代码，不解释思路。
        2. 使用 Manim 社区版 v0.18.0 语法。
        3. 一个 Scene 对应一个完整教学环节，拆成 4 个私有方法：
           _intro()、_discovery()、_proof()、_practice()。
        4. 所有公式用 MathTex，文字用 Tex，颜色统一用 Manim 内置调色板。
        5. 关键坐标用常量定义在类顶部，例如 LEFT_MARGIN = 3.2。
        6. 每行不超过 88 字符，函数名用 snake_case。
        7. 动画时长 3-5 秒 / 步，默认 ease-in-out。
        8.请严格按照以下模板：坐标轴范围、颜色、字体统一使用 Manim 社区版默认主题。
        9. 代码的最后，参考下面的代码添加相应内容:
        if __name__ == "__main__":
            # 全局指定 ctex 模板（一次性）
            config.tex_template = TexTemplateLibrary.ctex
            # 使用临时配置渲染场景（配置只在with块内有效）
            with manim.tempconfig(config):
                # 实例化场景类
                scene = UnitCircleSin()
                # 执行渲染流程（包含文件生成和预览）
                scene.render()
        
        本成的具体需求如下：
        场景描述：用 Manim 做一个 10 秒动画，展示单位圆上角度 θ 从 0→2π 时，sinθ 的实时高度变化，背景暗色，线条黄色。
      """

    # 创建聊天完成请求
    completion = client.chat.completions.create(
        model=ALY_LLM_MODEL_NAME,
        messages=[
            {
                "role": "user",
                "content": [
                    {"type": "text",
                     "text": prompt},
                ],
            },
        ],
        stream=True,
    )

    print("\n" + "=" * 20 + "思考过程" + "=" * 20 + "\n")

    for chunk in completion:
        # 如果chunk.choices为空，则打印usage
        if not chunk.choices:
            print("\nUsage:")
            print(chunk.usage)
        else:
            delta = chunk.choices[0].delta
            # 打印思考过程
            if hasattr(delta, 'reasoning_content') and delta.reasoning_content != None:
                print(delta.reasoning_content, end='', flush=True)
                reasoning_content += delta.reasoning_content
            else:
                # 开始回复
                if delta.content != "" and is_answering is False:
                    print("\n" + "=" * 20 + "完整回复" + "=" * 20 + "\n")
                    is_answering = True
                # 打印回复过程
                print(delta.content, end='', flush=True)
                answer_content += delta.content

    # print("=" * 20 + "完整思考过程" + "=" * 20 + "\n")
    # print(reasoning_content)
    # print("=" * 20 + "完整回复" + "=" * 20 + "\n")
    print(answer_content)