o +i8G@sddlmZddlmZddlmZddlmZmZm Z ddl Z ddl Z ddl m Z ddlmZddlmZmZed d d Zd d ddde jjdZe ddZde eeffddZdee fddZde fddZdedeefddZe jddZ dZ!e red e d!d"Z!ej"ed#gd#gd#gd$d%d&d'd%d(d'd)d%d*d'd+d,d'd-d%d.d'd%d/d'd0d1d2d'd1d3d'd4d%d5d'd%d6d'd7d8Z#de fd9d:Z$e$Z%Gd;d<dd>eZ'd?e&deefd@dAZ(dBedCeedee)fdDdEZ*edFdGdHZ+edIdJdKZ,e-dLd?e&fdMdNZ.e-dOdPed?e&fdQdRZ/edSdTdUZ0dVZ1e-dWdXe'fdYdZZ2ed[d\d]Z3e4d^kr[ddl5Z5e5j6ed_d`dadSdS)b)FastAPI)CORSMiddleware) BaseModel)ListOptionalDictN)contextmanager) ChatAnthropic) HumanMessage SystemMessagez PsyMed AI APIz0.4.0)titleversion localhostrootz fastcoin12*psychological_hubutf8mb4)hostuserpassworddatabasecharset cursorclassccs2tjdit}z |VW|dS|w)N)pymysqlconnect DB_CONFIGclose)connrr/var/www/html/psymed-ai/main.pyget_db_connections rreturnc CstF}|1}|di}|D]}|dr|dn|d}|||d<q|WdWdS1srCu1삼분기 심장기형 위험r0u*체중증가로 혈당조절 악화 가능)r>rLuEbstein 기형 위험u 중증 신기능저하시 금기)r>rD preferredu+임신중 안전성 데이터 가장 풍부u(심근경색 후 안전 (SADHART 연구))r>rGu부정맥 위험u녹내장 악화)rGrP) Valproate ParoxetineLithium SertralineTCAc Csz1t}|r/it}|D]\}}||vri||<|||qtdt|d|WSWtStyK}ztd|WYd}~tSd}~ww)u1DB에서 규칙 로드, 실패시 fallback 사용u DB에서 u개 약물 규칙 로드됨u#DB 로드 실패, fallback 사용: N)r;FALLBACK_RULESitemsupdater6lenr5)db_rulesmergeddrug conditionsr:rrrget_drug_ruless" rpc@seZdZUdZeed<dZeed<dZe ed<dZ e eed<dZ e ed <d Z eed <d Zeed <dZe ed<dZe ed<dZe ed<dZe ed<dZe ed<dZe ed<dZe ed<gZeeed<dS) PatientInforY diagnosisageFr>N trimesterr@none liver_diseasenormalrenal_functionrLrGrNhypothyroidismrRrPeating_disordercurrent_medications)__name__ __module__ __qualname__rrstr__annotations__rtintr>boolrurr@rwryrLrGrNrzrRrPr{r|rrrrrrqs             rqc@s2eZdZUeed<dZeed<dZee ed<dS)ExplainRequestpatientrYquestionN rules_result) r}r~rrqrrrrrdictrrrrrs  rrcCs,g}|jr |d|jr|d|jdkr|dn|jdkr(|dn |jdkr2|d|jdkr=|d n|jdkrH|d n |jdkrR|d |jrZ|d |jrb|d |jrj|d|jdkrt|d|j r||d|j r|d|j r|d|j r|d|S)Nr>r@mild liver_mildmoderateliver_moderatesevererC renal_mildrenal_moderaterDrLrGrNArSrzrRrPr{) r>appendr@rwryrLrGrNrtrzrRrPr{)rrorrrget_patient_conditionssD                       rrnroc Csjg}|tvr|St|}|D]$}||vr2||}||d|}||||d|d|ddq|S)N_kor3r1)rnr9r3r1 reason_ko) DRUG_RULESgetr)rnroresultsr7r9rulerule_korrrcheck_drug_for_conditionss"rz/healthcCsddttdS)Nokz0.2.0)statusr drugs_count)rkrrrrrhealth3srz /api/drugscCsttttdS)N)r(count)listrkeysrkrrrr list_drugs7srz/api/check-safetyc Cs t|}g}g}g}tD]G}t||}|D]=}|ddkr%||q|ddkr1||q|ddkr=||q|ddkrI||q|ddkrT||qqdd|D}d d|D} d d|D} ||| | ||t|t| t| t|t|d d S) Nr3r/avoidr0rbsafecSg|] }|ddkr|qSr3r/r.0wrrr Pz check_safety..cSrr3rrrrrrrQrcSrr3r0rrrrrrRr) banned_count avoid_count caution_countpreferred_count safe_count)ror/rr0rbrsummary)rrrrrrk) rro all_warningsrb safe_listrn drug_resultsresultr/rr0rrr check_safety;sF             rz/api/check-drugr"cCs6t|}t||}tdd|Ddk}||||dS)NcSsg|] }|ddvr|qS)r3)r/rrrrrrrrirz%check_single_drug..r)rnrowarningsis_safe)rrrk)r"rrorrrrrcheck_single_drugds rz/api/conditionscCsddddddddddd d gd d d dd gdd ddddddddddddddddddddddddddddddd dg iS)!Nror>r=r)keylabeltyper@r?rwu 간질환select)rvrrr)rrroptionsryu 신기능)rxrrrrLrKrGu 심장질환rNrMrzu갑상선저하증rRu 발작력rPrOr{u 섭식장애rtu나이numberrrrrrlist_conditionsrs          ruS당신은 정신건강의학과 전문의를 보조하는 약물 안전성 전문가입니다. 환자 정보와 규칙 기반 분석 결과를 바탕으로 임상적 설명을 제공합니다. 원칙: 1. 규칙 엔진의 금기/주의 판정은 절대적으로 존중합니다 (L1 = 사실) 2. 복합 상황에서는 우선순위와 대안을 제시합니다 (L2 = 판단) 3. 의사가 이해하기 쉽게 근거와 함께 설명합니다 (L3 = 설명) 4. 한국어로 답변하고, 약물명은 영문 그대로 사용합니다 5. 간결하게 핵심만 전달합니다 (200자 이내 권장) z /api/explainrequestc sts ddddS|jsdt|j}g}g}g}tD]/}t||}|D]%}|ddvr2||q$|ddkr>||q$|ddkrI||q$q|dd |Dd d |Dd d |D||d }n|j}|j} d | jd} | j rx| d7} | j r| d7} | j dkr| d| j 7} | j dkr| d| j 7} | j r| d7} | jr| d7} | jr| d| j7} d} |drdd |dD} | dd| d7} |drdd |dD} | d d| d7} |d!rd"d |d!D} | d#d| d7} |drd$d |dD} | d%d| d7} |jr |jnd&} | d'| d(| }zttd)t|d)g}t|}d*|j| | d+WSty_}zdt|ddWYd}~Sd}~ww),NFzAPI key not configured)successerror explanationr3)r/rr0rbrcSrrrrrrrrrz#explain_with_ai..cSrrrrrrrrrcSrrrrrrrrr)ror/rr0rbru환자: u세u, 임신u , 수유중rvu , 간기능 rxu , 신기능 u, 당뇨u, 심장질환u , 진단: u규칙 분석 결과: r/cS$g|]}|dd|ddqSrn(r)rrrrrr$u - 금기: z,  rcSrrrrrrrrru - 회피: r0cSrrrrrrrrru - 주의: cSrrrrrrrrru - 권장: uV이 환자에게 적합한 약물 선택에 대해 종합적으로 설명해주세요.z u 질문: )contentT)rrpatient_summary rules_summary)llmrrrrrrrrtr>r@rwryrLrGrrrjoinrr SYSTEM_PROMPTr invokerr5r)rrorrbrrnrrrrrrrir user_messagemessagesresponser:rrrexplain_with_ais                    rz/api/ai-statuscCstdutr ddSddS)NrZ) llm_availabler[)rrrrr ai_statuss r__main__z0.0.0.0i@)rport)7fastapirfastapi.middleware.corsrpydanticrtypingrrrosr contextlibrlangchain_anthropicr langchain_core.messagesr r appcursors DictCursorrrrrr+r,r;r4environrrXradd_middlewarerhrprrqrrrrrrpostrrrrrrr}uvicornrunrrrrs        !< $  (  c