]jdZddlZddlZddlZddlZddlZddlZddlmZddl m Z m Z m Z m Z mZmZmZddlmZddlmZmZddlmZddlZddlmZdd lmZmZmZd d lm Z d d l!m"Z"m#Z#m$Z$m%Z%m&Z&d d l'm(Z(d dl)m*Z*d dl+m,Z,m-Z-m.Z.m/Z/d dl0m1Z2ejfe4Z5ejljoejljqe9Z:ejljwe:ddddZ<e=e<5Z>ej~e>Z@dddhdZAdZBdhZCdZDdZEdZFdZGdZHdZIdeJdeKd e eKfd!ZLGd"d#ZMy#1swY;xYw)$ N)cached_property)DictOptionalListTupleAnyCallableUnion) BaseModel) defaultdict OrderedDict)etree)JSF)LSConfigParseException&LabelStudioXMLSyntaxErrorSentryIgnored'LabelStudioValidationErrorSentryIgnored)LabelStudioTag)ChatMessageTag ReactCodeTag ControlTag ChoicesTag LabelsTag) ObjectTag)LabelTag)AnnotationValue TaskValuePredictionValueRegioncreatez.._legacyschemazlabel_config_schema.json>LabelChoiceRelationz label-studiovideorectangleresultci}t|jt|jzD]M}i}||vr|j||||vr(||D] }|j|d|||z||<"|||<O|S)a& Merge two dictionaries with nested dictionary values into a single dictionary. Args: dict1 (dict): The first dictionary to merge. dict2 (dict): The second dictionary to merge. Returns: dict: A new dictionary with the merged nested dictionaries. Example: dict1 = {'sentiment': {'Negative': 1, 'Positive': 1}} dict2 = {'sentiment': {'Positive': 2, 'Neutral': 1}} result_dict = merge_nested_dicts(dict1, dict2) # {'sentiment': {'Negative': 1, 'Positive': 3, 'Neutral': 1}} r)setkeysupdateget)dict1dict2 result_dictkeyvaluesubkeys T/root/env/lib/python3.12/site-packages/label_studio_sdk/label_interface/interface.pymerge_labels_countersr6<s"K5::< 3uzz|#44 ! %< LLs $ %<* J % &! 4uSz&7I If  J! C ! cd|vrd|dvr&t|ddts|ddg|dd<nst|dtr`td|dDrKtt |dD]1}t|d|dtr|d|dg|d|d<3d|vrEt|dt rt |d|d<|S|dDcgc] }t |c}|d<|Scc}w)zb workaround for single choice https://github.com/heartexlabs/label-studio/issues/1259 Choicesr&c3$K|]}d|v yw)r&N).0 tag_choicess r5 z_fix_choices..js9 (3H #9 sView) isinstancelistallrangelenr _fix_choices)confignviews r5rErE^s2 F vi( ( 9 h '2 ,2)+DF^LTl40LF6N MMsC0ct|trdj|}dj|||jgS)N,|)r@rAjoinlower) from_nameto_nametypes r5get_annotation_tuplerQys6'4 ((7# 88Y6 77r7c t|}g}|jD]'\}}|jt||d|d)|S)NrOrP) parse_configitemsappendrQ label_configoutputsout control_nameinfos r5get_all_control_tag_tuplesr\sP<(G C%mmoV d ' d9otF|TUV Jr7ct|}g}|jD]'\}}|j|dj)|S)z) Get all types from label_config rP)rSrTrUrMrVs r5 get_all_typesr^sK<(G C%mmo) d 4<%%'() Jr7countrPreturnc,|sy|d||dkDrdSdS)zgHelper for displaying pluralized sources of validation errors, eg "1 draft" or "3 annotations" N rsr;)r_rPs r5 display_countres1 WAdV519C5 66"5 66r7ceZdZdZedMdZedZdNdefdZde ee ee e ee e ffde e fd Z ed Zed Zed Zed Zde de de de fdZdZdZdNdZdNdZdNdZde fdZ dOdeedeede fdZedZedZedZ dZ!dZ"d ede#e e e e$jJffd!Z&ed"Z'd#Z(d$Z)d%Z*d&Z+ed'Z,d(Z-ed)Z.dPdQd*Z/dPde e0e effd+Z1de efd,Z2d-Z3dPd.Z4dRde#e0e effd/Z5d0Z6dSd1Z7dTde#e0e effd2Z8dUd3Z9dPd4Z:dVd5Z;d6Zdee=fd8Z?dPd9e0dee=fd:Z@d;ZA dVd<ZBed=ZCed>ZDd?ZE dOd@e ee#fdAe ee#fdBeedCeede#eeeff dDZFdEZGdFZHdGZIdHZJdNdIZKdPdJZLdKZMdLZNy)WLabelInterfaceaYThe LabelInterface class serves as an interface to parse and validate labeling configurations, annotations, and predictions within the Label Studio ecosystem. It is designed to be compatible at the data structure level with an existing parser widely used within the Label Studio ecosystem. This ensures that it works seamlessly with most of the existing functions, either by directly supporting them or by offering re-implemented versions through the new interface. Moreover, the parser adds value by offering functionality to validate predictions and annotations against the specified labeling configuration. Below is a simple example of how to use the new API: ```python from label_studio_sdk.label_interface import LabelInterface config = " " li = LabelInterface(config) region = li.get_tag("chc").label("one") # returns a JSON representing a Label Studio region region.as_json() # returns True li.validate_prediction({ "model_version": "0.0.1", "score": 0.90, "result": [{ "from_name": "chc", "to_name": "txt", "type": "choices", "value": { "choices": ["one"] } }] }) ``` NcPtj||}t|tr dd|iiff|z}t|tst|trg} |D]} | \} } } | j dd}t|tr+||vr'dd|j |iif}| j |t|tr-||vr)dd|j |i| ff}| j || j | | }tj|}tj||S) z Simple way of create UI config, it helps you not to thing much about the name/toName mapping LabelInterface.create_simple({ "txt": "Text", "chc": choices("positive", "negative") }) )mappingHeaderr3nameNr?style)pretty) CEconvert_tags_descriptionr@strdictr.rUtree_from_tuplestree_to_string)clstagsrititlerlrmargskwargstuples new_tuplesttag attributeschildrenrk title_tag parent_tagtrees r5r"zLabelInterface.creates*,,T7C eS !7E!3R8:VCF eT "j&=J ),-)Z!~~fd3eT*tu}!)Weiio+G LI%%i0eT*tu}"(GUYYt_*F!MJ%%j1%%a( )" F""F+  f55r7c:|j|i|}|dd|i|S)aCreate instance is a shortcut to create a config and then parse it right away returning the LabelInterface object ``` li = LabelInterface.create_instance({ "txt": "Text", "lbl": labels(("person", "org")) }) lbl = li.get_control("lbl") reg = lbl.label("person", start=0, end=10) ``` The above returns a region that could be serialized to Label Studio JSON format and uploaded to Label Studio rFr;r!)rtrwrxrFs r5create_instancezLabelInterface.create_instances,T,V,+&+F++r7rFcd|_||_||_tj |\}}}||_||_||_|j|\}} } } |j|| | }t|j|_ t| j|_ ||_| |_| |_| |_y)a> Initialize a LabelInterface instance using a config string. Args: config (str): Configuration string. tags_mapping: Provide your own implementation of any tag, this is helpful in cases where you want to override one of the control tags, and have your custom .label() method implemented. The configuration string should be formatted as follows: ``` ``` This method will extract the predefined task from the configuration and parse the controls, objects, and labels used in it. FN) task_loaded_config _tags_mappingrgget_task_from_labeling_config_sample_config_task_sample_config_ann_sample_config_predparse_link_controlsr+r, _control_tags _object_tags _controls_objects_labels_tree) selfrF tags_mappingrwrx _task_data_ann_predcontrolsobjectslabelsrs r5__init__zLabelInterface.__init__s.! )#1"N"Nv"V D%#- "&#( *.**V*<''64&&x&A!1 /"   r7datar`c g}|jD]\}}|tjd|d"||jvrtj d|dJ|j|}t |t ttfrd|i}n7t |tr't|dkDrt |dt rd|i}n t |tr|g}|D]#}|j|jdi|%|S)a Takes raw data representation and maps keys to control tag names. If name is not found, it will be skipped Args: data (Dict): Raw data representation. Example: {"choices_name": "Positive", "labels_name": [{"start": 0, "end": 10, "label": "person"}]} raise_if_control_not_found (bool): Raise an exception if control tag is not found. zPayload for control tag 'z' is None: it can be ok if this payload is autogenerated by LLMs - try to adjust the prompt to generate valid payloads. Otherwise, it can signify an errorz Control tag 'z' not found in the configlabelrr;)rTloggerwarningrr[r@rpintfloatrArDrrUr)rrregionscontrol_tag_namepayloadcontrolitems r5create_regionszLabelInterface.create_regions5s)- 6 % g/0@/AB99: t~~5 m,<+==VWXnn%56G 'Ce#45"G,GT*w> !>>$' ' 4== ==& &.j9::r7c:|j||jS)aPRetrieves the object with the given name from `_objects`. This utilizes the `_get_tag` method to obtain the named object. Args: name (str, optional): The name of the object to be retrieved from `_objects`. Returns: object: The corresponding object if it exists in `_objects`. )rrrs r5 get_objectzLabelInterface.get_objects}}T4==11r7c$|j|S)z/Provides an alias for the `get_control` method.) get_controlrs r5 get_outputzLabelInterface.get_outputs%%r7c:|j||jS)a>Retrieves the control tag that the control tag maps to. This uses the `_get_tag` method to obtain the named control. Args: name (str, optional): The name of the control to be retrieved. Returns: object: The corresponding control if it exists in `_controls`. )rrrs r5rzLabelInterface.get_controls}}T4>>22r7c t|jt|jz}t|ts|gn|}|Dcgc]}|D]}t||s|c}}Scc}}w)aFinds tags by their class type. The function looks into both `self.objects` and `self.controls` to find tags that are instances of the provided class(es) Args: tag_class (class or list of classes): The class type(s) of the tags to be found. Returns: list: A list of tags that are instances of the provided `tag_class`(es). )rArrr@)r tag_classlst tag_classesr|rts r5find_tags_by_classz!LabelInterface.find_tags_by_classs[4<< 4 #66)3It)Dyk) "RkRsZS=QRRRRs A'A'tag_typematch_fnc|j|j|j|jd}|j|t|jt|jz}|tt ||}|S)aaFinds tags that match the given function in the entire parsed tree. This function searches through both `objects` and `controls` based on `tag_type`, and applies the `match_fn` (if provided) to filter matching tags. Args: tag_type (str, optional): The type of tags to be searched. Categories include 'objects', 'controls', 'inputs' (alias for 'objects'), 'outputs' (alias for 'controls'). If not specified, searches both 'objects' and 'controls'. match_fn (Callable, optional): A function that takes a tag as an input and returns a boolean value indicating whether the tag matches the required condition. Returns: list: A list of tags that match the given type and satisfy `match_fn`. )rrrrX)rrr.rAfilter)rrr tag_typesrs r5 find_tagszLabelInterface.find_tagssl0|| ll}}  mmHd4<<&84 ;N&NO  vh,-C r7c(|jddS)Nrc:|jjdvS)N)rhypertextlabelsr|rMr{s r5z)LabelInterface.ner_tags..sAEEKKMEb4br7rrs r5ner_tagszLabelInterface.ner_tags s~~j*bccr7c(|jddS)Nrc<|jjdk(S)Nimagerrs r5rz+LabelInterface.image_tags..s155;;=G3Kr7rrs r5 image_tagszLabelInterface.image_tagss~~i)KLLr7c(|jddS)Nrc<|jjdk(S)Npdfrrs r5rz)LabelInterface.pdf_tags..s155;;=E3Ir7rrs r5pdf_tagszLabelInterface.pdf_tagss~~i)IJJr7ctj|}d|_|jD]>}|js|j |vs|j |j |_@|S)aLoads a task and substitutes the value in each object tag with actual data from the task, returning a copy of the LabelConfig object. If the `value` field in an object tag is designed to take variable input (i.e., `value_is_variable` is True), the function replaces this value with the corresponding value from the task dictionary. Args: task (dict): Dictionary representing the task, where each key-value pair denotes an attribute-value of the task. Returns: LabelInterface: A deep copy of the current LabelIntreface instance with the object tags' value fields populated with data from the task. T)copydeepcopyrrvalue_is_variable value_namer.r3)rtaskrobjs r5 load_taskzLabelInterface.load_tasks\*}}T"<< 5C$$4)? HHS^^4  5 r7c |jjDcgc]\}}|jr|}}}d|jjDcic]\}}||jc}}|dScc}}wcc}}w)z Converts the current LabelInterface instance into a JSON Schema. Returns: dict: A dictionary representing the JSON Schema. object)rP propertiesrequired)rrTis_output_requiredto_json_schema)rrkrrequired_outputss r5rzLabelInterface.to_json_schema8s'+nn&:&:&< "T7))   DHNNDXDXDZ3@4g,,..)    s A9A? config_stringc 2 tj|}iit t }}}g}ttd}|jD]} | jr,d| jvr|j| jdtj| r2tj| |j|| jd<t!j| r2t!j| |j|| jd<| jj#dr| j$j'|vrt!| j$t | j| jj#d| jj#d| jj#d|| jd<t)j| st)j| |} | s| || j*| j,<i||} | j/D]\} } t1| d d j'}||vr&| |vr||}|| | g| | d  || <| |vsFt!| j$t | j2| | j2j#d| j2j#d|| <||||fS#tj$r}tt |d}~wwxYw) aParses the received configuration string into dictionaries of ControlTags, ObjectTags, and Labels, along with an XML tree of the configuration. Args: config_string (str): the configuration string to be parsed. Returns: Tuple of: - Dictionary where keys are control tag names and values are ControlTag instances. - Dictionary where keys are object tag names and values are ObjectTag instances. - Dictionary of dictionaries where primary keys are label parent names and secondary keys are label values and values are LabelTag instances. - An XML tree of the configuration. N)chat reactcode indexFlag)rrk valueListr3)r|attrrkr3r|rd)rktoName)rkrOr)r fromstringXMLSyntaxErrorrrpr rqrriterattribrUr validate_node parse_noderrr.r|rMr parent_namer3rTgetattrr)rrxml_treeerrr variables_SELF_REFERENCING_TAGSr|lb all_parsedrk parsed_tag tag_lower control_classs r5rzLabelInterface.parseKs  A'' 6H%'K,=6 #%"  ==? :CzzkSZZ7  K!89'',/9/D/DSW[WiWi/jF+,((-.7.B.B3UYUgUg.h 6*+'CGGMMO?U,U/8cjj)/**..cjjnnW6MN / 6*+'',((h779F2>>*28845 :>-,8, * 0 0 2  D* E26<<>I 668# 6y A !.!F"&$7" 7" )"joo.$//--k:??;N;Nw;WX !  &&(22K## A8Q@ @ AsK**L=LLc tj|}|tjdtj j |}t|}|S#t$rtjdwxYw)rbzcan only parse stringszxml is empty or incorrect)rr TypeError ParseErrorxmljson badgerfishrrE)rtrxmlrFs r5parse_config_to_jsonz#LabelInterface.parse_config_to_jsonsx =""=1C ;""#>? ?##((-f%  =""#;< < =s AA9c * tj|}tj|ty#t j tf$r}tt|d}~wtjj$r}t|jr|jdjn |j}djdj!t#t|j$|j'dd}t|d}~wwxYw)rbNzValidation failed on {}: {}/@rd)rgr  jsonschemavalidate_LABEL_CONFIG_SCHEMA_DATArr  ValueErrorrrp exceptionsValidationErrorrDcontextmessageformatrLmappathreplace)rrrFexc error_messages r5_schema_validationz!LabelInterface._schema_validations I#88GF   (A B  *- D9#c(C C$$44 I7:3;;7GCKKO33S[[M9@@S#((+,m.C.CC.LM:-H H  Is"/2D A D?BD  Dc tjd|}t|}tjd|}|D]6}|jdD] }||vst d|dt |8y)rbzname="([^"]*)"ztoName="([^"]*)"rJztoName="z" not found in names: N)refindallr+splitrsorted)rr all_namesnamestoNamestoName_rs r5_to_name_validationz"LabelInterface._to_name_validationsJJ0-@ I**0-@ G!--, &A"6(*@P  r7ctjd|}tt|t|k7rt ddj |zy)rbz(?:^|[^\w])name="([^"]*)"z(Label config contains non-unique names: , N)r"r#rDr+rrL)rrr&s r5_unique_names_validationz'LabelInterface._unique_names_validationsMJJ;]K s9~ #i. 09:TYYy=QQ  1r7cg}|jrK|jjD].}t|ds|j|j 0|rt dj |y)zCValidate tag-specific attribute values (e.g. Video playback speed).validate_configrN)rrhasattrextendr/rrL)rerrorsrs r5_tag_attribute_validationz(LabelInterface._tag_attribute_validationsh ==}}++- 93 12MM#"5"5"78 9 9$))F:KL L r7cD |jy#t$rYywxYw)rbTF)rrrs r5is_validzLabelInterface.is_valids%  MMO6  s  c|j}|j||j||j||j y)aValidates the provided configuration string against various validation criteria. This method applies a series of validation checks to `_config`, including schema validation, checking for uniqueness of names used in the configuration, and the "to_name" validation. It throws exceptions if any of these validations fail. Raises: Exception: If any validation fails, specific to the type of validation. N)rr r-r*r3)rrs r5rzLabelInterface.validatesC   . %%m4   / &&(r7ctrNotImplemented)rtrFs r5validate_with_dataz!LabelInterface.validate_with_datas r7c|jD]0}|js|dj|jd0yd|vr|j syd|vr|j syy)rbrNF annotations predictionsT)rrr.rvalidate_annotationvalidate_prediction)rrvalidate_regions_onlyrs r5 validate_taskzLabelInterface.validate_tasksn<< C$$f)9)9#..$)O)W  D )A)A)C D )A)A)Cr7cH|j|}|r|St|dk(S)a Validates an object (annotation/prediction) and returns boolean or error messages. Args: obj (dict): The object to validate return_errors (bool): If True, returns list of error messages. If False, returns boolean. Returns: Union[bool, List[str]]: If return_errors=False, returns True/False. If return_errors=True, returns list of error messages. r)_validate_object_logicrD)rr return_errorsr2s r5_validate_objectzLabelInterface._validate_objects+,,S1 Mv;!# #r7c g} t|vr|jdtd|S|jt}t|ts|jdtd|Sd|vr4|d}|- t |}d|cxkrdksn|jd|g}g}t|D]\}} t| ts!|jd t| d | 7| jd d k7r>|j| d||| d} |j| | rw|j| |j|| f|D].\}} |j| |d|} | s|j| 0 |S#t tf$r|jd |Y wxYw#t$r(} |jdt| Yd} ~ |Sd} ~ wwxYw)z2Core validation logic that returns error messages.zMissing required field ''z' must be a listscoreNgg?z3Score must be between 0.00 and 1.00 inclusive, got z"Score must be a valid number, got z!Region must be a dictionary. Got z: rPrelationT)r)region)rD region_indexr)rDrelation_indexz$Unexpected error during validation: ) RESULT_KEYrUr.r@rArrr enumeraterqrPvalidate_regionr1validate_relationrrp)rrr2r)rH score_floatrrelations_to_validateirJ region_errorsrIrelation_errorrs r5rCz%LabelInterface._validate_object_logic*s2 K$ 8 AFG WWZ(Ffd+ *-=>? #~G $T&+El #{ 9c 9"MM,_`k_l*mnG$& !'v. > 6!&$/MM$Ed6l^SUV\U]"^_::f%3$($8$8d+1V D%9%MMM-0(v.*00!V= >$ 5 2 8!%!7!7'Y]no!7!p!MM.1 2 C' 2T (J5'&RST< K MM@QI J J  KsT"F,?F,' F,3,FBF,%AF,,F,#F)%F,(F))F,, G5GGc$|j|S)aUValidates the given annotation against the current configuration. This method applies the `validate_region` method to each region in the annotation and returns False if any of these validations fail. If all the regions pass the validation, it returns True. Args: annotation (dict): The annotation to be validated, where each key-value pair denotes an attribute-value of the annotation. Returns: bool: True if all regions in the annotation pass the validation, False otherwise. rE)r annotations r5r>z"LabelInterface.validate_annotationes$$$Z00r7c&|j||S)a Validates the given prediction against the current configuration. Args: prediction (dict): The prediction to be validated return_errors (bool): If True, returns a list of error messages instead of boolean Returns: Union[bool, List[str]]: If return_errors=False, returns True/False. If return_errors=True, returns list of error messages. rW)r predictionrDs r5r?z"LabelInterface.validate_predictionys$$Z??r7c g}gd}|D]}||vs|jd|d|d!|rd|fS |j|d} |j|d } |jj } |d j } | | k7r!|jd|d | d| d|dd |d |j vr-|jd|d |d d|dd|j  |j|d|st|ddj } | dvrg} |d}dD]9}||vs t||}|dks|dkDr| j|d||;| r1|jd|d|dddj| zdznm|d}|j|}|jd|d|dd |d!|n6|d}|j|}|jd|d|dd |d!|t|dk(|fS#t$r)}|jd|d|ddd|fcYd }~Sd }~wwxYw#t$r)}|jd|d |d dd|fcYd }~Sd }~wwxYw#t$r| j|d||Y{wxYw#t$r0}|jd|d"|ddt|Yd }~d }~wwxYw)#aHelper method to perform region validation logic. Args: region (dict): The region to be validated. region_index (int): Index of the region for error reporting. context (dict, optional): Additional context forwarded to ``control.validate_value()``. Returns: tuple: (is_valid, errors) where is_valid is bool and errors is list of strings. )rNrOrPr3zRegion : Missing required field 'rGFrNz: 'from_name' 'z' not found in configurationNrOz : 'to_name' 'rPz: Type 'z ' does not match expected type 'z' for control 'z' is not valid for control 'z'. Valid options: r3rr|rd) rectanglerectanglelabels)xywidthheightrdrbz : Invalid geometry for control 'z': r,z out of bounds [0, 100]z: Invalid value for control 'z'. Got: z. Valid options: z&: Error validating value for control ')rUrrrr|rMrOvalidate_valuerrrL_get_valid_values_for_controlrprD)rrJrKrr2required_fieldsfieldrrr expected_type actual_typerout_of_bounds_fieldsr3v invalid_value valid_valuess r5_validate_region_logicz%LabelInterface._validate_region_logics3D$ ZEF"  ~5OPUwVWXY Z &=  !&&vk':;G  !//&"34C  ))+ Vn**, - ' MMGL>+FfgtfuvEFLMXFYEZZ[\ ] ) GOO 3 MMGL>vi?P>QQmntvAoBnCCUV]VeVeUfg h" z))&/7)K#GUB7==?  @@+-("7OE!>W E>W$)%,$7#$q5AG$8$?$?5'5QV<.@Y$Z W, %l^3STZ[fTgShhkl"ii(<=>78 )/w '+'I'I''R  %l^3PQWXcQdPeemn{m|}NO[N\]%+7OM#'#E#Eg#NLMM!,/LVT_M`LaaijwixyJKWJXY 6{a''u ! MMGL> AT@UUqr s&=  ! ! MMGL>vi?P>QQmn o&=  !8$-W 4 ; ;ugQuU|n1WX^_jXkWllopstupvowx y y zsyHI -AJ+51J&B"J+ I !I?I I  I?I:4I?:I?"J($J+'J((J++ K$4&KK$c t|dr|jr |jSt|drd|jjSy#t$rYywxYw)zDGet valid values for a control tag to provide better error messages.r _value_classzexpected structure: zno specific validation ruleszunknown validation rules)r0rrq__name__rrrs r5rfz,LabelInterface._get_valid_values_for_controls] .w)gnn!..)*.1-g.B.B.K.K-LMM5 .- .s$A $A AAc<|j|||\}}|r|S|S)aValidates a region from the annotation against the current configuration. The validation checks the following: - Both control and object items are present in the labeling configuration. - The type of the region matches the control tag name. - The 'to_name' in the region data connects to the same tag as in the configuration. - The actual value for example in tag is producing start, end, and labels. Args: region (dict): The region to be validated. return_errors (bool): If True, returns a list of error messages instead of boolean region_index (int): Index of the region for error reporting (used when return_errors=True) context (dict, optional): Additional context passed through to ``control.validate_value()``. May contain ``result`` (the full result list) and ``region`` (the current region dict) so that control tags can inspect sibling regions. Returns: Union[bool, List[str]]: If return_errors=False, returns True/False. If return_errors=True, returns list of error messages. r])ro)rrJrDrKrr5r2s r5rOzLabelInterface.validate_regions.. 66v|U\6]& MOr7c g} gd}|D]}||vs|jd|d|d!|jddk7r'|jd|d|jddd}|jd |vr*|jd|d |d |jd d|Dcic] }d |vs|d |} }|jd } | | vr|jd|d| d|jd} | | vr|jd|d| dt |dk(|fScc}w#t$r*} |jd|dt| Yd} ~ Cd} ~ wwxYw)apHelper method to perform relation validation logic. Args: relation (dict): The relation to validate regions (list): List of validated regions relation_index (int): Index of the relation for error reporting Returns: tuple: (is_valid, errors) where is_valid is bool and errors is list of strings. )rP directionfrom_idto_idz Relation r\rGrPrIz : Type must be 'relation', got ')leftrightbirvz: Direction must be one of z, got 'idrwz : 'from_id' 'z' not found in regionsrxz : 'to_id' 'z: Error validating relation - Nr)rUr.rrprD) rrIrrLr2rgrhvalid_directionsrregion_mappingrwrxrs r5_validate_relation_logicz'LabelInterface._validate_relation_logic s ^GO( b(MMIn-==WX]W^^_"`a b ||F#z1 .)99YZbZfZfgmZnYoopqr 7 ||K(0@@ .)99TUeTffmnvnznz|GoHnIIJKL3:GQTQYagqjGNGll9-Gn, .)9wiOefgLL)EN* .)9UGKabc 6{a''H ^ MMIn%55STWXYTZS[\ ] ] ^s6 D-BD-* D(4D(rrrF)rrXannotation_dcts r5generate_sample_annotationz)LabelInterface.generate_sample_annotationsh$%002 $..0  # #N 3! ! LL-n-==abfbmbmano pr7raise_on_failurec|j}|s|r tdy|j}|s |r td|j}|s |r td||r|gng|r|gdSgdS)aGenerate a complete sample task with annotations and predictions. This method combines the generation of a sample task, sample prediction, and sample annotation into a single method call, returning a complete task structure. Args: raise_on_failure: If True, will raise ValueError when any step (including annotation/prediction generation) fails Raises: ValueError: If raise_on_failure is True and any generation step fails. Example: { 'data': {'text': 'Sample text for labeling'}, 'annotations': [ {'was_cancelled': False, 'ground_truth': False, 'completed_by': -1, 'result': [ {'id': 'b05da11d-3ffc-4657-8b8d-f5bc37cd59ac', 'from_name': 'sentiment', 'to_name': 'text', 'type': 'choices', 'value': {'choices': ['Negative']}} ] } ], 'predictions': [ {'model_version': 'sample model version', 'score': 0.95, 'result': [ {'id': 'e7bd76e6-4e88-4eb3-b433-55e03661bf5d', 'from_name': 'sentiment', 'to_name': 'text', 'type': 'choices', 'value': {'choices': ['Neutral']}} ] } ] } NOTE: `id` field in result is not required when importing predictions; it will be generated automatically. NOTE: for each control tag, depends on tag.to_json_schema() being implemented correctly zJLabelInterface.generate_sample_task failed to generate a valid sample taskNzOLabelInterface.generate_sample_prediction failed to generate a valid predictionzOLabelInterface.generate_sample_annotation failed to generate a valid annotation)rr<r=)rrrr)rr sample_tasksample_predictionsample_annotations r5generate_complete_sample_taskz,LabelInterface.generate_complete_sample_tasks\//1  !mnn ;;= %5no o ;;= %5no o 2C-.2C-.  JL  r7cjt|}|jjD]\}}||jvry|j|}|j|jk7ry|j|jk7ryt |j j|j ryy)z/Detect essential changes of the labeling config)rFTF)rgrrTr|rr+rissubset)rnew_config_strnew_obj new_tag_namenew_tagold_tags r5!config_essential_data_has_changedz0LabelInterface.config_essential_data_has_changeds 7%,%6%6%<%<%>  !L'4>>1nn\2G{{gkk)'//1w~~&//? r7ctrr8)rWrrs r5"generate_sample_task_without_checkz1LabelInterface.generate_sample_task_without_checks r7cidd}}}|jd}|dk\r|n|jd}|dz }|dk\r||djdnd}d|cxkr||zkrNnnK tj||||z}d |vxsd |v}d |vr|d n|rdn|}d |vr|d nd}d |vr|d nd}|||fS#t$rYwxYw) zGet task, annotations and predictions from labeling config comment, it must start from "" Nzrr=r<r)findjsonloadsr) rtrF task_datar<r=startendbody dont_use_roots r5rz,LabelInterface.get_task_from_labeling_configs  /1$ ;  H%!W)=  ,1QJfUVn!!%(B u "us{ " Uzz&"=> !. 5 N$9N ~L"/$T 6Cd5Jd=1PT 5Bd5Jd=1PT +{22  s"B11 B=<B=c6tj|d}|jd}|D]K}|j}||j |tj |dj d}M|jddjd dS) NF) forbid_dtdz //comment()html)methodzutf-8rrd )rrxpath getparentremovetostringdecoder)rcrcommentsps r5config_line_stippedz"LabelInterface.config_line_stipped6se4::m, DA A} tF3::7CA  D yyr"**444r7cdtjDcgc]}|jc}Scc}wr)rras_tuple)rWr|s r5r\z)LabelInterface.get_all_control_tag_tuplesCs *.--83 888s- control_type object_type name_filterto_name_filterc|jD]a}|j||s|jD]<}|j||s|j|j|jfccSct d|d|d)a Reads config and fetches the first control tag along with first object tag that matches the type. Args: control_type (str or tuple): The control type for checking tag matches. object_type (str or tuple): The object type for checking tag matches. name_filter (function, optional): If given, only tags with this name will be considered. Default is None. to_name_filter (function, optional): If given, only tags with this name will be considered. Default is None. Returns: tuple: (from_name, to_name, value), representing control tag, object tag and input value. )name_filter_fn)to_name_filter_fnzNo control tag of type z and object tag of type z found in label config)rmatchrrkrr)rrrrrr| object_tags r5get_first_tag_occurencez&LabelInterface.get_first_tag_occurenceGs,== PCyykyB"%++PJ!'' ~'V"xx*:O:OOOP P %l^3KK=Xn o  r7c|jDcic]}|js|jd}}|j|fScc}w)rbT)r dynamic_valuerkr)rrdynamic_valuess r5get_all_labelszLabelInterface.get_all_labelsgs<04 Q1!&&$,QQ||^++Rs AAc6|jjSr)rr,rs r5get_all_object_tag_namesz'LabelInterface.get_all_object_tag_namesls}}!!##r7c|jSr)rrs r5extract_data_typesz!LabelInterface.extract_data_typesps }}r7cBd}|j|}t|S)rbcB|jjtvSN)r|rM_VIDEO_TRACKING_TAGS)r|s r5rz9LabelInterface.is_video_object_tracking..vssww}}2FFr7)r)rbool)rrrus r5is_video_object_trackingz'LabelInterface.is_video_object_trackingts!F~~x~0Dzr7ctrr8)rrs r5is_typezLabelInterface.is_type{sr7c|jsy|j}|j}|j}|j ||j ||y)z0Validate current config using LS Project SummaryFN)rcreated_labelscreated_labels_draftscreated_annotations validate_annotations_consistencyvalidate_lables_consistency)rsummarystrictrrannotations_summarys r5validate_config_using_summaryz,LabelInterface.validate_config_using_summarysQ}} // ' = =%99 --.AB ((9NOr7c t||}|jD]\}}|j|}|r,|s*tdt |j dd|dg}|jD]9\}}|dkDs |j j|dr)|j|;t|stddj|dt|d y) rbz There are rz! annotation(s) created with tag "z", you can't remove itNz3These labels still exist in annotations or drafts: rJz$Please add labels to tag with name="z".) r6rTrrsumr labels_attrsr.rUrDrLrp) rrrcontrol_tag_from_datalabels_from_datacontrol_from_configremoved_labels label_name label_values r5rz*LabelInterface.validate_lables_consistencys /~?TU 8F7K7K7M"  3 !#3"&"2"23H"I (;= %5%<%<%>!B CD-..EG  N+;+A+A+C 6' K?+>+K+K+O+O, #))*5  6$>"=J388TbKcJd:3?T;U:VVXZ?" r7cg}t|}|D]j}|jd\}}}|jdk(r, |j|}|r|j |s|j d|d|d|l|rd j|} td | y#t $r0} |j d|d|d|dt| Yd} ~ d} ~ wwxYw) rbrKtextareazwith from_name=z , to_name=z, type=z*Error occurred while processing from_name=z , error: NrzNCreated annotations are incompatible with provided labeling schema, we found: ) r+r$rMrrrUrrprLr) rrerrannotations_from_dataannrNrOrrexdiff_strs r5rz/LabelInterface.validate_annotations_consistencys #$7 8( C+.99S> (Iw~~:- **95g&8&8&AJJ))JwiwxjY , yy~H9abjakl    @ :V]U^^efneooxy|}zAyBC s>B C(&CC)NNNTr)NN)F)rr)rN)FrN)r)Fr)uploadF)Orr __module__ __qualname____doc__ classmethodr"rrprrr rr rpropertyrFrrrrrrrrrrrr rrrrrrrrr_Elementrr r r*r-r3r5rr:rArrErCr>r?rorfrOrrPrrrrqrrrrrrrr\rrrrrrrrrr;r7r5rgrgs&P&6&6R,,"-s-`,4U3d3id3S-T(T#U,Z^_eZf,\::8866tdDT( +;4 2& 3StS(NR$  $8@8J$ $LddMMKK< &W33W35tT5>>1Q+RW3r   I M)(&$E$S / 5 59+/-1  CJ' 3:& h'  !*  sC}   @, $ P,\$r7rg)Nrosrloggingr"rr functoolsrtypingrrrrr r r pydanticr collectionsr rlxmlrr jsfr#label_studio_sdk._legacy.exceptionsrrrbaser control_tagsrrrrr object_tagsr label_tagsrrrrrr rdr"rn getLoggerrrrrdirnamerealpath__file__dir_pathrL file_pathopenfloadr _LABEL_TAGS _DIR_APP_NAMErrMr6rErQr\r^rrprergr;r7r5rsB  %DDD1 !# HH   8 $ 77??277++H5 6 GGLL4H>X Y  )_- ) ! -.  ()  D68 77C7HSM7FFc--s :EE