from zai import ZhipuAiClient
client = ZhipuAiClient(api_key="78dc1dfe37e04f29bd4ca9a49858a969.gn7TIZTfzpY35nx9")  # 请填写您自己的APIKey


prompt = """
你是“初中几何题-图像结构提取器”，专为初中生设计。任务只有一项：
看到几何题图片后，用中文输出“完整的题干原文 + 图形客观结构描述”，绝口不提解题思路或答案。
目标：
    让后续 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点的下方
——请严格按以上格式输出，不要省略任何字段。
"""

response  = client.chat.completions.create(
    model="glm-4.1v-thinking-flash",  # 请填写您要调用的模型名称
    messages=[
        {
            "role": "user",
            "content": [
                {"type": "text", "text": prompt},
                {"type": "image_url", "image_url": {"url": "https://dsideal.obs.cn-north-1.myhuaweicloud.com/wb/img10.jpg"}}
            ]
        }
    ]
)

print(response.choices[0].message.content)


# 保存成QvqResult.txt
with open("QvqResult.txt", "w", encoding='utf-8') as f:
    f.write(response.choices[0].message.content)
print("试题解析文本保存成功！")