{-j vdZddlmZmZmZddlmZdededefdZ dd ed ee d edeeeffd Z dS)z'Shared utilities for DocVLM predictors.)AnyDictList) MODEL_GROUP model_name group_namereturncH|tj|tvS)z5Check if model_name belongs to the given model group.)rgetset)rr s f/var/www/html/banglarbhumi/venv/lib/python3.11/site-packages/paddlex/inference/models/doc_vlm/utils.py is_in_grouprs SUU;; ;;T model_predssrc_dataadd_input_pathct|ts|g}t|ts|g}g}|D]X}t|}|r0|dd}t|tr||d<||Yt |t |kr0tdt |dt |dd|dD}g|d <t||D]X\}}|D]#} || || $|d |Y|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. imageN input_pathzModel predicts z results while src data has z samples.ci|]}|gSr).0ks r z.format_doc_vlm_result_dict..;s;;;q";;;rrresult) isinstancelistdictr strappendlen ValueErrorkeyszip) rrr input_infodatarrst_format_dict data_sample model_predrs rformat_doc_vlm_result_dictr+s k4 ( ($"m h % %:J  Dzz  +HHWd++E%%% +%*\"$ ;3z??** fc+.. f fCPZOO f f f   <;jm&8&8&:&:;;;O "OH#&z;#?#?55 Z!!## 6 6A A  % %k!n 5 5 5 5!((4444 rN)T) __doc__typingrrr constantsrr boolrrr+rrrr0s.-""""""""""""""""