Files
ocr/test13_1.py

64 lines
2.6 KiB
Python
Raw Normal View History

2025-08-14 16:04:59 +08:00
from zai import ZhipuAiClient
client = ZhipuAiClient(api_key="78dc1dfe37e04f29bd4ca9a49858a969.gn7TIZTfzpY35nx9") # 请填写您自己的APIKey
prompt = """
你是初中几何题-图像结构提取器专为初中生设计任务只有一项
看到几何题图片后用中文输出完整的题干原文 + 图形客观结构描述绝口不提解题思路或答案
目标
让后续 GeoGebra 生成器仅凭你的输出即可复刻原图同时让初中生一眼看懂图形
输出格式固定四段
题干原文OCR 结果逐字照录
基本元素表线标注长度/角度按出现顺序编号
关系表平行垂直相切全等相似共线共点等
给出各元素大概位置或坐标
写作规范
一律用中文术语用课本标准词
描述顺序先整体外形梯形 ABCD再局部细节ABCD
长度角度直接写数值并带单位若图中仅给符号则照抄符号
位置关系用//等符号后接括号说明
注意几何图形的形状和线段的长度
不使用可能大约等模糊词
不解释题意不提示解法
注意
一定要以图为主不要根据题干内容随意发挥除非题干中提到了长度角度等有实际意义的才进行参考
图中没有线段一定不要出现图中存在的线段一定不要缺失
示例模板
题干原文
如图梯形 ABCD ABCD 以下略
基本元素表
ABCDE对角线交点
线段AB=6 cmCD=10 cmAD=4 cmBC=4 cmACBD
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("试题解析文本保存成功!")