o bi#@sddlZddlZddlmZddlZddlmZmZddlZddl Z ddl m Z e e ZddZddZd d Ze d krEd ZeedSdS) N) natsorted)TONGYI_API_KEY OCR_MODEL)OpenAIc Cs tdtdt|td|d|ddd|id|igdg}td zt}td ttjjt t|d d }t|}td ||j dkrZt d|j |j WdStd|j t |drt |jdrt|jjdkrd|jjdvrd|jjdjvrt|jjdjddkr|jjdjddd}t|jdd}t|jdd} td|| t|tdt|dkr|dddn|||| fWSt dtd|WdSty} ztj d t| d!d"t d#tWYd} ~ dSd} ~ ww)$uM 调用通义千问OCR接口识别图片中的表格内容 Args: system_prompt (str): 系统提示词 ocr_result (str): OCR识别结果的描述 image_url (str): 图片URL地址 Returns: tuple: (识别结果内容, 输入token数, 输出token数) 或 (None, 0, 0) 当调用失败时 u!开始调用通义千问OCR接口#OCR系统提示词长度: %d 字符 图片URL: %ssystemZrolecontentuserimagetextu-构建完成多模态消息，准备调用APIu.发送OCR请求到通义千问API，模型: %s皙?)api_keymodelmessages temperatureu-通义千问API调用完成，耗时: %.2f秒uC通义千问OCR接口调用失败，状态码: %d, 错误信息: %s)NrruAPI响应状态码: %doutputchoicesrmessager input_tokens output_tokensuQOCR识别成功，输入token: %d, 输出token: %d, 响应内容长度: %d 字符uOCR识别结果前100字符: %sdN...u0API响应格式异常，无法提取有效内容u完整响应对象: %su.调用通义千问OCR接口时发生异常: %sTexc_info异常堆栈信息: %s)loggerinfodebuglentimer dashscopeZMultiModalConversationcallr status_codeerrorr hasattrrrrgetattrusage Exceptionstr traceback format_exc) system_prompt ocr_result image_urlr start_timeresponse api_call_timer rrer57/home/project/hydrological_data_analyse/src/call_llm.pycall_tongyi_ocrsv          (   r7c csBtdtdt|td|tdt|d|dddd |id d |d gdg}zttd d}t}tdt|jj j t|dddidd}t|}td|g}d} d} td|D]R} | j r| j dj j dur| j dj j } || tdt| dkr| dddn| | Vqi| jrt| jdd} t| jdd} t| jdd} td| | | qid|}td | | t|td!t|d"kr|dd"dn|| | fVWdWdS1swYWdSty }ztjd#t|dd$td%td&VWYd}~dSd}~ww)'u4 调用通义千问OCR接口识别图片中的表格内容（流式版本） 优化点： 1. 分离内容流式返回和token统计功能 2. 简化yield逻辑，专注于内容流 3. 使用上下文管理器确保资源清理 4. 添加更详细的错误处理 Args: system_prompt (str): 系统提示词 ocr_result (str): OCR识别结果的描述 image_url (str): 图片URL地址 Yields: str: 内容片段（流式返回） tuple: (输入token数, 输出token数) 在流结束时返回 u3开始调用通义千问OCR接口（流式模式）rru OCR结果描述长度: %d 字符rr r r0url)typer0r )r9r z1https://dashscope.aliyuncs.com/compatible-mode/v1)rbase_urlu4发送OCR流式请求到通义千问API，模型: %sTZ include_usager)rrstreamZstream_optionsru3通义千问流式API连接建立，耗时: %.2f秒ru开始处理流式响应Nu收到内容片段: %s2rZ prompt_tokensZcompletion_tokens total_tokensuA收到用量统计，输入token: %d, 输出token: %d, 总计: %duZOCR流式识别完成，总输入token: %d, 总输出token: %d, 总内容长度: %d 字符uOCR识别完整结果: %sru4调用通义千问OCR流式接口时发生异常: %srr)rr)rrr r!rrr"rZchatZ completionscreaterdeltar appendr)r(joinr*r&r+r,r-)r.r/r0rclientr1r;r3 content_partsrrchunkr r=Z full_contentr4r5r5r6call_tongyi_ocr_streambs|         (  ( &1rFc Csd}tt|D]e}|dvrq t}td|dd|ds%q |ddd }d |}t|d |}t tj ||d d dd }| |Wdn1sWwYtdt|ddt dq dS)Nu 你是一个专业的OCR工具，识别图片里表格的内容： 1. 检测出图片里完整的表格，如果表格不完整，则直接忽略 2. 表格数据前31行表示一年中每一天的数据，竖向记录，一列表示一个月，识别时也要竖向识别，按月份返回数据。切记这是真实数据，只有一月、三月、五月、七月、八月、十月、十二月，有31天。二月只有28或者29天，其余月份只有30天 3. 第32行记录的是平均数，若均值不存在，则对应值记录为None 数据返回格式为:{"1月": {"data": [0.1, 0.2], "avg": 0.3}, "2月": {...}}，每个月份数据的长度，要与这个月的天数相符。如果某个月份存在数据缺失，则用null进行填充 )z102.jpgz103.jpgu处理图片: r>)end)z.pngz.jpgz.jpeg.rzGhttps://questions-1309102452.cos.ap-nanjing.myqcloud.com/images/ruiwen/u识别图中的表格z_processed.jsonwzutf-8)encodingu ，耗时: z.2fu 秒)roslistdirr"printlowerendswithrsplitr7openpathrBwritesleep) image_pathr.filenamer1Z file_prefixr0resfr5r5r6llm_ocrs"     r[__main__z*D:\project\hydrological_data_analyse/input)rMr#Znatsortrr"varrrloggingr,Zopenair getLogger__name__rr7rFr[rWr5r5r5r6s    Tg