o 0j @shdZddlmZmZmZddlmZdededefdd Z dd ed ee d edeeeffddZ dS)z'Shared utilities for DocVLM predictors.)AnyDictList) MODEL_GROUP model_name group_namereturncCs|t|tvS)z5Check if model_name belongs to the given model group.)rgetset)rrr g/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddlex/inference/models/doc_vlm/utils.py is_in_groupsrT model_predssrc_dataadd_input_pathc Cst|ts|g}t|ts|g}g}|D]}t|}|r+|dd}t|tr+||d<||qt|t|krHtdt|dt|ddd|d D}g|d <t ||D]\}}|D] } || || qd|d |q\|S) aZFormat model predictions and source data into a result dict. Args: model_preds: Model predictions (single or list). src_data: Source data list. add_input_path: If True, add input_path from image when image is a path string. Returns: Dict with keys from src_data plus "result" containing predictions. imageNZ input_pathzModel predicts z results while src data has z samples.cSsi|]}|gqSr r ).0kr r r ;sz.format_doc_vlm_result_dict..rresult) isinstancelistdictr strappendlen ValueErrorkeyszip) rrrZ input_infodatarZrst_format_dictZ data_sampleZ model_predrr r r format_doc_vlm_result_dicts.      r!N)T) __doc__typingrrr constantsrrboolrrr!r r r r s