o 0j9%@sddlmZddlmZmZmZmZddlmZddl m Z ddl m Z m Z dd lmZdd lmZmZdd lmZdd lmZmZmZdd lmZddlmZddlmZddlm Z ddl!m"Z"ddl#m$Z$ddl%m&Z&ddl'm(Z(ddl)m*Z*ddl+m,Z,ddl-m.Z.ddl/m0Z0ddl1m2Z2ddl3m4Z4ddl5m6Z6ddl7m8Z8ddl9m:Z:m;Z;ddlm?Z?dd l@mAZAdd!lBmCZCdd"lDmEZEdd#lFmGZGdd$lHmIZImJZJdd%lKmLZLdd&lMmNZNdd'lOmPZPdd(lQmRZRdd)lSmTZTdd*lUmVZVd+eWd,eWfd-d.ZXd/eWd,eeWeffd0d1ZY 2dDd2d2d2d2d2d2d2d3d/eeWd4eeeWefd5eeWd6eeWd7eeeWefd8ee d9eeZd:eeeeWefe fd;ed,efdd?Z\d4ed,efd@dAZ]d4ed,efdBdCZ^d2S)E)Path)AnyDictOptionalUnion)logging) parse_config) HPIConfigPaddlePredictorOption)AnomalyDetectionPipeline)PedestrianAttributeRecPipelineVehicleAttributeRecPipeline) BasePipeline)BaseChatBaseGeneratePrompt BaseRetriever)DocPreprocessorPipeline)DocUnderstandingPipeline)FaceRecPipeline)FormulaRecognitionPipeline)ImageClassificationPipeline)%ImageMultiLabelClassificationPipeline)InstanceSegmentationPipeline)KeypointDetectionPipeline)LayoutParsingPipeline)BEVDet3DPipeline)%MultilingualSpeechRecognitionPipeline)ObjectDetectionPipeline) OCRPipeline)OpenVocabularyDetectionPipeline)"OpenVocabularySegmentationPipeline)PaddleOCRVLPipeline)PP_ChatOCRv3_PipelinePP_ChatOCRv4_Pipeline)PP_DocTranslation_Pipeline)ShiTuV2Pipeline)RotatedObjectDetectionPipeline)SealRecognitionPipeline)SemanticSegmentationPipeline)SmallObjectDetectionPipeline)TableRecognitionPipelineTableRecognitionPipelineV2)TextToSpeechPipeline)TSAnomalyDetPipeline) TSClsPipeline) TSFcPipeline)VideoClassificationPipeline)VideoDetectionPipeline pipeline_namereturncCs4ttjjjd|d}t|sdS|S)a Get the full path of the pipeline configuration file based on the provided pipeline name. Args: pipeline_name (str): The name of the pipeline. Returns: str: The full path to the pipeline configuration file or None if not found. zconfigs/pipelines.yamlN)r__file__parentresolveexists)r5 pipeline_pathr=e/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddlex/inference/pipelines/__init__.pyget_pipeline_path<s  r?pipelinecCsF|ds|dst|}|durtd|dn|}t|}|S)a Load the pipeline configuration. Args: pipeline (str): The name of the pipeline or the path to the config file. Returns: Dict[str, Any]: The parsed pipeline configuration. Raises: Exception: If the config file of pipeline does not exist. z.ymlr7NzThe pipeline (zC) does not exist! Please use a pipeline name or a config file path!)endswithr? Exceptionr )r@r<configr=r=r>load_pipeline_configPs  rDNrCdeviceengine engine_config pp_optionuse_hpip hpi_configrCrFrGrHrIrJrKkwargsc Ks|dur |dur td|durt|}n|dur+|d|kr+td|d||d|}|d} |dur>|dd}n|dd|durO|dd}n|dd|dur`|dd}n||d<|duro|d d}n||d <t| d |||||||d |}|S) a,Create a pipeline instance from a pipeline name or config. Args: pipeline (Optional[str], optional): Pipeline name (e.g. "OCR", "object_detection") or path to a YAML config file. Required if `config` is None. config (Optional[Dict[str, Any]], optional): Pipeline configuration dict. If provided, `pipeline` name is optional and used only for validation. If both are provided and pipeline names differ, `config` takes precedence. device (Optional[str], optional): Device to run on (e.g. "gpu", "cpu", "npu:0"). engine (Optional[str], optional): Inference engine for models in the pipeline. engine_config (Optional[Dict[str, Any]], optional): Engine-specific config. pp_option (Optional[PaddlePredictorOption], optional): Paddle predictor options. use_hpip (Optional[bool], optional): Whether to use HPIP. hpi_config (Optional[Union[Dict[str, Any], HPIConfig]], optional): HPIP configuration. Returns: BasePipeline: The created pipeline instance. Nz=Both `pipeline` and `config` cannot be None at the same time.r5ziThe pipeline name in the config (%r) is different from the specified pipeline name (%r). %r will be used.rJFrKrGrHrEr=) ValueErrorrDrwarningcopypopgetr) r@rCrFrGrHrIrJrKrLr5r=r=r>create_pipelineisN     rRcO.d|vr t|d|d}t||}|S)aCreates an instance of a chat bot based on the provided configuration. Args: config (Dict): Configuration settings, expected to be a dictionary with at least a 'model_name' key. *args: Additional positional arguments. Not used in this function but allowed for future compatibility. **kwargs: Additional keyword arguments. Not used in this function but allowed for future compatibility. Returns: BaseChat: An instance of the chat bot class corresponding to the 'model_name' in the config. Zchat_bot_config_errorapi_type)rMrrQ)rCargsrLrTZchat_botr=r=r>create_chat_bots  rVcOrS)a Creates a retriever instance based on the provided configuration. Args: config (Dict): Configuration settings, expected to be a dictionary with at least a 'model_name' key. *args: Additional positional arguments. Not used in this function but allowed for future compatibility. **kwargs: Additional keyword arguments. Not used in this function but allowed for future compatibility. Returns: BaseRetriever: An instance of a retriever class corresponding to the 'model_name' in the config. Zretriever_config_errorrT)rMrrQ)rCrUrLrTZ retrieverr=r=r>create_retriever  rWcOrS)a Creates a prompt engineering instance based on the provided configuration. Args: config (Dict): Configuration settings, expected to be a dictionary with at least a 'task_type' key. *args: Variable length argument list for additional positional arguments. **kwargs: Arbitrary keyword arguments. Returns: BaseGeneratePrompt: An instance of a prompt engineering class corresponding to the 'task_type' in the config. Zpe_config_error task_type)rMrrQ)rCrUrLrYZper=r=r>create_prompt_engineeringrXrZ)N)_pathlibrtypingrrrrutilsrZ utils.configr modelsr r Zanomaly_detectionrZattribute_recognitionrrbaser componentsrrrZdoc_preprocessorrZdoc_understandingrZface_recognitionrZformula_recognitionrZimage_classificationrZimage_multilabel_classificationrZinstance_segmentationrZkeypoint_detectionrZlayout_parsingrZm_3d_bev_detectionrZmultilingual_speech_recognitionrZobject_detectionr Zocrr!Zopen_vocabulary_detectionr"Zopen_vocabulary_segmentationr#Z paddleocr_vlr$Z pp_chatocrr%r&Zpp_doctranslationr'Z pp_shitu_v2r(Zrotated_object_detectionr)Zseal_recognitionr*Zsemantic_segmentationr+Zsmall_object_detectionr,Ztable_recognitionr-r.Ztext_to_speechr/Zts_anomaly_detectionr0Zts_classificationr1Zts_forecastingr2Zvideo_classificationr3Zvideo_detectionr4strr?rDboolrRrVrWrZr=r=r=r>s                                    M