tjIUdZddlmZddlZddlmZddlmZedZddd d d d d ddddddddZ de d<ej dZ ej dZ ej dZej dZej dZej dZej dZej dZej dZej d Zej d!Zej d"Zej d#Zej d$Zej d%Zej d&Zej d'Zej d(Zej d)Zej d*Zej d+Z ej d,Z!ej d-Z"hd.Z#dDd3Z$dDd4Z%dDd5Z&dDd6Z'dDd7Z(dDd8Z)dDd9Z*dDd:Z+dDd;Z,dDd<Z-dDd=Z.dDd>Z/dDd?Z0dEdAZ1dFdCZ2dS)Gz;Classify finance-assistant questions into response intents.) annotationsN)Literal)is_pdf_overview_question)statement_summaryspending_summarymerchant_analysiscategory_spending date_lookupincometransaction_search comparisoninsightsfinancial_advice document_qaperiod_coverage payment_countgeneralzStatement SummaryzSpending SummaryzMerchant AnalysiszCategory Spendingz Date LookupIncomezTransaction Search ComparisonInsightszFinancial Advicez Document Q&AzPeriod Coveragez Payment CountGeneralzdict[str, str]FINANCE_INTENT_LABELSzo(?i)\b(grocer(?:y|ies)|food|taxi|cab|uber|ola|bills?|recharge|transfer|zepto|blinkit|swiggy|zomato|instamart)\bzd(?i)\b(merchant|paid\s+to|sent\s+to|who\s+did\s+i\s+pay|payments?\s+to|upi\s+to|money\s+sent\s+to)\bzV(? aanmycabdaytheupibilldatefoodlasttaxithatthisweekyearbillsmoneymonthtotalaccountaveragegrocerypaymentpaymentsrechargetransfer groceries statement transfers transaction transactionsquestionstrreturnboolcTtt|pdSN)r<_FINANCIAL_ADVICE_REsearchr9s 3/var/www/html/CA-Chatbot/app/saas/finance_intent.pyis_financial_advice_questionrDs# $++HN;; < <<cN|pd}|sdSt|rdSt|}t |dkrt jd|rdSt |dkrt jd|rdSdS)Nr?FTz4(?i)\b(only|not|why|contain|include|cover|missing)\bzL(?i)\b(only\s+contain|only\s+includes?|why\s+only|not\s+august|not\s+july)\b)strip_PERIOD_COVERAGE_RErA_MONTH_WORD_REfindalllenre)r9qmonthss rCis_period_coverage_questionrQs R  A u!!!$$t  # #A & &F 6{{aBI&]_`aat 6{{aBIW t 5rEcTtt|pdSr>)r<_PAYMENT_LIST_RErArBs rCis_payment_list_questionrTs#  ''B77 8 88rEc|pd}tjd|sdSt|rdStjd|rdSt|rtjd|sdSdS)uNTrue for analyse/summarize the document — not spend totals or payment lists.r?z-(?i)\b(analy[sz]e|overview|summar(?:y|ize))\bFz8(?i)\b(how\s+much|how\s+many|list\s+all|total\s+spent)\bz(?i)\banaly[sz]e\bT)rIrNrArT _SPEND_REr9rOs rCis_analyse_overview_askrXs R  A 9Eq I Iu""u yLaPPu29-BA#F#Fu 4rEcTtt|pdSr>)r<_PAYMENT_SINGULAR_CLARIFY_RErArBs rC is_vague_single_payment_questionr[ s# ,33HNCC D DDrEcTtt|pdSr>)r<_PAYMENT_COUNT_RErArBs rCis_payment_count_questionr^$s# !((R88 9 99rEc|pd}t|rdStjd|rtjd|rdSdS)Nr?Tz(?i)\b(?:bank\s+)?statement\bz3(?i)\b(?:my\s+name|account\s+holder|who\s+am\s+i)\bF)_STATEMENT_IDENTITY_RErArNrWs rCis_statement_identity_questionra(s_BA$$Q''t y1155")>;;t 5rEcTtt|pdSr>)r<_PROFIT_LOSS_RErArBs rCis_profit_loss_questionrd4s" &&x~266 7 77rEcTtt|pdS)uETrue for ranked expense lists — not a restatement of header totals.r?)r<_BIGGEST_EXPENSES_RErArBs rCis_biggest_expenses_questionrg8s# $++HN;; < <r:: ; ;;rEc|pd}t|rdSttjd|ptjd|S)Nr?Tz9(?i)\bwhat\s+(?:is|was)\s+the\s+(?:statement\s+)?period\bz(?i)\bstatement\s+period\b)_STATEMENT_PERIOD_RErAr<rNrWs rCis_statement_period_questionrmBs^BA""1%%t  NPQRR 7 92A 6 6  rEcn|pd}|sdSt|rdSt|}|sdSt jdd|d}|r|dvrdS|tvrdSdS)zRTrue when the user asks about someone else's spend (not their uploaded statement).r?F\s+ rH>iweyouanyoneanybodysomeoneT) rI_FIRST_PERSON_RErA_THIRD_PARTY_SPEND_RErNsubgrouplower_MERCHANT_STOP)r9rOmsubjects rC!is_unrelated_party_spend_questionrLs R  A uq!!u$$Q''A ufVS!''!**--3355;;==G g!SSSu.  u 4rEcZ|pd}|sdStjd|rjt|dkrEt |s+t |st|sdSt|}|dvS)zUTrue when the question fully specifies its own finance ask (not a pronoun follow-up).r?Fz6(?i)\b(it|that|this|those|these|them|previous|above)\b > r rr rrrr rrr ) rIrNrArMsplit_DATE_RE _CATEGORY_REextract_merchant_queryclassify_finance_intent)r9rOintents rCis_analytical_self_containedr^s R  A u yJANNSVWXW^W^W`W`SaSaegSgSgq!! ,*=*=a*@*@ I_`aIbIb 5 $Q ' 'F    rE str | Nonec |pd}|sdSgd}|D]k}tj||}|s|dd}tjdd|}dtjd|D}|rN|d d vr2||r|d d v2|std |Dr|d tvrt|dkr-d |}t|dkri|cSdS)zRPull a merchant/payee name from 'spent on X' / 'paid to X' / 'pay Nykaa' phrasing.r?N)z(?i)spend(?:ing)?\s+(?:this\s+much\s+|so\s+much\s+|too\s+much\s+)?on\s+([A-Za-z0-9][A-Za-z0-9&.'\-]{1,40}(?:\s+[A-Za-z0-9&.'\-]{1,30}){0,3})zq(?i)(?:spent|spend|paid|pay|payment)\s+on\s+([A-Za-z0-9][A-Za-z0-9&.'\-]{1,40}(?:\s+[A-Za-z0-9&.'\-]{1,30}){0,3})z(?i)(?:cut|reduce|trim|save)\s+(?:back\s+)?on\s+(?:my\s+)?([A-Za-z0-9][A-Za-z0-9&.'\-]{1,40}(?:\s+[A-Za-z0-9&.'\-]{1,30}){0,3})z(?i)(?:how\s+much\s+(?:did\s+i\s+)?)?(?:paid|pay|sent|send)\s+(?:to\s+)?([A-Za-z0-9][A-Za-z0-9&.'\-]{1,40}(?:\s+[A-Za-z0-9&.'\-]{1,30}){0,3})z](?i)money\s+sent\s+to\s+([A-Za-z0-9][A-Za-z0-9&.'\-]{1,40}(?:\s+[A-Za-z0-9&.'\-]{1,30}){0,3})zZ(?i)payments?\s+to\s+([A-Za-z0-9][A-Za-z0-9&.'\-]{1,40}(?:\s+[A-Za-z0-9&.'\-]{1,30}){0,3})z<(?i)\b(?:on|at)\s+([A-Za-z][A-Za-z0-9&.'\-]{2,40})\s*[?.!]*$rHz ?.,!;:"'zT(?i)\s+(last|this|previous|in|on|during|for)\s+(month|year|week|july|august|june).*$cg|]}||Sr.0ts rC z*extract_merchant_query..s888a8!888rEro>r#r&r'r(r+r,pleasec3LK|]}|tvV dS)N)r{r|rs rC z)extract_merchant_query..s/;;qqwwyyN*;;;;;;rErrprG) rIrNrArzryrr{popallr|rMjoin)r9rOpatternspatr}rawtokensnames rCrrvs R  A t   H Ic1    ggajj|,,f c     %'' 98RXfc22888 ))++0    JJLLL ))++0      ;;F;;; ; ;   !9??   . .3v;;!3C3C xx%%'' t99q==  4rE FinanceIntentc|pd}|sdStjdd|}t|rdSt|rdSt |rdSt|rdSt|rdSt|st|rdSt|rdSt|r1t|stjd |rd St|rd St|st|rdSt |rtjd |rd St|rPt"|s4t$|st&|rd Stjd|rtjd|rd St(|r2t"|sd|vrdSt-|}|rPt"|s4t.|st0|rdSt.|r6t"|st2|rdSt0|r1t"|stjd|rdSt4|stjd|rt"|rd St$|rdSt7|rdSt"|rdSt8|rdSt:|rdSt&|rdStjd|rdStjd|rtjd|sdSdS)Nr?rz(?i)\b,\s*ade\bz maderrrrz[(?i)\b\d{1,2}(?:st|nd|rd|th)?\s+(?:jan|feb|mar|apr|may|jun|jul|aug|sep|oct|nov|dec)[a-z]*\br rz(?i)\b(spend|spent|paid|payment|received|merchant|statement|category|month|july|august|june|total|outflow|inflow|this\s+pdf|uploaded)\br z(?i)\b\d{1,2}(?:st|nd|rd|th)?\s+(?:jan|feb|mar|apr|may|jun|jul|aug|sep|oct|nov|dec)[a-z]*\b|(?:jan|feb|mar|apr|may|jun|jul|aug|sep|oct|nov|dec)[a-z]*\s+\d{1,2}\bzN(?i)\b(payments?|transactions?|spent|spend|paid|pay|activity|outflow|inflow)\bzon r rz (?i)\b(paid|pay|sent|send|upi)\bz,(?i)\b(last|previous|this|current)\s+month\br r rz(?i)^\s*what\s+is\s+a\s+pdf\bz5(?i)\b(my\s+)?(pdf|document|statement|invoice|page)\bz(?i)^\s*what\s+is\s+(?:a|an)\s+r)rIrNryrr@rArQrdrgrarmrjr^rrrX _COMPARE_RErV _INCOME_RE _SEARCH_RErr{r _MERCHANT_RE_MERCHANT_PROPER_RErwrKrT _INSIGHT_RE _SUMMARY_RE)r9rOmerchants rCrrs R  A y !7A..A)++y""1%%"!!"1%%!  q!!z#A&&z%a((#,H,K,K#"""1%%#"" ## 9 K   } ##""#&=a&@&@#""!  ""|qy//22j6G6G6J6JjN_N_`aNbNb} y R   )Y    }1#9#3#3A#6#6#%17799:L:L""%a((H#Y%%a((#L,?,?,B,B#FYF`F`abFcFc#""1#9#3#3A#6#6#:J:Q:QRS:T:T#""!!!$$#)*:*:1*=*=#KnpqArAr#"" a   9Da H H   1  }x""$##"!!!z!#""$## y1155y yI1MMVXV_*AWW} 9rE)r9r:r;r<)r9r:r;r)r9r:r;r)3__doc__ __future__rrNtypingr app.pdf.calcrrr__annotations__compilerrrrrVrrrrr@rfrrKrJrSrZr]rir`rcrlrwrxr|rDrQrTrXr[r^rardrgrjrmrrrrrrErCrsOAAA"""""" 111111  $-*,, .*!($))"rz9 rz  !bj] 2:    BJ    RZR bj4 bj[ bjT "rz    "rz RZ  ]!bj    2:     *rz*LMMBJ    !bj   $    "*   "rz 2:- # " !!!H===="9999    EEEE::::    8888==== <<<< $02222j[[[[[[rE