dj)& ndZddlZddlZddlmZmZddlmZddlm Z ddl m Z ddl mZddlmZdd lmZdd lmZdd lmZdd lmZej2Zej6Ze j8Ze j:Zej<Gd dZGddej@Z!Gddej@Z"de"defdZ#Gddej@Z$ej<GddZ%ejLdejNe$ejNejPfejLdejPejRejNe"fejLdejPfejTdejNe"gdfZ+GddZ,y) z!MediaPipe language detector task.N)ListOptional)category)classification_result)classifier_options)classifier_options_c base_options)base_options_c)mediapipe_c_bindings)mediapipe_c_utils)serial_dispatchercPeZdZUejGddZeeed<y)LanguageDetectorResultc&eZdZUdZeed<eed<y) LanguageDetectorResult.Detection$A language code and its probability. language_code probabilityN)__name__ __module__ __qualname____doc__str__annotations__floatp/root/repos/pocs/video-evals/.venv/lib/python3.12/site-packages/mediapipe/tasks/python/text/language_detector.py Detectionr&s.rr detectionsN)rrr dataclasses dataclassr rrrrrrr#s-9orrcHeZdZdZdej fdej fgZy)LanguageDetectorPredictionCrrrN)rrrrctypesc_char_pc_float_fields_rrrr%r%2s&,(fnn%(rr%cTeZdZdZdej efdejfgZy)LanguageDetectorResultCzLanguage detector output. predictionspredictions_countN) rrrrr&POINTERr%c_uint32r)rrrr+r+;s.!nfnn%@ABFOO,(rr+c_resultreturnctg}t|jD]f}|j|}tj |j j d|j}|jj|h|S)zGConverts a C LanguageDetectorResult to a Python LanguageDetectorResult.)r!utf-8)rr) rranger-r,r rdecoderr!append)r0 py_resulti c_prediction py_predictions r+_convert_to_python_language_detector_resultr;Ds%3) ++ ,/a''*L*44"0077@ ,,5M . / rcDeZdZdejfdej fgZy)LanguageDetectorOptionsCr rN)rrrbase_options_c_module BaseOptionsCclassifier_options_c_moduleClassifierOptionsCr)rrrr=r=Ss',99:8KKL(rr=ceZdZUdZeed<dZeeed<dZ ee ed<dZ ee ed<dZ eeeed<dZeeeed<d efd Zy) LanguageDetectorOptionsaOptions for the language detector task. Attributes: base_options: Base options for the language detector task. display_names_locale: The locale to use for display names specified through the TFLite Model Metadata. max_results: The maximum number of top-scored classification results to return. score_threshold: Overrides the ones provided in the model metadata. Results below this value are rejected. category_allowlist: Allowlist of category names. If non-empty, classification results whose category name is not in this set will be filtered out. Duplicate or unknown category names are ignored. Mutually exclusive with `category_denylist`. category_denylist: Denylist of category names. If non-empty, classification results whose category name is in this set will be filtered out. Duplicate or unknown category names are ignored. Mutually exclusive with `category_allowlist`. r Ndisplay_names_locale max_resultsscore_thresholdcategory_allowlistcategory_denylistr1c (|jj}tjt j |j |j|j|j|j}t}||_||_ |S)z,Generates a ctypes LanguageDetectorOptionsC.)rDrErFrGrH) r to_ctypesr@convert_to_classifier_options_cclassifier_options_moduleClassifierOptionsrDrErFrGrHr=r)selfr r c_optionss rrJz!LanguageDetectorOptions.to_ctypesvs&&002N#CC % 7 7%)%>%> ,, $ 4 4#'#:#:"&"8"8   )*I+I#7I r)rrrr _BaseOptionsrrDrrrEintrFrrGrrHr=rJrrrrCrCZso&(, ,#+x}#%)/8E?),0htCy)0+/Xd3i(/1rrCMpLanguageDetectorCreateMpLanguageDetectorDetectMpLanguageDetectorCloseMpLanguageDetectorCloseResultceZdZUdZej ed<ejed<dej dejfdZ e de ddfd Z e d e ddfd Zd e defd ZdZdZdZdZy)LanguageDetectoraClass that predicts the language of an input text. This API expects a TFLite model with TFLite Model Metadata that contains the mandatory (described below) input tensors, output tensor, and the language codes in an AssociatedFile. Input tensors: (kTfLiteString) - 1 input tensor that is scalar or has shape [1] containing the input string. Output tensor: (kTfLiteFloat32) - 1 output tensor of shape`[1 x N]` where `N` is the number of languages. _lib_handlelibhandlec ||_||_yN)rX_detector_handle)rNrZr[s r__init__zLanguageDetector.__init__sDI"Dr model_pathr1cL|jtt|S)aCreates an `LanguageDetector` object from a TensorFlow Lite model and the default `LanguageDetectorOptions`. Args: model_path: Path to the model. Returns: `LanguageDetector` object that's created from the model file and the default `LanguageDetectorOptions`. Raises: ValueError: If failed to create `LanguageDetector` object from the provided file such as invalid file path. RuntimeError: If other types of error occurred. )model_asset_pathr )create_from_optionsrCrP)clsr`s rcreate_from_model_pathz'LanguageDetector.create_from_model_paths'"  " "%zB  roptionsctjt}|j}t j }|j t j|t j|t||S)aCreates the `LanguageDetector` object from language detector options. Args: options: Options for the language detector task. Returns: `LanguageDetector` object that's created from `options`. Raises: ValueError: If failed to create `LanguageDetector` object from `LanguageDetectorOptions` such as missing the model. RuntimeError: If other types of error occurred. )rZr[) mediapipe_c_bindings_moduleload_shared_library_CTYPES_SIGNATURESrJr&c_void_prRbyrefrW)rdrfrZctypes_optionsdetector_handles rrcz$LanguageDetector.create_from_optionssd" & 9 9:L MC&&(Noo'O   ^$ _% O <r rhr r BaseOptionsrPCategoryClassificationsrMr#r Structurer%r+r;r=rCCStatusFunctionr.rkr' CFunctionrjrWrrrrs( !TnhlKO[99"..  # #.>>-??    &"2"2f.. %  v// ---b&%%" FNN3 4 FNN6?? + &%%" OO OO FNN2 3 &%%! OO   ' / 01 -