o ƧjI@sUddlmZddlZddlZddlZddlZddlZddlZddlm Z m Z ddl m Z m Z mZmZmZmZmZddlZddlZddlZddlmZeidddd d d d d dddddddddddddddd dd dd ddddd ddddd!Zed"d"d#d$d$d%d&d&d&d'd"d(d'd) Zed*Zd+Zdd/d0Z dd1d2Z!dd3d4Z"dd5d6Z#dd7d8Z$dd9d:Z%dd>d?Z&ddCdDZ'ddEdFZ(ddHdIZ)ddJdKZ*ddMdNZ+dOZ,ddRdSZ-ddTdUZ.ddXdYZ/dd]d^Z0ddadbZ1dddfdgZ2dddjdkZ3GdldmdmZ4Gdndodoe4Z5Gdpdqdqe4Z6da7dre8ds<e9Z:ddtduZ;dd{d|ZdddZ?iZ@de8d<e9ZAdddZBGdddZCdddZDdddZEdddZF ddddZGdddZHdddddddddddd dddƄZIdS)) annotationsN) dataclassfield)DictAnyListOptionalTupleIterableCallable)YOLOu۰0u۱1u۲2u۳3u۴4u۵5u۶6u۷7u۸8u۹9u٠u١u٢u٣u٤u٥u٦)u٧u٨u٩uیuکuهuوuاuلا) uيuىuكuةuۀuؤuإuأuٱuءuئuﻻuـz/[\u0610-\u061A\u064B-\u065F\u0670\u06D6-\u06ED]u‌sstrreturncCstdd|pdS)Nz\s+ r)resubstriprr &/var/www/html/kyc_policy_engine_llm.pycollapse_spaces;sr"cCs|pdtS)Nr) translate_FA_AR_TO_EN_DIGITSrr r r! to_en_digits?r%cCs:|pd}t|}|t}|td}td|}t|S)Nrr)r%r#_AR_FA_LETTERSreplace_ZWNJ_DIACRITICS_RErr"rr r r!normalize_fa_textCs    r+cCs(t|}tjdd|tjd}t|S)Nz0[^\w\u0600-\u06FF\u0750-\u077F\u08A0-\u08FF\s\-]rflags)r+rrUNICODEr"lowerrr r r!normalize_nameLs r0cCst|}tdd|S)Nz[^0-9]r)r%rrrr r r!normalize_digits_onlyRsr1cCst|}tdd|S)Nz [^0-9A-Za-z]r)r%rrupperrr r r!normalize_id_alphanumWsr3vrboolcCsBt|tr|St|ttfr|dkSt|tr|dvSdS)Nr>ryonyestrueenabledrequiredF) isinstancer5intfloatrrr/)r4r r r! is_truthy\s  r?abr>c Cs|pd}|pd}||krdS|r|sdSttt|d}t|ddD]5\}}|g}t|ddD]%\}}||kr;dnd}|t||d||dd||d|q1|}q$|d} d| tt|t|S)Nr?)startr)listrangelen enumerateappendminmax) r@rAprevicacurjcbcostdistr r r!levenshtein_ratiojs4rVcCsrt|}t|}|r |sdSt|}t|}t||@}t||B}|r,||nd}d|dt||S)NrC333333?g?)r0setsplitrIrV)r@rAsasbinterunionrRr r r!token_set_similarity~s    r^set[str]cCs,t|}|s tSdd|ddDS)NcSsh|]}|r|qSr r ).0tokr r r! z"_name_token_set..-r)r0rXr(rY)rnormr r r!_name_token_setsrfcCs,t|}t|}|r |sdS||kp||kS)u True when one normalized name token set is a strict subset of the other. Example: "محمد غلام" vs "محمد غلام نژاد". F)rf)r@rAtatbr r r!names_have_strict_subsets riOptional[Tuple[int, int, int]]c Cs8t|pd}|dddd}td|}t|dkrdSdd |ddD}|d |d |d f|d |d |d f|d |d |d f|d |d |d f|d |d |d f|d |d |d fg}|D]0\}}}d |krxdkrnqid |krdkrnqid |krdkrnqi|||fSqidS)z~ Accepts common formats and returns (YYYY, M, D) if parseable. This is only for validation/matching, not for OCR. rrd/.z\d{1,4}NcSg|]}t|qSr r=r`xr r r! rcz$parse_date_loose..rrDii )r%r(rfindallrI)rnumsvalspermsr6mdr r r!parse_date_looses$   Hr|)348343344code Optional[str]cst|}tD]}||rd|Sqt|dkrdStt|dkr'dSdd|Dtfddtd D}|d }d }|d krO||krMdSdS|d |krWdSdS) Nz*iran_national_code_sanctioned_city_prefix: iran_national_code_invalidrDcSrnr ror`cr r r!rrrcz5iran_national_code_invalid_reason..c3s |] }|d|VqdS)rNr )r`rOdigitsr r! sz4iran_national_code_invalid_reason.. rs)r1#IRAN_NATIONAL_CODE_BLOCKED_PREFIXES startswithrIrXsumrH)rprefixrrcheckr rr!!iran_national_code_invalid_reasons   rcCs t|duSN)r)rr r r!iran_national_code_is_valids rpts np.ndarraycCs||tj}|jdd}|t|}|t|}tj|ddd}|t|}|t|}tj||||gtjdS)NrD)axisrFdtype) astypenpfloat32rargminargmaxdiffreshapearray)rrtlbrr{trblr r r!order_points_clockwises  rwr=hcCsXt|dddfd|d|dddf<t|dddfd|d|dddf<|S)NrrD)rclip)rrrr r r! clip_pointss**rimg_bgrquadcCs |jdd\}}|tj}t|||}t|}|\}}}}tj||}tj||} tt || } tj||} tj||} tt | | } t | d} t | d} tj ddg| ddg| d| dgd| dggtjd}t ||}t j ||| | ft jdS)NrsrrDrr,)shaperrrrrlinalgrer=rMrcv2getPerspectiveTransformwarpPerspective INTER_CUBIC)rrHWrrrrwidthAwidthBmaxWheightAheightBmaxHdstMr r r!warp_quad_to_rects      8 rimgmax_dimcCsv|jdd\}}t||}||kr|S|t|}tdtt||}tdtt||}tj|||ftjdS)Nrs) interpolation)rrMr>r=roundrresize INTER_AREA)rrrrrzscalenwnhr r r!_resize_max_dims  rU jpeg_qualitycCsVt||d}td|ttjt|g\}}|stdt| d}d|S)zD Converts an image to a compact data URL to feed to an LLM. )rz.jpgzcv2.imencode(.jpg) failedasciizdata:image/jpeg;base64,) rrimencoder=IMWRITE_JPEG_QUALITY RuntimeErrorbase64 b64encodetobytesdecode)rrrokbufb64r r r!_encode_jpeg_data_url s   rc@eZdZd ddZdS) FaceMatcherBase doc_photo_bgrr selfie_bgrrDict[str, Any]cCstr)NotImplementedErrorselfrrr r r!matchszFaceMatcherBase.matchNrrrrrr__name__ __module__ __qualname__rr r r r!rrc@r) FaceMatcherUnavailablerrrrrcCsdddddS)Nface_module_unavailablescore01cosinereasondetailsr rr r r!r#r&zFaceMatcherUnavailable.matchNrrr r r r!r"rrc@s<eZdZddddZed d ZedddZdddZdS)InsightFaceMatcherrFdet_sizeTuple[int, int] prefer_gpur5cCs@ddlm}|r ddgndg}|d|d|_|jjd|ddS) Nr) FaceAnalysisCUDAExecutionProviderCPUExecutionProvider buffalo_l)name providersrF)ctx_idr)insightface.apprappprepare)rrrrrr r r!__init__(s zInsightFaceMatcher.__init__cCs|sdSdd}t||dS)NcSs"|j\}}}}t||||Sr)bboxr>)fx1y1x2y2r r r!area3sz.InsightFaceMatcher._largest_face..areakey)rM)facesrr r r! _largest_face.s z InsightFaceMatcher._largest_facer@rrArr>cCsP|tj}|tj}tj|d}tj|d}tt||||S)Ng& .>)rrrrrer>dot)r@rAnanbr r r!_cosine9s  zInsightFaceMatcher._cosinerrrc Cs|j|}|j|}||}||}|dur%ddddt|idS|dur4ddddt|idS||j|j}|dd}t|t|dddS) Nno_face_in_document_photo doc_facesrno_face_in_selfie selfie_facesrB@r)rgetrrIr normed_embeddingr>) rrrr  sel_facesfdfscosrr r r!rAs     zInsightFaceMatcher.matchN)rF)rrrr5)r@rrArrr>r)rrrr staticmethodrr rr r r r!r's   rzOptional[FaceMatcherBase] _FACE_MATCHERc Csnt+tdurtWdSztddaWn ty#taYnwtWdS1s0wYdS)NF)r) _FACE_LOCKrr Exceptionrr r r r!get_face_matcherVs  $rdoc_image_pathselfie_image_pathdoc_photo_quadOptional[List[List[float]]]rcCs|s dddddS|sdddddSt|}t|}|dur'dddddS|dur2dddddSt|tj|tjd}t}|||S)z^ Computes face match pack once. Caller decides whether to use it depending on policy. Nselfie_missingrdoc_photo_not_detectedcannot_read_doc_imagecannot_read_selfie_imager)rimreadrrrrrr)rrrdoc_imgsel_imgcropmatcherr r r!compute_face_packbs    r)c@seZdZUded<ded<ded<dZded <d Zded <d Zded <dZded<dZded<dZ ded<dZ ded<e e dZ ded<dZded<dZded<dZded<d Zded!<dZd"ed#<dS)$ FieldRule List[str] class_namesrrr5r;?r> min_det_confg? min_ocr_conftextocr_kindoptional match_type333333?match_thresholdrr= expected_lenNr validatordefault_factoryr constraintsrBweightF must_match match_gaterDmax_candidateszOptional[List[str]] input_aliases)rrr__annotations__r.r/r1r3r5r6r7rdictr:r;r<r=r>r?r r r r!r*s"            r*c@seZdZUded<ded<dZded<dZd ed <d Zd ed <d Zd ed<dZd ed<dZ d ed<dZ d ed<dZ d ed<dZ ded<dZ ded<dZd ed<eedZded<d Zded!<d"Zd ed#<dS)$ DocConfigrdoc_idzList[FieldRule]rulesNz Optional[int]min_detected_fields_countrBr>approve_min_coverage(\?approve_min_extractionq= ףp?approve_min_match_corerCapprove_min_match_all?review_min_coverage?reject_below_coverage皙?approve_no_input_extra_bufferFr5require_face_matchr face_metricface_match_thresholdr8zList[Tuple[str, str]] swap_pairsTenable_name_swapQ?name_swap_margin)rrrr@rErFrHrJrKrMrOrQrRrSrTrrGrUrVrXr r r r!rBs"             rBpayloadc Csg}|dgD]f}t|}|dp|d}t|tr#|g|d<nt|p'g|d<|dd|d}|dur?i|d<n%t|tr[z t||d<WntyZi|d<Yn wt|tsdi|d<| t d$i|q|dpug}dd|D}t d$idt|dp|d pd d|d |d dd t |d d dt |dddt |d|dddt |dddt |dddt |dddt |dddt |dddt|dddt |ddd|d t |d d!d"t |d"d#S)%NrDr, class_namer:rUcSs.g|]}t|ttfrt|dkrt|qS)rs)r<rGtuplerIrpr r r!rrs.z+doc_config_from_payload..rCiddocrErFrBrHrGrJapprove_min_matchrIrKrCrMrLrOrNrQrPrRFrSrrTrVTrXrWr )rrAr<rrGpopjsonloadsrrKr*rBr>r5)rYrDrrrcncstrUr r r!doc_config_from_payloadsn             rezDict[str, 'KYCEngine'] _ENGINE_CACHE model_path 'KYCEngine'cCsLt|}t|tvrt|t|<t|WdS1swYdSr)r _ENGINE_LOCKrf KYCEngine)rgrr r r! get_engines  $rkc@sjeZdZd7ddZed8dd Zd9ddZd:ddZeddd;dd Z ! " # $dd5d6Z d!S)?rjyolo_model_pathrcCsTt||_|jj}t|trdd|D|_n ddt|D|_t |_ dS)NcSsi|] \}}t|t|qSr )r=r)r`kr4r r r! sz&KYCEngine.__init__..cSsi|] \}}|t|qSr )r)r`rOnr r r!rns) r modelnamesr<rAitemsr,rJ threadingLock _infer_lock)rrlrqr r r!rs  zKYCEngine.__init__rqr>rcCstdtdt|S)NrCrB)rMrLr>rqr r r!_clamp01 szKYCEngine._clamp01ruler* user_inputDict[str, str]rcCs@||j}|r |S|jr|jD] }||}|r|SqdSr)rrr?)rrxryr4akvvr r r! _get_expecteds   zKYCEngine._get_expected image_pathconfioudevicemax_detr=List[Dict[str, Any]]c Cs>|j|jj|||||dd}Wdn1swY|s#gS|d}g}t|dddur|jdur|jj} |jj} |jj} t| t j rP| } t| t j r^| } t| t j rl| } tt| D]$} t| | } || |j| t| t| | | | tdqr|St|dddur|jdur|jj}|jj} |jj} t|t j r| }t| t j r| } t| t j r| } tt|D]8} dd|| D\}}}}t| | } ||g||g||g||gg}|| |j| t| t| | |dq|S) NF)sourcerrrrverboserobb)class_idrZrrboxescSrnr )r>rpr r r!rrJrcz$KYCEngine.detect..)rurppredictgetattrrxyxyxyxyclsrr<torchTensordetachcpunumpyrHrIr=rKr,rrr>rtolistrxyxy)rr~rrrrpredsrdetsquadsrcfrOcidrrrrrrr r r!detectsb             zKYCEngine.detectfa) locale_hint fields_planrrc Cs(dd||ddgddddd d d }|S) a Returns a single, self-contained bundle to be sent to your LLM layer. You can convert it to OpenAI Responses API (or any other provider) on your server. fields_plan: list of: { "key": str, "kind": "text|numeric|date|mrz", "candidates": [{"idx": int, "det_conf": float, "image_data_url": str}, ...] } Expected LLM response: { "fields": { "": { "candidate_idx": 0, "text": "...", # what you read "confidence": 0.0-1.0, # optional but recommended "normalized": "..." # optional; server can re-normalize anyway }, ... } } rD kyc_field_ocrr`fieldszint (required)zstring|null (required)zfloat 0..1 (optional)zstring|null (optional)) candidate_idxr0 confidence normalized)typetop_level_keys field_object)schema_versiontaskrroutput_contractr )rrschemar r r!build_llm_ocr_prompt_bundleXsz%KYCEngine.build_llm_ocr_prompt_bundleNF\rdoc_cfgrBrllm_ocr*Callable[[Dict[str, Any]], Dict[str, Any]]Optional[Dict[str, str]]debugr5rrc ) s|pi}t|} | durtd|i} |D] } | | dg| q| D] } | | jdddq(i} d7d d }g}it| jdd d}i}|jD]g}j D] }| | |gqV|seqOfdd|D}|jddd|d|}|sqOj pd }|dkr|d}j dt|ddddd|ddd|j<|d|dj<qOj|gd}g| j<t|D]1\}}tj|dtjd}t| |}t|||d}|d|t|d|d| j|q||qO|j|dd}||pi}t|tr| d nd}t|tsi}d8d$d%}d9d:d)d*} |jD] j p2d }|dkr<q*j| vrDq*| jpLi}!t|!tsUi}!|! d+}"zt|"}"Wn tyld}"Ynw| jpug}#|#s{q*|"dks|"t|#krd}"|#|"}t| dd,}$|! d}%|%dur|! d#}%|%durd-}%t|%}%|! d.}&|&dur| ||%tjpɈj d/pdd0}&zt|&}&Wnty| ||%tjpj d/pdd0}&Ynw||&}&|! d1}'|'dus t|' d-kr|||%}'n||t|'}'j d|$|&|% |'d2|d|"d|j<|d|dj<q*||d3}(|rM||(d4<||(d5<||(d6<|(S);z 1) Uses YOLO detections + rules to crop candidates. 2) Builds ONE bundle and calls llm_ocr(bundle). 3) Maps response back to per-field extraction pack (same shape as before). NzCannot read image: rZcSt|dd SNrrCr>rrvr r r!z,KYCEngine.extract_with_llm..rrxr*rr=cSs$t|jpd}|dkr d}t|dS)NrDr)r=r>rL)rxrmr r r!top_ks z)KYCEngine.extract_with_llm..top_krs)rdoc_image_shapecs*g|]}t|ddtjkr|qS)rrC)r>rr.rrxr r!rrs*z.KYCEngine.extract_with_llm..cSrrrrvr r r!rrr0nonerrr)rZdet_confocr_conf value_raw value_norm ocr_methodrrr)rkind candidatesr)rrr)idxrimage_data_urlr)rrrrrvaluecSs`|dkrt|S|dkrtt|S|jdvrt|S|jdvr,|dkr(t|St|St|S)Nnumericdate first_name last_name full_name) id_number passport_no)r1r"r%rr0r3r+)rxrrr r r!normalize_values   z3KYCEngine.extract_with_llm..normalize_valuerr6r>cSs|pd}|s dS|dkrt|}|rt||krdSdS|dkr*t|r(dSdS|d kr>d |vr.heuristic_confrrCrrlength)r6rllm)r internals detections llm_bundlellm_response_raw)rxr*rr=)rxr*rrrrrr)r)rrr0rr6r=rr>)!rr$r setdefaultrKsortrGrrDr,extendrr1r/r>rrJrrrrrrr<rAr=rrIrr6r:rwr))rrrrrryrrrrby_classr{rmcandidates_for_keyrrrrrrcrbest plan_entryrdetrr'data_urlbundlellm_resp llm_fieldsrrrespcand_idxdet_listrraw_textr norm_textoutr rr!extract_with_llms                         ( ,    zKYCEngine.extract_with_llmextracted_fields face_packOptional[Dict[str, Any]]c< Cs|pi}dd|jD}d}t}g} g} g} i} g} g}dJd d}dKdd}|jD]a}||j}|sBddddddd| |j<q*|t|dd}|t|jk}|r\||j|jp`d }|dkrwd}t |d}d}d}d}n4|t|ddpd}|dpd}|d pd}|t|j kot t|}t |td!|}|jr|r|d"7}|||}d}d}d}d}d} d}!|jd#kr|dkrt|pd}"|"du}|"}|"durd}|"}|r|d$ks|jd%vrt|nt|}#t|#}$|$du} |$}!|$durd}|dusd&|$vr|$}|||||\}%}&|%s)d}|p(|&}|s8|d'|jd(|| t|t|jfd}'d)}(|r|dkr|jpUd* })|)d+kr|pad}*|}+|d$krrt|*}*t|+}+n|jd,vrt|*}*t|+}+nt|*}*t|+}+|*|+krdnd}'|'dkrd-nd.}(nk|)d/krt|jpid0},|jd,vr|,rt|pd|rd}'d1}(nDtt|pd|}'|'t|jkrd/nd2}(n-|)d3krt |pd}-t |}.|-r|.r|-|.krdnd}'|'dkrd4nd5}(nd}'d6}(|'dur(| t|'t|jf|j!r(| t|'t|jf|j"rM|'dkr?|(d7vr?| d8|j|(d9vrM| d8|j|||||t||'|(||d: }/|jrl|/#|j||| |!d;|r|/#|||d<|d=|d>|d?d@|/| |j<q*|r|td"t$|nd}0|| }1| r|| nd}2| r|| nd}3d}4|j%durt$|t&|j%k}4d}5d}6|dur|dA}6|j'dBkr|dB}5n|dC}5t|3dur|3n|2pd}7|dDt|1dD|7}8|8}9|j(r|5dur|dE|8dF|t|5}9n|8}9t)||0|1|2|3|4t$|| ||5|6dG \}:};|:|;|0|1|2|3t$||4|5|j'|8|9dH | dIS)LNcSsg|]}|jr|qSr )r;)r`rr r r!rrXsz#KYCEngine.score..rrxr*rrrawrrerTuple[bool, Optional[str]]cSs|jpi}|s dS|p d}|pd}|d}|d}|d} |durBzt|} t|| kr7dd| fWSWn tyAYnw|durdzt|} t|| krYdd| fWSWn tycYnw| durzt| } t|| kr{dd | fWSWn tyYnw|d } | rt| } t|| sdd | fS|d } | rztt| t|durWd SWn tj yYnw|dkr|ddrt |durdSdS)N)TNrr min_length max_lengthFlength_expected_ min_length_ max_length_rprefix_regex)Fregex_mismatchr must_parse)F date_unparsed) r:rr=rIrrrr fullmatcherrorr|)rxrrrerraw_snorm_srrrLrprefrr r r!validate_constraintsesj             z-KYCEngine.score..validate_constraintspairsList[Tuple[float, float]]r>cSs@|sdStdd|D}|dkrttdd|D|SdS)NrCcss|]\}}|VqdSrr )r`_rr r r!rz0KYCEngine.score..wavg..rcss|] \}}||VqdSrr )r`r4rr r r!rs)rr>)rswr r r!wavgs&zKYCEngine.score..wavgFrCmissing)presentdet_okocr_ok extractionrrrr0rTrBrrrrg{Gz?rDiran_national_coder>rrsanctioned_city_prefixzinvalid:: no_user_inputr2exactr exact_matchexact_mismatchfuzzyreject_incomplete_namename_incompletefuzzy_below_thresholdr date_match date_mismatchr)rrz mismatch:)rr) r r r rrr rrvalid valid_reason)r7 validator_okvalidator_reasonexpected_validator_okexpected_validator_reasonrZrrr)rrrZrrrrrrrNrgffffff?) rcoverager  match_all match_core count_gate_okdetected_countmismatch_flags invalid_flags face_value face_reason) r!r r"r#detected_fields_countr$facerS doc_score final_score)decisionreasonsscores per_field) rxr*rrrrrerrr)rrrr>)*rDrXrrrwr>r.addr1r/mathsqrtr/r5rrrMr;r}r7rr1rKr;r3r0r+r?r:rir^r5r|r=r<updaterIrEr=rSrRdecide)rr>)rxr*ryrzrr) r~rrr>rr>rrrr=rr)rrrrrr)NFrr)rrrrBrrrrryrrr5rr=rr=rr) rrBrrryrrrrr5rr) rrrrrrwr}rrrrPr r r r!rjs     ?4 Hrjrr!r r"Optional[float]r#r$r%r&r+r'r(r)Tuple[str, List[str]]c Csg} |jdur|s| d| d|d|jd| fS||jkr,| dd| fS|rH| |ddtdd|DrDd| fSd | fS|rW| |ddd | fS|jr| durl| d | pdd d | fSt| t|jkr| d | d | dd|jdd | fS||jkr||j kr|dur||j |j krd| fS| dd | fS||j krd| fS| d|dur| d|dd | fS||j kr||j kr| d||jkr| dd | fS| dd | fS)Ndetected_fields_count_too_lowz detected=z_min=REJECTcoverage_too_lowrtcss|]}d|vVqdS))iran_national_code_sanctioned_city_prefixNr )r`flagr r r!rrzdecide..REVIEWzface_required_but_unavailable:unknownface_below_thresholdzface=z.3fz_thr=APPROVEno_core_user_inputcore_match_below_thresholdz match_all=extraction_lowpartial_coveragecoverage_insufficient) rErKrOranyrRr>rTrFrHrQrJrM) rr!r r"r#r$r%r&r'r(r)r/r r r!r6}sX              r6rrrNonecCs~||vr||vr||||||<||<|di}||vr0||vr0||||||<||<|dg|||ddS)Nr swap_events)r@rAr)rrrK)rrr@rArqr r r!swap_field_keyss  recfgcCs&|jD] \}}t||||ddqdS)Nconfig_swap_pairsr)rUre)rrrfr@rAr r r!apply_config_swapssrirW ocr_first_raw ocr_last_raw exp_firstexp_lastmarginc Cst|pd|pd}t|p d|pd}||d}t|pd|pd}t|p%d|p(d} || d} | ||kr@d|| || ddSd|| ||ddS)NrrT)z first->lastz last->first)swap score_normal score_swappedrF)z first->firstz last->last)r^) rjrkrlrmrns11s22normals12s21swappedr r r!best_name_assignments   rxryrzc Cs|jsdSd|vs d|vrdS|dpd}|dpd}|r!|s#dS|ddp+d}|ddp4d}t|||||jd}|dr]t||dddd|d gddd|d dSdS) Nrrrr)rnroheuristic_name_swaprhrc)r@rArmeta)rVrrxrXrerrK) rrrfryrlrmf_rawl_rawinfor r r!apply_name_swap_if_neededs  r~r-rWTF) doc_config_payload doc_configrryr yolo_confyolo_iourr cache_enginerrCrrrOptional[DocConfig]r4Optional[Callable[[Dict[str, Any]], Dict[str, Any]]]rrrrrrrcCs|dur t|}n |dur|}ntd|durtd| r"t|nt|}|j||| | | d}|j|||||| d}|d}|d}t|||t||||pPid}|jrv|ro|d dpf|d d }t |||d }nddd dd }|j ||||| d }||d|d|d|dd}|jrdnd}g}|jr|s| d|ddkr| d||d<||d<| r||d<||d<||d<||d<| d|d<| d|d<|S)NzBdoc_config_payload or doc_config must be provided (server-driven).z0llm_ocr callback is required in LLM-OCR edition.)rrrr)rryrrrr doc_photophoto)rrrr r)ryrrr.r0r/r1)rCr.r0r/r1r;disabledupload_selfie_requiredrTretake_document_photo selfie_mode next_stepsrr+rr) re ValueErrorrkrjrrrir~rRrr)rPrK)rgrrCrrrryrrrrrrrrfenginerpackrrrrscoringresultrrr r r!run_kycsd       r)rrrr)r4rrr5)r@rrArrr>)rrrr_)r@rrArrr5)rrrrj)rrrr)rrrr5)rrrr)rrrr=rr=rr)rrrrrr)r)rrrr=rr)rr)rrrr=rr=rr)rr)rrrrrrrr)rYrrrB)rgrrrh)rrBr!r>r r>r"rQr#rQr$r5r%r=r&r+r'r+r(rQr)rrrR) rrrrr@rrArrrrrb)rrrrrfrBrrb)rW) rjrrkrrlrrmrrnr>rr) rrrrrfrBryrzrrb)rgrrrrCrrrrrrrryrrrrr>rr>rrrr=rr5rr5rr)J __future__rosrr3r`rrs dataclassesrrtypingrrrrr r r rrrr ultralyticsr r maketransr$r'compiler*r)r"r%r+r0r1r3r?rVr^rfrir|rrrrrrrrrrrrr@rtrrr)r*rBrerfrirkrjr6rerirxr~rr r r r!s $                    +    7   M