Ljh 8dZddlmZddlZddlZddlZddlZddlZddlmZm Z ddl m Z ddl Z ddlmZddlmZmZmZmZddlmZdd lmZdd lmZdd lmZdd lmZddlZej@jCej@jEe#Z$eej@jKe$d ejLddZ'ejLddZ(dZ)dZ*ddddZ+dZ,e-ejLddZ.ej@jKe$dZ/ej`ejbdejddZ3ejhd ejjZ6d!Z7ejpe9ej@jKe/d"Dcic]}|d#| c}Z:e jpej@jKe/d$Z;e;d%Z|d'c}Z?e=Dcgc]}e:e>|jd(d)c}ZAejhd*ZBd+ZCd,ZDeDZEdfd-ZFd.ZGd/d0d1id0d2id1d3d4d1d3d4d5d0d6id3d7d8d9d:gd;ZHed<=ZIejbjd>d?ZJejbjd@d?ZKed?dABfdgdCZLGdDdEeZMGdFdGeZNGdHdIeZOdJZPdKZQdLZRdde*fdMZSdNZTdOZUeIjeJdPdQZVeIjeJdReeLgSdhdTZXeIjeJdUeeLgSdidVZYeIjeJdWeeLgSdXZZeIjeJdYeej@jKe$dZ[dZ\eIjeJxsd]d^Z\eIjeJd_d`Z]e^dak(rddl_Z_e_jdbdcddeyycc}wcc}wcc}w)juu QuitSure AI Support — API service. One stateless endpoint the app frontend calls. Given a user message (+ optional history), returns an answer grounded ONLY in the curated knowledge base, plus an escalation decision so the devs know when to hand the chat to a human coach (in Freshchat). Never connects to Freshchat. Never learns from coach transcripts (Ram's call). Design rationale: docs/DESIGN.md Frontend contract: docs/API.md Run: python3 scripts/build_kb.py && python3 scripts/build_index.py # one-time / on KB change uvicorn app:app --port 8082 A simple test chat UI is served at http://localhost:8082/ ) annotationsN)datetimetimezone)Optional) load_dotenv)DependsFastAPIHeader HTTPException) FileResponse) StaticFiles) BaseModel)genai)typesz.env GEMINI_MODELzgemini-2.5-flashGEMINI_EMBED_MODELzgemini-embedding-001izDQuitSure Original, the 6-day intensive program (users quit on Day 6)zCQuitSure Relaxed, the 42-day gradual program (users quit on Day 42)zLeansure, the fitness program)P3P9P11c6|sy|j}|dvr|Sy)aeThe KB tags smoking content as either Original (P3) or Relaxed (P9). P9/LTP is the ONLY 42-day Relaxed program; every other smoking program (P1, P2, P4, P5, P7...) is a 6-day Original-style program, so it maps to P3 for both retrieval filtering and naming. P11 is Leansure (separate). Keeps a P1/P5 user from seeing none of the Original answers.N)rrr)upper)codecs '/var/www/python/quitsure-support/app.pycanonical_programr7s%  AM RETRIEVAL_FLOORz0.55dataGEMINI_API_KEYi N)timeout)api_key http_optionsuosuicid|kill(?:ing)? myself|end(?:ing)? my life|hurt(?:ing)? myself|harm(?:ing)? myself|self.?harm|want to die|wanna die|don.?t want to live|no reason to live|end it all|take my (?:own )?life|khud ?kushi|khud.?khushi|aatmahatya|atmahatya|marna chaht|mar jana chaht|jeena nahi|jina nahi|jeene ka mann nahi|mujhe (?:ab )?jeena nahi|ab nahi jeena|khud ?ko ?khatam|khud ?ko ?maar|khud ?ko ?nuksan|jaan de ?d|jaan deni|zindagi khatam|आत्महत्या|खुदकुशी|मरना चाहत|जीना नहीं|खुद को खत्म|खुद को मार|जान दे|जान दूंctjd|tjryttj|S)Nz"smok\w*\s+(?:to|and)\s+hurt myselfF)researchIbool_CRISIStexts r is_crisisr,_s0 yy6bddC t$ %%rzkb.jsonidz index.npzvectorsidsprogramsplatformallz [a-z0-9]+cHtj|jSN) _TOKEN_REfindalllowerr*s r_tokr8us   TZZ\ **rc 4dtDDcgc]F}tdj|d|jdd|jdd|dgH}}t |}t j |Dcgc] }t |c}t j}|rt|jnd }i}|D]*}t|D]}|j|d d z||<,|jDcic]*\}}|tjd ||z d z|d zz z,} }}|D cgc]} i} } t|D]*\} }|D] }| | j|d d z| | |<",| | ||d Scc}wcc}wcc}}wcc} w)Nc3@K|]}tt|ywr4)KBint).0is r z_build_bm25..{s4QbQj4s questiontopictagsanswerdtyperg?)tfidfdlavgdl) INDEX_IDSr8joingetlennparrayfloat32floatmeansetitemsmathlog enumerate) edocsndrLrMdftrrK_rJjs r _build_bm25rdys4)4 6 1Z=!%%*OQRS[Q\]^ _ 6D 6 D A 4(a3q6( ;B !E"'') sE B %Q %AFF1aL1$BqE %%CE((* M$!Q1dhhqAECKAG445 5 MC M " B $+1 +A!uyyA*BqE!H ++SU ;; 6) N sA F;F /F; Fctt|}tdtdtdtdf\}}}}tjt |tj }tt |D]c} |d|z ||r|| |z ndzzz} d} |D]=} || j| } | s| |j| d| |dzzz| | zz z } ?| || <e|S)zOOkapi BM25 score of the query against every indexed doc (aligned to INDEX_IDS).rJrKrLrMrFrIrH) rWr8_BM25rRzerosrQrTrangerP)queryk1bqrJrKrLrMscoresrcnormsrafs r bm25_scoresrqs DKAteElE$KwOBR XXc"gRZZ 0F 3r7^QUQ5"Q%%-cBBC  CA1 ! ASWWQ_R!V 5TBB Cq  MraCYou are the QuitSure Assistant, an automated support assistant inside the QuitSure smoking-cessation app. You are NOT a human and NOT a named coach, never claim to be. Never output placeholders or merge-fields (e.g. %name%, ____). Never use em-dashes or en-dashes in your replies. Use commas, periods, or parentheses instead. Your ONLY job is to help with: the QuitSure app and programs, the user's quit-smoking journey, and their subscription / billing / account. Nothing else. TONE: Be warm, encouraging, and human, like a caring quit-smoking coach who is genuinely on the user's side. Acknowledge how they feel, celebrate their effort, and be patient and reassuring. Sound friendly and natural, never robotic, curt, or clinical. You may rephrase for warmth, but never add any fact, step, price, or policy that is not in the approved answers. Even for simple or factual questions (how to log out, gift cards, a tech step), keep a warm human touch, a short kind opener or closer. Never give a cold one-line answer. LANGUAGE: Reply in the SAME language and script the user writes in. If they write in Hinglish (romanized Hindi mixed with English, in Latin script), reply in that same natural romanized Hinglish style, NOT in pure English and NOT in Devanagari Hindi. If they write in Devanagari Hindi, reply in Hindi. Keep the exact approved facts, steps, and links, just expressed in their language. Answer ONLY from the APPROVED ANSWERS provided below. Match their warm, supportive tone. Never invent facts, prices, steps, or policies that are not in the approved answers. Decide in this order: 1. OFF-TOPIC / time-pass: if the message is NOT about QuitSure, quitting smoking, the app, or the user's account (e.g. general knowledge, news, sports, math, coding, jokes, weather, recipes, casual chit-chat), do NOT answer it. Give a brief, friendly one-line redirect to what you can help with. -> escalate=false, escalate_reason="out_of_scope". Do NOT send these to a coach. 2. Otherwise set escalate=true when ANY of these is true: - It IS about QuitSure but the approved answers don't contain the info -> "no_confident_answer" - Medical/clinical questions -> "medical_topic", UNLESS the approved answers directly cover the question, in which case answer from that approved content (e.g. an approved answer about using Champix or nicotine substitutes with the program). Escalate a medical question only when the approved answers do NOT cover it (e.g. personal dosing or a drug interaction we have no answer for). Normal quit-smoking experiences (irritability, cravings, sleep, mood, appetite) are not medical and may be answered warmly. - Self-harm, suicidal thoughts, or crisis -> "crisis" - The user explicitly asks for a human / coach / agent -> "user_requested" - The user is clearly frustrated or angry -> "user_frustrated" - Needs a human to look up or fix THIS user's specific account state: charged but subscription not active, "I paid but no access", "where is my refund" / refund requested but not received, a double charge -> "account_action". (General billing questions, and how to REQUEST a refund or "I want a refund", are ANSWERED from the approved refund process, not escalated. Only a refund STATUS problem that needs a human to check the account is escalated.) 3. Otherwise, answer directly from the approved answers. ANSWER BEFORE YOU ESCALATE: if the approved answers contain concrete steps that would resolve the user's issue, GIVE those steps rather than escalating, even if the message also sounds discouraged, upset, or emotional. Many complaints that sound emotional are really technical/account issues with a fix in the approved answers (e.g. "I lost all my progress", "my days reset", "everything is gone", "it logged me out") -> give the recovery steps first (you may briefly acknowledge the feeling), and only escalate if the steps do not cover their situation. Questions about how QuitSure works, how effective it is, or how it compares to willpower, nicotine patches, or other methods ARE covered by the approved answers (how it works, how it helps, how it is different). Answer these from that content, do not escalate them. CONTINUITY: Use the whole conversation, including your own previous replies. If the user sends a short reply that continues the current topic (e.g. "yes", "sure", "help me", "ok", "tell me more", "which one") or accepts an offer you just made, keep helping using what has already been discussed and the approved answers. Do NOT escalate a simple continuation, and never offer help and then hand off instead of giving it. Only escalate when the user genuinely needs information not covered. If the user asks which option or program suits them or asks you to help them decide, HELP them like a coach: briefly ask about the concrete factors that actually decide it (how much time they can give each day, how soon they want to quit), then RECOMMEND the program that fits and say why, using the approved answers. Do not just ask them to pick a pace or bounce the choice back to them. That is grounded help, not a reason to escalate. MULTI-PART QUESTIONS: If the user asks several things at once, answer every part you can from the approved answers. Escalate ONLY if a specific part genuinely needs a human (an account lookup or a medical topic), and even then, answer the other parts first and tell the user which one part you are passing to the team. Never escalate the whole message just because one part needs a human, and never invent an answer for a part the approved answers do not cover. USER CONTEXT: If a USER CONTEXT block is provided (the user's program, current day, subscription status), treat those facts as true and use them to answer questions about the user's own account or progress directly, e.g. "am I subscribed" -> use subscription status, "what day am I on" -> use the day. Only escalate account PROBLEMS that need a human to fix or look up (a specific charge, a refund to process, a payment not reflecting), not simple status questions you already have the answer to. IMPORTANT: Whatever the approved answers cover, you SHOULD answer, including medical questions the approved answers address (e.g. using Champix or nicotine substitutes with the program, common symptoms the doc explains). Escalate only what the approved answers do NOT cover, plus these always-escalate cases even if related content exists: self-harm or crisis, an explicit request for a human, and account-specific payment/refund problems for this user. If the user greets you (hi/hello), thanks you, or makes small talk, reply warmly in one line as the QuitSure Assistant and invite their question, and set escalate=false with escalate_reason "greeting". Do NOT mention coaches or escalation unprompted, only at the moment you actually need to hand off. When escalating: a short warm handoff line as "answer" (e.g. "Let me get a member of our support team to help you with this.") plus a one-sentence "coach_summary". For "crisis", gently encourage immediate help and that someone from the team is coming. In "used_ids", list the id number(s) of the APPROVED ANSWERS you actually drew on to write your answer (from the "[id N | ...]" labels). Leave it empty for greetings, off-topic declines, and escalations where you used none. Respond ONLY as JSON: {"answer": str, "escalate": bool, "escalate_reason": str|null, "coach_summary": str|null, "used_ids": int[]} objecttypestringbooleanT)rsnullablerSinteger)rsrXrv)rEescalateescalate_reason coach_summaryused_idsrErx)rs propertiesrequiredzQuitSure AI Support)title MOUNT_PREFIXrCAPI_KEYz X-API-Key)defaultaliasc>tr|tk7r tddyy)Nizinvalid or missing API key) status_codedetail)rr ) x_api_keys r require_keyrs!9'4PQQ(wrc"eZdZUded<ded<y)Turnstrroler+N)__name__ __module__ __qualname____annotations__rrrrs I IrrczeZdZUdZded<ded<gZded<dZded<dZded <dZded <dZ d ed <dZ ded <y) ChatRequestN Optional[str]user_idrmessagez list[Turn]historyprogramr1localez Optional[int] program_daysubscription_status) rrrrrrrr1rrrrrrrr"sN!G]! LGZ!G]!"Hm" FM !%K%)--rrcVeZdZUdZded<dZded<ded<ded<ded<dZded <y) FeedbackNrr message_idrrrEratingnote)rrrrrrrrrrrr-s/!G]! $J $ L K KD-rrcdtjtjj Sr4)rnowrutc isoformatrrr_nowr6s << % / / 11rcJtjjt|gt j dt }tj|jdjtj}tjj|}|r||z S|S)NRETRIEVAL_QUERY) task_typeoutput_dimensionalitymodelcontentsconfigrrF)clientmodels embed_content EMBED_MODELrEmbedContentConfigDIMrRasarray embeddingsvaluesrTlinalgrn)r+respvr^s r embed_queryr:s} == & &''2C[^_ ' D 4??1%,,BJJ?A qA1q51rc<|jDcgc]}|jdk(s|j c}dd}|jDcgc]!}|jdk(s|jdd#c}dd}dj||z|jgzScc}wcc}w)z`Anchor retrieval to the recent context so short follow-ups ('sure', 'which one?') retrieve well.userN assistantr@)rrr+rOr)reqra recent_userlast_assistants rretrieval_queryrEs#&;;Ca!&&F2B166CBCHK,/KKQq166[;PaffTclQRTRUVN 88K.0CKK=@ AADQsBBBBct|}t|}t|zg}tt D]G}|rdt |vr |t |vr|rt |dk7r t ||k7r7|j|I|sgdfSt|t|fd}tfd|Dfd}d} |Dcic]}|dc}t|D]\} }|xxd| | zz z cc<t|D]\} }|xxd| | zz z cc<t|fd d|} | Dcgc])}ttt|t|f+} }ttfd |D} | | fScc}wcc}w) Nr2rHc| Sr4r)r>simss rzretrieve..^s d1gXr)keyc34K|]}|dkDs |yw)rNr)r=r>bms rr?zretrieve.._s6ABqEAIa6s c| Sr4r)r>rs rrzretrieve.._s r!ufr<g?c| Sr4r)r>fuseds rrzretrieve..fs q rc3(K|] }| ywr4r)r=r>rs rr?zretrieve..js.T!W.s)rr INDEX_VECSrhrQINDEX_PROGRAMSINDEX_PLATFORMappendrqsortedr[r;r<rNrUmax)rirr1kqvallowedr>cos_rankbm_rankrrf_krtophitsbestrrrs @@@rretrieverLs(G U B ?DG 3t9  uN1$55'XYIZ:Z  q)U2~a7HH7T q  3w U Bg#56H66.osR Aq qwis1W:485VAjM?%PQRZP[}]s.0)rO)rs r build_contextrns ;; rctjjdd|d<|jdsd|d<n|jddvrdnd |d<|jd r |d dd |d <|S) zSAdd a message id (for feedback linking) and a suggested channel; cap coach_summary.N rrxsuggested_channelry)account_action service_error technicalcoachingrzi,)uuiduuid4hexrP)outs r _finalizerus (("-C  77: #'  3677;L3MQt3tK)3   ww"?3DS9O Jrz/healthc:dttttdS)Nok)status kb_entriesrretrieval_floor)rQr; ANSWER_MODELrrrrhealthrs#b'L. 00rz/chat) dependenciesc t|jrutdddd|jgd}tj|dt |j |j|j|jddd|d  |S |jxsd }t|}t|||j \}}|jr:t|j||j \}}||kDr|||j}}}g}|jr|jd |jdtjt!||} |jd| d|j"|jd|j"d|j$r|jd|j$d|rddj'|zdznd} dj'd|jddDd|jz} | dt)|d| } t*j,j/t0| t3j4t6dt8d !} t;j<| j>}|jAd"d|jAd#d|jd"d$k(rd%|d&<d'}|d& xr|jd"|v}|r.|tBd(zkr"|jEdd)d*d+|j,|d&s|jd"|vr|Dcic]\}}|tBkDs|d-|}}}|jd.xsgDcgc] }||vs||}}|s%|r#tGtI|jKg}|Dcgc]}|d/d0|d1c}|d2<ng|d2<t|}tj|dt |j |j|j|j||d3tM|d&|d"|d 4 |Scc}}wcc}wcc}w#tN$r}td5dd6d7|jgd}tj|dt |j |j|j|jd8dd6|d  |cYd}~Sd}~wwxYw)9NaI'm really glad you reached out, and you don't have to go through this alone. Let me connect you with a member of our team right now. If you need immediate support, please reach out to a helpline. In India you can call Tele-MANAS at 14416 (free, 24x7) or AASRA at +91 9820466726. If you are outside India, please contact your local emergency number or a nearby suicide-prevention helpline. You matter, and help is available.TcrisiszCRISIS / self-harm signal: )rErxryrzsourcesr crisis_filterrIrE) rtsrrr1rgaterxryrEr)rr1zUser's app language: z (reply in this language).zUser's program: zK. An approved answer written for this program by name applies to this user.zUser is on day z of the program.zUser's subscription: .zUSER CONTEXT:  rrCc3pK|].}|jjd|jd0yw)z: rN)rrr+)r=ras rr?zchat..s,RA166<<>*"QVVHB7Rs46izUSER: zAPPROVED ANSWERS: z --- CONVERSATION: zapplication/jsong?)system_instructionresponse_mime_typeresponse_schema temperaturerryrz out_of_scopeFrx)rgreetingg?no_confident_answerz>Let me get a member of our support team to help you with this.z User asked: )rxryrErzr-r{rz / rArr) rrrrr1rrbest_simrrxryrEzWI'm having a bit of trouble right now, so let me get someone from our team to help you.rzService error while handling: error)(r,rrdblog_chatrrrr1rrrrr PROGRAM_NAMESrPrrrrOrrrgenerate_contentrrGenerateContentConfig SYSTEM_RULESRESPONSE_SCHEMAjsonloadsr+ setdefaultrupdatenextiterrr< Exception)rr eff_programrqrr hits_mbest_mctxpnamectx_strconvopromptrexemptansweredr\roabover>pickeds rchatr's$l &*h,G }*Ubdfg s<0TVS[[Z]ZeZe,, /\]!)#h- A Tkk)T S !!"kCLLQh ;;%ckk;QTQ]Q]^NFF %+VS[[h ::szz$9#**E_"`a!!"3K"@+N %eW-+, - ?? & _S__DUUe3f(g  " "CJJ1FsG^G^F__`/a$bCF$tyy~5>BRRSAQRRW]^a^i^i]jUkk9/ d0C/DD\]b\cd}}--..#/#5 / .  jj# ($/ - 77$ % 7#C O .:&S3773D+EV+S ?T#99 JJ6K^'3CKK=%A  C :377+<#=V#K,0HDAqA4GQtWaZHEH),)<)BR1qEzeAhRFRetELLN345JPQQ8 S:@QC NC Nn s<0TVS[[Z]ZeZe,, RZbC O(Q2Q S  A4S?S S z /feedbackctjt|j|j|j |j ddiS)Nrr)r  log_feedbackrrrrr)fbs rfeedbackr+s3OODFBMM2::ryy"''J d rz/statsc*tjSr4)r  get_statsrrrstatsr.s <<>rz/staticstatic) directory)name/c^ttjjtddS)Nr/z index.htmlr ospathrOROOTrrrhomer8s  T8\B CCrz /analyticsc^ttjjtddS)Nr/zanalytics.htmlr4rrr analyticsr:s  T85EF GGr__main__zapp:appz0.0.0.0i)hostport)g?g?)rr)rr)r*r)a__doc__ __future__rrrYr5r%rrrtypingrnumpyrRdotenvrfastapirr r r fastapi.responsesr fastapi.staticfilesr pydanticrgoogler google.genairr r6dirnameabspath__file__r7rOgetenvrrrTOP_KrrrUrDATAClientenviron HttpOptionsrcompiler'r)r,loadopenr;_idxrrNr<rrPrr5r8rdrfrqrrapprrrrrrrrrrrrrpostr'r+r.mountr8r:ruvicornrun)r\r>s00rr[si# ';;*+  wwrwwx01 BGGLLv &'ryy);< bii,.DE    Q O *    "3V<= ww||D&! bjj)9:#45#4#4U#C E "**KDD &$$))DdI)F$GHIQagqjIrwwrww||D+./ )_ K 2; !"0#0 \N%  0D/EFdGdN\N) $GK4H3IJK L>  0D/EFG  \N' "K"'',,tX:V$W^f g DDL> $%H&H zGKK 5E J =Gs' P 1P "P