ɫxiGdZddlZddlZddlZddlmZddlmZmZm Z m Z  ddl m Z GddZGddZy#e$rdZ YwxYw) z Credit Card Recommender Engine Provides grounded recommendations based on JSON card data. Optionally uses Gemini (via google-generativeai) to generate more natural chat responses, while still grounding all facts in the JSON data. N)Path)DictListOptionalTupleceZdZddefdZededefdZededeefdZ ededefdZ de d e de fd Z de d e defd Zde d e deefd Zd e de fd Zy)CardRecommendercards_json_pathct|dd5}tj||_ddd|jj dg|_y#1swY+xYw)z%Initialize recommender with card datarzutf-8)encodingNcards)openjsonloaddatagetr)selfr fs 5/var/www/html/chatbot/card-advisor-bot/recommender.py__init__zCardRecommender.__init__sJ /3 9 %Q ! DI %YY]]7B/  % %s AAtextreturncF|xsdjjS)zNormalize text for comparison)lowerstrip)rs rnormalize_textzCardRecommender.normalize_texts  !!#))++cNtj|jddjdd}tjd|}|r&t t |jddzStjd|}|rt|jdSy) u Extract monthly income from text Accepts: "1.5L", "150000", "2 lakh", "2l", "₹2,00,000", "2.5 lakhs" , u₹z+(\d+(?:\.\d+)?)\s*(lakh|lakhs|lac|lacs|l)\biz(\d{2,})N) r rreplaceresearchroundfloatgroupint)rt lakh_match num_matchs rextract_income_inrz"CardRecommender.extract_income_inr!s  * *4 0 8 8c B J J5RU VYYMqQ z//23f<= =IIk1- yq)* *rc,tj|tfddDrytfddDrytfddDrytfd d Dry tfd d DrytfddDryy)z2Detect user's primary card goal from their messagec3&K|]}|v ywN.0wordr+s r z.CardRecommender.detect_goal..;sETtqyE)cashbackz cash backcbr8c3&K|]}|v ywr1r2r3s rr6z.CardRecommender.detect_goal..>sVTtqyVr7)travelflighthotelmileslounger;c3&K|]}|v ywr1r2r3s rr6z.CardRecommender.detect_goal..AsPTtqyPr7)forexzinternational spendmarkupc3&K|]}|v ywr1r2r3s rr6z.CardRecommender.detect_goal..DsKTtqyKr7)tataneu bigbasketcromarDc3&K|]}|v ywr1r2r3s rr6z.CardRecommender.detect_goal..Gs@Ttqy@r7)upirupayzru payrJc3&K|]}|v ywr1r2r3s rr6z.CardRecommender.detect_goal..KsFTtqyFr7)best recommendsuggestunknown)r rany)rr+s @r detect_goalzCardRecommender.detect_goal5s  * *4 0 E%DE E V%UV V P%OP P K%JK K @%?@ @ F%EF Frcarduserc|jd}t|ttfr=|jdijd}t|ttfr||kryy)z9Check if user meets card eligibility (only checks income) monthlyIncome eligibilitymin_monthly_incomeFT)r isinstancer*r()rrRrSmonthly_income min_incomes rcard_eligibility_passz%CardRecommender.card_eligibility_passPsU/2 nsEl 3-4889MNJ*sEl38Srcd}|jdd}|jdd}|jdijd}|jdijd d }|jd ijd ijd d }|jd ijd ijd d } |jd ijd} |dk(r |dk(s|r|dz }|dkr|dz }|dvr|dz}n|dk(rM|dvr|dz }|r|dz }| r|dz }t| ttfr|t dd| dzz z }|dkDr|dz }n|dk(r0d|j |jddvr|dz }|rX|dz }nR|dk(r0d|j |jddvr|dz }|r#|dz }n|s|dk(r|d z }|dkr|dz }|r|d!z }|jd"} t| ttfrR|jd#ijd$} t| ttfr| | z } | d%k\r|d&z }|S| d'k\r|dz }|S)(z&Score a card based on user preferencesggoalrO annual_feer reward_modeltypecashback_benefitsincludedFtravel_benefitsdomestic_loungeinternational_loungeforex_markup_percentr8<i)pointsr> r;)r>ri#'rDnamerrJnetworkrUrVrWg?g?)rrXr*r(maxr)rrRrSscorer]r^ reward_type has_cashback lounge_dom lounge_intlrArYrZratios r score_cardzCardRecommender.score_cardYsxx *XXlA. hh~r266v> xx 3R8<z/CardRecommender.build_reasons..sQUU>15Mr)keyrz% cashback on categoryz spendsrarbz+Cashback-focused rewards on everyday spendsr;rdrez"Access to domestic airport loungesz'Access to international airport loungesrfzForex markup around z% on international spendsrDc3jK|]*}dj|jddvs'|,yw)rDrrN)rr)r4r rs rr6z0CardRecommender.build_reasons..s/ZqVt/B/B155UWCX/Y%YZs(33N currency_namerewardsz % back in z on Tata brandsrJz2RuPay network (UPI-ready if enabled by the issuer)r`zRewards structure focused on uAnnual fee around ₹r! spend_based thresholdperiodyearu/Annual fee can be waived on spends of about ₹z per u"Designed for income from about ₹z per monthzWorks well for: z, rs) rrtappendrXr*r(nextjoinsetadd)rrRrSreasonsr]r_r;rVr}r^r~rZrules best_rulerxryrA tata_rulecurrencyrvrrseenunique_reasonsr s` r build_reasonszCardRecommender.build_reasonssg  xx *xx39r +R06Bhh}b17R 88J+1rXXlA.3! XXlB/52  __%9:  :  $$%5r:@bE+MN )N";!>zJJ **%;R@DDZPKCDHIJJ56E%#u.!5eW >>& !] 2"{3I^^Hf5F)c5\2!PQZ[\P]]bcibjkl  NN? 1~ZX Y  NN-dii! .E-FG Hu$& )A} %%a( ) bq!!rcx|jDcgc]}|j||s|}}|r|n |j}d}td}|D]}|j||}||kDs|}|} |j ||} d} |j d} |st | ttfrd| dd} || | dScc}w)z` Recommend the single best card for user Returns: {card, reasons, note} Nz-infrUuNote: based on ₹r!z5/month, this is the closest fit from available cards.)rRrnote)rr[r(r{rrrXr*) rrSceligiblepool best_card best_scorerRrurrrYs rrecommend_one_cardz"CardRecommender.recommend_one_cards $zzQ!T-G-G4-PAQQ#x 6]  !DOOD$/Ez!"   !$$Y5/2J~U|D'q'99noD  -Rs B7B7N)z cards.json)__name__ __module__ __qualname__strr staticmethodrrr*r.rQrboolr[r(r{rrrr2rrr r s00 ,S,S,,#&##4$dt@t@4@E@DN"$N"dN"tCyN"`  t    rr c eZdZdZdZdedeedeedefdZ dedefd Z dedeedeedeefd Z d Z d edefd Z y)ConversationSessionz5Manages conversation state for asking max 2 questionschd|_d|_i|_t|_d|_d|_y)NidlerF)stepaskedrSr recommender_llm_configuredfinishedrs rrzConversationSession.__init__s1   *,$ rrRrrrcJd|jdd|jdd|jdd|jdd d g}|r|jd |d jd |D}dtjj|d|dS)zGBuild a grounded prompt for the LLM to create a natural-sounding reply.zName: rozIssuer: issuerz Network: rpuAnnual fee: ₹r^rr!zNote:  c3&K|] }d| yw)- Nr2)r4r s rr6z8ConversationSession._build_llm_prompt..$s ;a2aS ;szYou are a friendly, concise credit card assistant inside a chat UI. ONLY use the factual details given below about the recommended card. Do NOT invent benefits, fees, limits, or numbers that are not listed. Do not use emojis. Recommended card details: z' Grounded reasons for recommendation: u Write a natural chat reply in a warm, helpful tone: - Start with: Best pick: ** - Then explain 2–3 key benefits, using or paraphrasing the reasons above - If there is a note, include it in a separate friendly sentence. - You may add 1 short closing sentence inviting the user to ask another question. Use simple markdown where helpful (like *bold* for the card name or key phrases). Do not add any extra factual details beyond what is provided.)rrroslinesep)rrRrr card_lines reasons_texts r_build_llm_promptz%ConversationSession._build_llm_promptsTXXf%& 'txx)* ++, -dhh|Q7: ;     tfo .yy ;7 ;;  *zzz*+,5nL L r base_textcTt|Stjd}|s|S |jstj|d|_tj d}d|d}|j |}t|dd}|s|S|jS#t$r|cYSwxYw) z Ask the LLM to rephrase a simple system-style reply into a more natural chat response, while keeping meaning the same. Falls back to the original text on any error or if LLM is unavailable. NGEMINI_API_KEYapi_keyTgemini-flash-latestzRewrite the following assistant reply so it sounds like a natural, friendly chat message in 1-3 short sentences. Keep the meaning and any important numbers the same. Do not add new facts or emojis. Here is the reply: ""r) genairgetenvr configureGenerativeModelgenerate_contentgetattrr Exception)rrrmodelpromptresponsers r _llm_rephrasez!ConversationSession._llm_rephrase7s = )),-  ''0'+$))*?@EKr#  --f5H8VT2D  ::<    sA%B B B'&B'cdtytjd}|sy |jstj|d|_tj d}|j |||}|j|}t|dd}|sy|jS#t$rYywxYw)z If Gemini is available and configured via GEMINI_API_KEY, ask it to craft the final chat response. Falls back to None on any error. NrrTrr) rrrrrrrrrrr) rrRrrrrrrrs r_maybe_llm_responsez'ConversationSession._maybe_llm_responseXs =)),- ''0'+$))*?@E++D'4@F--f5H8VT2D::<   sA1B#B## B/.B/c<d|_d|_i|_d|_y)zReset conversation staterrFN)rrrSrrs rresetzConversationSession.resetts    rrc<tj|}|dvr#|jd}|j|S|jdk(r*tj ||j d<d|_nn|jdk(r_tj|}|r5||j d<d|_|dkr-d |_d }|j|Sd }|j|S|jdk(rtj |}|j jdr|j jdd k(r||j d<|j jdd k(r>|jd kr/|xjdz c_ d|_d}|j|S|j jddv}|rmt|j jdttfs>|jd kr/|xjdz c_ d|_d}|j|S|jj|j }|d}|d} |d} |j!|| | } | rd|_d |_| Sd|ddg} | r| j#| | D]} | j#d| | j#d| j#dd|_d |_dj%| Sd}|j|S)zs Process incoming message and return bot response Asks max 2 questions before recommending )rrestartz start overz`Reset done. Tell me what you want: cashback, travel, Tata/Neu, RuPay/UPI, or 'best card for me'.ask_goalr]r ask_incomerUrmTz7Income is too low for the cards available in this demo.zGPlease share your *monthly income in INR* (for example, 80000 or 1.2L).rOrr#uVQuick one: what do you want most — *cashback* or *travel perks* (lounges and miles)?)r;rOrDrJr8zTWhat is your *monthly income in INR*? You can just send a number like 80000 or 1.5L.rRrrz Best pick: *ro*rrzXIf you want, tell me your typical spend split (online/offline/travel) and I'll re-check.rzjTell me whether you care more about cashback, travel, Tata/Neu, RuPay/UPI, or just say 'best card for me'.)r rrrrrQrSr.rrrrXr*r(rrrrr)rr normalizedbaseincome inferred_goal needs_incomeresultrRrr llm_answerlinesreasons rprocess_messagez#ConversationSession.process_message{s %33D9  ; ; JJLuD%%d+ + 99 " / ; ;D ADIIf DI YY, &$77=F-3 /*" E>$(DMTD--d33`))$// 99 +77=M99==(DIIMM&,AY,N$1 &!yy}}V$ 1::>JJ!OJ *DIsD--d33 99==04ffLJtyy}}_/MPSUZ|$\::>JJ!OJ ,DIqD--d33%%88CF&>DY'G&>D11$FJ" $ !!$DL>34E T"! , r&]+ , LL  LLs tDI DM99U# #{!!$''rN)rrr__doc__rrrrrrrrrrr2rrrrs? d T#Y hsm X[ <ssBtCyQT ZbcfZg8R(CR(CR(rr)rrrr%pathlibrtypingrrrrgoogle.generativeai generativeair ImportErrorr rr2rrrsV ..' v v r(({ Es=AA