import asyncio
import logging

from openai import AsyncOpenAI
from zai import ZhipuAiClient

from Config.Config import ZHIPU_API_KEY

# 设置日志
logger = logging.getLogger(__name__)

def create_llm_client() -> AsyncOpenAI:
    """初始化并返回异步OpenAI客户端"""
    return AsyncOpenAI(
        #api_key=LLM_API_KEY,
        #base_url=LLM_BASE_URL
        api_key="sk-f6da0c787eff4b0389e4ad03a35a911f",
        base_url="https://dashscope.aliyuncs.com/compatible-mode/v1"
    )

# 【5】如果在使用Segment创建两个点的线段时，需要指定一个标签名，然后使用ShowLabel隐藏这个标签名，注意使用的是ShowLabel而不是SetVisibleInView，而且只是Segment创建两个点的线段时才隐藏标签名
async def getGgbCommand(QvqResult):
    """异步版本的getGgb函数 - 支持流式输出"""
    custom_prompt = """ 
        # Role: GeoGebra 几何图形生成器
        ## Profile
        - language: 中文
        - description: 专门用于根据初中几何题的题干和几何图形的描述来生成Geogebra指令集的角色，确保生成的指令集严格遵循几何约束和动态坐标系构建规则，并具备错误熔断机制。
        - background: 拥有深厚的几何学和计算机科学背景，熟悉Geogebra软件的操作和指令集编写。
        - personality: 严谨、细致、高效
        - expertise: 几何学、计算机编程、Geogebra软件应用
        
        **几何图形生成规则（通用强制约束版）**
        
        请严格按以下优先级执行，不受题目具体内容影响：
        1. **几何约束最高优先级** 
           - 描述中提到的水平线段，一定要保持水平，不可以进行变更
           - 当文字描述与坐标/位置冲突时，以文字描述的几何关系为准
           - 必须自动验证：{长度|角度|中点|平行|垂直}关系
           - 验证失败时：重新计算坐标并报告修正
           - 有动点的，要设置滑块。没有动点的，不要设置滑块。
           - 检查描述中提到的最长的边做为X轴，最长边的左端点做为原点
        
        2. **动态坐标系构建**
           ```mermaid
           graph TB
           A[定位原点] --> B{原点坐标明确？}
           B -->|是| C[直接采用]
           B -->|否| D[根据相对关系推算]
           E[建立坐标系] --> F{存在水平/垂直要求？}
           F -->|是| G[调整坐标满足]
           F -->|否| H[保持默认方向]
        ```
        
        3、错误熔断机制
        
        检测到矛盾 → 触发修正流程：
        
        if not satisfy_constraint(description):
            new_coords = recalculate_by_geometry()
            log_adjustment(old_coords, new_coords, reason)
        
        4、元素生成协议
        
        顺序：固定点 → 固定线 → 动态点(初始位) → 动态元素
        
        依赖检查：确保父元素先于子元素生成
        
        5、输出结构（强制）
        
        [修正报告] （若有调整）
        [坐标摘要] （最终使用坐标）
        [Geogebra指令] （完整可执行代码，以```geogebra 开头，以 ```结束。【注意】```geogebra 前面一定要加上 \n 换行！） 
        注意：
        【1】GGB指令集中不要添加注释！不要添加除指令集外的其它描述性文字，需要可以直接批量执行的！
        【2】仔细检查原图中是线段还是直线，要严格进行识别！
        【3】仔细检查原图中的每一条边是否都正确进行了线段标识！
        【4】不显示网格（使用ShowGrid(false)），不显示坐标轴（使用ShowAxes(false)）        
        【5】注意不要使用Perpendicular，需要使用PerpendicularLine来替代
        【6】注意不要使用SetLabel，因为会提示“未知的指令”。
        【7】注意不要使用Angle，无需标注角度。
        【8】如果有滑块使用Slider
        【9】如果没有动点就无需添加滑块。
        【10】注意不要使用TangentPoint，因为会提示“未知的指令”。
        
        6、题目的文字描述信息
        ```
        {QvqResult}
        ```
    """

    try:
        # 创建异步OpenAI客户端
        client = create_llm_client()

        # 使用异步流式调用
        completion = await client.chat.completions.create(
            model="deepseek-r1-0528",
            #model="deepseek-reasoner",
            messages=[{"role": "user", "content": custom_prompt.replace("{QvqResult}", QvqResult)}],
            stream=True
        )

        is_answering = False
        async for chunk in completion:
            delta = chunk.choices[0].delta
            # 打印思考过程
            if hasattr(delta, 'reasoning_content') and delta.reasoning_content != None:
                yield delta.reasoning_content
            else:
                # 开始回复
                if delta.content != "" and is_answering is False:
                    print("\n" + "=" * 20 + "完整回复" + "=" * 20 + "\n")
                    is_answering = True
                # 打印回复过程
                yield delta.content
    except Exception as e:
        print(f"异步getGgb调用错误: {str(e)}")
        yield f"错误: {str(e)}"


async def translate_to_ggb(natural_language_cmd):
    """将自然语言指令翻译成GeoGebra指令集（异步版本）"""
    # 精简的prompt，专注于单句指令翻译
    custom_prompt = f"""
    请将以下自然语言指令翻译为GeoGebra指令集：
    "{natural_language_cmd}"
    
    要求：
    1. 仅返回GeoGebra指令，不包含任何解释性文字
    2. 指令必须可直接在GeoGebra中执行
    3. 保持指令简洁明确
    4. 以```geogebra 开头，以 ```结束 【注意】```geogebra 前面一定要加上 \n 换行！
    5. 用户提到的元素，比如点A,线段AB等，都视为已存在，不要重新创建    
    """

    try:
        # 创建异步OpenAI客户端
        client = create_llm_client()

        # 使用异步流式调用
        completion = await client.chat.completions.create(
            model="deepseek-r1-0528",
            messages=[{"role": "user", "content": custom_prompt}],
            stream=True
        )

        is_answering = False
        async for chunk in completion:
            delta = chunk.choices[0].delta
            # 打印思考过程
            if hasattr(delta, 'reasoning_content') and delta.reasoning_content != None:
                yield delta.reasoning_content
            else:
                # 开始回复
                if delta.content != "" and is_answering is False:
                    print("\n" + "=" * 20 + "完整回复" + "=" * 20 + "\n")
                    is_answering = True
                # 打印回复过程
                yield delta.content
    except Exception as e:
        print(f"翻译指令错误: {str(e)}")
        yield f"错误: {str(e)}"


# 示例用法
if __name__ == '__main__':
    # file_name = "QvqResult_14.txt"
    # with open(file_name, 'r', encoding='utf-8') as file:
    #     QvqResult = file.read()  # 直接获取整个文件内容为字符串
    #
    # async def test():
    #     async for content in getGgbCommand(QvqResult):
    #         if content:
    #             print(content, end="", flush=True)
    #
    #
    # asyncio.run(test())


    async def test_translate():
        async for content in translate_to_ggb("把A和B连接起来"):
            if content:
                print(content, end="", flush=True)


    # 运行测试
    asyncio.run(test_translate())



async def process_geometry_image(image_url: str):
    """
    真正的流式处理几何题图片
    """
    client = ZhipuAiClient(api_key=ZHIPU_API_KEY)

    prompt = """你是"初中几何题-图像结构提取器"，专为初中生设计。任务只有一项：
    看到几何题图片后，用中文输出"完整的题干原文 + 图形客观结构描述"，绝口不提解题思路或答案。
    【返回格式】：标准markdown+latex格式
    目标：
        让后续 GeoGebra 生成器仅凭你的输出即可复刻原图，同时让初中生一眼看懂图形。
        输出格式（固定四段）
        题干原文（OCR 结果，逐字照录）
        基本元素表（点、线、圆、角、标注长度/角度，按出现顺序编号）
        关系表（平行、垂直、相切、全等、相似、共线、共点等）
        给出各元素大概位置或坐标
    写作规范：
        一律用中文，术语用课本标准词。
        描述顺序：先整体外形（如"梯形 ABCD"），再局部细节（如"AB∥CD"）。
        长度、角度直接写数值并带单位；若图中仅给符号，则照抄符号。
        位置关系用"//、⊥、≅、∼、∈"等符号，后接括号说明。
        注意几何图形的形状和线段的长度
        不使用"可能、大约"等模糊词。
        不解释题意，不提示解法。
    注意：
        一定要以图为主，不要根据题干内容随意发挥。除非题干中提到了：长度、角度等有实际意义的才进行参考
        图中没有线段一定不要出现，图中存在的线段一定不要缺失。
    示例模板：
    【题干原文】
    如图，梯形 ABCD 中，AB∥CD …（以下略）
    【基本元素表】
    点：A、B、C、D、E（对角线交点）
    线段：AB=6 cm，CD=10 cm，AD=4 cm，BC=4 cm，AC，BD
    角：∠DAB=60°，∠ABC=120°
    圆：无
    【关系表】
    AB // CD
    AD = BC
    AC 与 BD 交于 E
    ∠DAB 与 ∠ABC 互补
    【元素位置】
    A点在左上，B点在右上，C点在A点和B点的下方
    ——请严格按以上格式输出，不要省略任何字段。"""

    try:
        messages = [{
            "role": "user",
            "content": [
                {"type": "text", "text": prompt},
                {"type": "image_url", "image_url": {"url": image_url}}
            ]
        }]

        # 使用智谱SDK的流式调用（同步调用包装为异步）
        stream = await asyncio.to_thread(
            client.chat.completions.create,
            model="glm-4.1v-thinking-flash",
            messages=messages,
            stream=True
        )

        # 异步处理流式输出
        for chunk in stream:
            if chunk.choices and chunk.choices[0].delta.content is not None:
                content = chunk.choices[0].delta.content
                yield content

    except Exception as e:
        logger.error(f"流式处理图片时发生错误: {str(e)}")
        yield ""