{-j9%DddlmZddlmZmZmZmZddlmZddl 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-ZXd.eWd,eeWeffd/ZY d>d0d0d0d0d0d0d0d1d.eeWd2eeeWefd3eeWd4eeWd5eeeWefd6ee d7eeZd8eeeeWefe fd9ed,efd:Z[d2ed,efd;Z\d2ed,efd<Z]d2ed,efd=Z^d0S)?)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_namereturncttjjjdz |dz }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)r6 pipeline_paths d/var/www/html/banglarbhumi/venv/lib/python3.11/site-packages/paddlex/inference/pipelines/__init__.pyget_pipeline_pathr@<sh X$+    ! ! ! "gii     % % ' 't pipelinec|ds:|ds%t|}|td|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.ymlr9NzThe pipeline (zC) does not exist! Please use a pipeline name or a config file path!)endswithr@ Exceptionr )rBr>configs r?load_pipeline_configrGPs   f % %!):):7)C)C!)(33  nnnn  ! ! - ( (F MrANrFdeviceengine engine_config pp_optionuse_hpip hpi_configrFrIrJrKrLrMrNkwargsc b||td|t|}nE|/|d|kr#tjd|d||d|}|d} ||dd}n|dd||dd}n|dd||dd}n||d<||d d}n||d <tj| 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.r6ziThe pipeline name in the config (%r) is different from the specified pipeline name (%r). %r will be used.rMFrNrJrKrH) ValueErrorrGr warningcopypopgetr) rBrFrIrJrKrLrMrNrOr6s r?create_pipelinerWis>FN K   ~%h//  F?$;x$G$G O{''     ?+M::j%00 :t$$$ZZ d33  <&&& ~Hd++!x ?D99 "/.| .. #     H OrAcd|vrt|d|d}tj||}|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. chat_bot_config_errorapi_type)rRrrV)rFargsrOrZchat_bots r?create_chat_botr]sK&(( 78999j!H%x|H%%f--H OrAcd|vrt|d|d}tj||}|S)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. retriever_config_errorrZ)rRrrV)rFr[rOrZ retrievers r?create_retrieverrasM  6)) 89:::j!H+ !(++F33I rAcd|vrt|d|d}tj||}|S)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. pe_config_error task_type)rRrrV)rFr[rOrdpes r?create_prompt_engineeringrfsM F"" 12333{#I *   * *6 2 2B IrA)N)_pathlibrtypingrrrrutilsr utils.configr modelsr r anomaly_detectionrattribute_recognitionrrbaser componentsrrrdoc_preprocessorrdoc_understandingrface_recognitionrformula_recognitionrimage_classificationrimage_multilabel_classificationrinstance_segmentationrkeypoint_detectionrlayout_parsingrm_3d_bev_detectionrmultilingual_speech_recognitionr object_detectionr!ocrr"open_vocabulary_detectionr#open_vocabulary_segmentationr$ paddleocr_vlr% pp_chatocrr&r'pp_doctranslationr( pp_shitu_v2r)rotated_object_detectionr*seal_recognitionr+semantic_segmentationr,small_object_detectionr-table_recognitionr.r/text_to_speechr0ts_anomaly_detectionr1ts_classificationr2ts_forecastingr3video_classificationr4video_detectionr5strr@rGboolrWr]rarfrQrAr?rs------------((((((55555555777777CCCCCCCCCC555555777777------;;;;;;======RRRRRR??????999999111111000000RRRRRR555555FFFFFFLLLLLL------DDDDDDDD999999((((((DDDDDD555555??????@@@@@@SSSSSSSS000000666666,,,,,,((((((======333333SS(34S>4#J(,  .215#=AJJJsmJ T#s(^ $J SM J SM J DcN+ J-.JtnJtCH~y89:JJJJJJZDh&  .  rA