o e7@sdZddlmZddlZddlZddlmZmZddlm Z m Z ddl Z ddl mZddlmZe r8ddlmZGd d d eZGd d d eeZGd ddeeZd"ddZd#ddZd$ddZd$d d!ZdS)%zPContains classes and methods related to interpretation for components in Gradio.) annotationsN)ABCabstractmethod) TYPE_CHECKINGAny)utils) components) Interfacec@s(eZdZdddZddZdddZdS) InterpretablereturnNonecCs |dSN)set_interpret_parametersselfrED:\Projects\ConvertPro\env\Lib\site-packages\gradio/interpretation.py__init__s zInterpretable.__init__cCsdS)z Set any parameters for interpretation. Properties can be set here to be used in get_interpretation_neighbors and get_interpretation_scores. Nrrrrrrsz&Interpretable.set_interpret_parametersxr neighborslist[Any] | Nonescores list[float]listcKs|S)a Arrange the output values from the neighbors into interpretation scores for the interface to render. Parameters: x: Input to interface neighbors: Neighboring values to input x used for interpretation. scores: Output value corresponding to each neighbor in neighbors Returns: Arrangement of interpretation scores for interfaces to render. r)rrrrkwargsrrrget_interpretation_scoress z'Interpretable.get_interpretation_scoresN)r r )rrrrrrr r)__name__ __module__ __qualname__rrrrrrrr s r c@s(eZdZedddZedd d Zd S)TokenInterpretablerrr tuple[list, list, None]cCs ggdfS)z Interprets an input data point x by splitting it into a list of tokens (e.g a string into words or an image into super-pixels). Nrrrrrrtokenize.s zTokenInterpretable.tokenizetokensrbinary_mask_matrix list[list]cCsgSr r)rr#r$rrrget_masked_inputs6sz$TokenInterpretable.get_masked_inputsN)rrr r )r#rr$r%r r)rrrrr"r&rrrrr-s  rc@seZdZedddZdS) NeighborInterpretablerrr tuple[list, dict]cCsgifS)a Generates values similar to input to be used to interpret the significance of the input in the final output. Parameters: x: Input to interface Returns: (neighbor_values, interpret_kwargs, interpret_by_removal) neighbor_values: Neighboring values to input x to compute for interpretation interpret_kwargs: Keyword arguments to be passed to get_interpretation_scores rr!rrrget_interpretation_neighbors<s z2NeighborInterpretable.get_interpretation_neighborsN)rrr r()rrrrr)rrrrr';sr' interfacer raw_inputrc stjtrއfddtjDdIdHdtjdkr+ggg}}ttjD]\\}}|dkr5jttt r |\}}g}g} |D]C} | <fddtjD} d| IdHdtjdkrgfd dtjD} | | | t q`| | | j |||d q8ttr-|\}} g}g} |D]C} | <fd dtjD} d| IdHdtjdkrgfd dtjD} | | | t q| | d d|D}| j ||fi| q8tdd|dks?|dkrzddl}WnttfyY}ztd|d}~wwjtt smtdd |\}}fdd}t}||td|f}|jtd|ftj|dd}|dusJd| j d|d|d | gq8|dur| d| gq8td|||fSjrfddtjDj}|}tdkr|g}|gfStd)a1 Runs the interpretation command for the machine learning model. Handles both the "default" out-of-the-box interpretation for a certain set of UI component types, as well as the custom interpretation case. Parameters: raw_input: a list of raw inputs to apply the interpretation(s) on. cg|] \}}||qSr preprocess.0iinput_componentr+rr Q z!run_interpret..rN predictiondefaultcr,rr-r/neighbor_raw_inputrrr4gr5cr,r postprocessr0r1output_componentneighbor_outputrrr4tr5)masksr#cr,rr-r/r9rrr4r5cr,rr;r=r?rrr4r5cSsg|]}| qSrr)r0scorerrrr4sz Component z does not support interpretationshapZshapleyzVThe package `shap` is required for this interpretation method. Try: `pip install shap`zInput component z' does not support `shap` interpretationcsttsJ|}g}|D]/}t}||<tjd|}|d}t j dkr5|g}t |}| |qt |S)Nrr6r7) isinstancerr&copydeepcopyr. client_utilsZsynchronize_async call_functionlenoutput_components&get_regression_or_classification_valueappendnparray)Z binary_maskZ masked_xspredsZmasked_xZprocessed_masked_inputZ new_outputpred)r1r2r*original_outputprocessed_inputr#rrget_masked_predictions"    z,run_interpret..get_masked_predictionT)ZnsamplesZsilentz#SHAP values could not be calculatedzUnknown interpretation method: cr,rr-r/r3rrr4r5z#No interpretation method specified.)rDinterpretationr enumerateZinput_componentsrHrIrJziprr"rLquantify_difference_in_labelrr'r) ValueErrorrC ImportErrorModuleNotFoundErrorZKernelExplainerrMZzeros shap_valuesZonesintZnum_shaptolist)r*r+rZalternative_outputsrZinterpZneighbor_valuesrAZinterface_scoresZalternative_outputZneighbor_inputZprocessed_neighbor_inputZprocessed_neighbor_outputZinterpret_kwargsrCerr_rSZnum_total_segmentsZ explainerr[ interpreterrTr) r1r2r*r@r:rQrRr+r#r run_interpretIs0                            raoriginalr perturbedr int | floatcCs8z t|t|}W|Styt||k}Y|Swr )floatrXr\)rbrcrBrrrdiffs rfrQperturbed_outputc Cs|jd}||d}||d}t|tjr>|d}|d}d|vr7|d|}|d|} || } | St||} | St|tjrKt||} | Std|)Nrlabel confidencesAThis interpretation method doesn't support the Output component: )rJr<rDrLabelrfNumberrX) r*rQrgr>post_original_outputpost_perturbed_outputoriginal_labelperturbed_labelZoriginal_confidenceZperturbed_confidencerBrrrrWs&       rWc Cs|jd}||d}||d}t|tjr=|d}|d}d|vr6t|d|r0dS|d|St||}|Std|)zIUsed to combine regression/classification for Shap interpretation method.rrhrirj) rJr<rDrrkmathisnanrfrX) r*rQrgr>rmrnrorprBrrrrK.s"   rK)r*r r+r)rbrrcrr rd)r*r rQrrgrr rd)__doc__ __future__rrErqabcrrtypingrrnumpyrMZ gradio_clientrrGZgradiorr r rr'rarfrWrKrrrrs&      @