o cBjÛ ã@sºdZddlmZddlZddlmZmZmZmZm Z ddl m Z d4d d „Z d5dd„Z d6dd„Z d7d8dd„Zd9d d!„Z d7d:d$d%„Zd;d)d*„Z   dd?d2d3„ZdS)@zNScore customer sentiment from transcript using keywords, LLM, or hybrid rules.é)Ú annotationsN)ÚAnyÚDictÚListÚOptionalÚTuple)ÚSentimentConfigHandlerÚtextÚstrÚreturnú List[str]cCs&|sgSt dt|ƒ¡}dd„|DƒS)Nz[\n,;|]+cSs g|] }| ¡r| ¡ ¡‘qS©)ÚstripÚlower)Ú.0Úpr r ús z#_split_keywords..)ÚreÚsplitr )r Úpartsr r rÚ_split_keywords srÚhaystackÚkeywordsÚmax_hitsÚintúTuple[int, List[str]]cCsXg}|r|s d|fS| ¡}|D]}t|ƒ|krn |r%||vr%| |¡qt|ƒ|fS)Nr)rÚlenÚappend)rrrÚhitsr Úkwr r rÚ_count_keyword_hitss   € r!ÚvaluercCs@t|pdƒ ¡ ¡}d|vs|dvrdSd|vs|dvrdSdS)NÚÚpositive>ÚposÚgoodÚhappyÚnegative>ÚbadÚnegÚangryÚneutral)r rr)r"r r r rÚ_normalize_sentiment_labels r-Ú transcriptÚspeaker_segmentsúOptional[List[Dict]]cCsÊ|st|pdƒSg}|D]K}t| d¡pdƒ ¡}t| d¡p%| d¡p%dƒ ¡}t| d¡p5| d¡p5dƒ ¡}|sÚconfigúDict[str, Any]Ú analysis_textúTuple[str, Dict[str, Any]]cCs4tdt| d¡p dƒƒ}t| d¡pdƒ}t| d¡pdƒ}t| d¡p%dƒ}t|||ƒ\}}t|||ƒ\}} t|||ƒ\} } tdt| d¡pHd ƒƒ} tdt| d ¡pTd ƒƒ} || krb||krbd }n*|| krm||krmd }n| d kr||d kr||d kr|d}n||krƒd }n ||krŠd }nd}|| | ||| ddœ}||fS)NéÚmax_keyword_hitséÚpositive_keywordsr#Únegative_keywordsÚneutral_keywordsÚkeyword_positive_thresholdéÚkeyword_negative_thresholdr(r$rr,Úkeyword)Ú positive_hitsÚ negative_hitsÚ neutral_hitsÚpositive_countÚnegative_countÚ neutral_countÚmethod)Úmaxrr8rr!)r?rArÚpos_kwÚneg_kwÚneu_kwÚ pos_countÚpos_hitsÚ neg_countÚneg_hitsÚ neu_countÚneu_hitsÚ pos_thresholdÚ neg_thresholdÚlabelÚdetailsr r rÚkeyword_sentiment_labelBs:ù rbÚkeyword_detailsúOptional[Dict[str, Any]]cCsì| d¡pd ¡}t| d¡pdƒ ¡}|pi}ddg}|r&| d|›¡| d¡r0| d ¡|dkr^|r^| d ¡p ÿ üÿ ürvÚbidú Optional[Any]Úhandlerú Optional[SentimentConfigHandler]c CsŽ|pt|piƒ}| ¡| |¡}t| dd¡ƒ}|r t||ƒnt|p$dƒ}t||ƒ\}} t||| ƒ} |||| | t| d¡p?dƒ  ¡dkdœS)zFBuild analysis text, keyword scan, and optional prompt block for Nova.rhTr#rfrgrL)r?rArqrcÚ prompt_blockÚinclude_in_llm_prompt) rÚ ensure_tableÚget_config_or_defaultsÚboolr8r>r rbrnr) rwr.r/r?ryÚcfgÚuse_customer_onlyrArqrcr{r r rÚprepare_sentiment_analysis¶s"  ÿ ý úr‚Úpreparedc Cs„|dur t|||||d}|d}t|||d|dƒ\}} t| d¡p%dƒ| d<t| d d ¡ƒ| d <t| d ¡p:d ƒ| d <|| fS) z4Return final sentiment label and detection metadata.N)r?ryr?rqrcrAr#Úanalysis_text_lengthrhTrfrg)r‚rvrr8rr ) rwr.r/ror?ryrƒr€r`rar r rÚresolve_call_sentimentÕs& ûür…)r r r r )rr rr rrr r)r"rr r )N)r.r r/r0r r )r?r@rAr r rB)r?r@rAr rcrdr r ) r?r@rorprqr rcr@r rB)NNN) rwr r.r r/r0r?rxryrzr r@)NNNNN)rwr r.r r/r0rorpr?rxryrzrƒrdr rB)Ú__doc__Ú __future__rrÚtypingrrrrrÚsentiment_config_handlerrrr!r-r>rbrnrvr‚r…r r r rÚs0      þ , ý / û"ù