Update app.py

app.py

@@ -4,6 +4,29 @@ import pandas as pd
 from sklearn.linear_model import LogisticRegression
 from xgboost import XGBClassifier
 import time
+import os
+import re
+import json
+
+# ============================================================
+# 0-A. Claude API 클라이언트 초기화 (5세대용)
+# ============================================================
+# 환경변수 ANTHROPIC_API_KEY 가 설정되어 있으면 실제 API 호출 모드
+# 없거나 호출 실패 시 자동으로 시뮬레이션 모드로 fallback
+try:
+    from anthropic import Anthropic
+    _api_key = os.environ.get("ANTHROPIC_API_KEY")
+    if _api_key:
+        claude_client = Anthropic(api_key=_api_key)
+        CLAUDE_AVAILABLE = True
+    else:
+        claude_client = None
+        CLAUDE_AVAILABLE = False
+except ImportError:
+    claude_client = None
+    CLAUDE_AVAILABLE = False
+
+CLAUDE_MODEL = "claude-sonnet-4-6"  # 2026년 5월 현재 권장 모델
 
 # ============================================================
 # 0. 공통 설정
@@ -361,89 +384,240 @@ def render_gen4(amount, hour, new_payee_bin, ratio, prob3):
     </div>
     """
 
-def render_gen5(amount, hour, new_payee_bin, ratio, prior_avg):
-    """5세대 LLM 시뮬레이션 (안정성을 위해 API 호출 없이 하드코딩)"""
-    is_high_risk = (new_payee_bin == 1 and (hour <= 6 or hour >= 22) and amount >= 500)
-    prob = 0.95 if is_high_risk else min(prior_avg + 0.02, 0.98)
+# ============================================================
+# 5세대 — Claude API 실제 호출
+# ============================================================
+def build_claude_prompt(amount, hour, new_payee_bin, ratio, prior_avg, prior_dec):
+    """Claude에게 보낼 시스템 프롬프트와 사용자 메시지 구성"""
+    system_prompt = (
+        "당신은 한국 은행의 FDS(이상금융거래탐지시스템) 분석 전문가입니다. "
+        "주어진 거래 정보와 1-4세대 모델의 사전 판단을 바탕으로 사기 여부를 종합 판정하세요. "
+        "특히 부동산 잔금 송금, 사업자 대금 결제 같은 정상 거래 패턴과 보이스피싱·대포통장 패턴을 구분해야 합니다. "
+        "반드시 다음 JSON 형식으로만 응답하세요(다른 텍스트 금지):\n"
+        '{\n'
+        '  "risk_score": <0.0~1.0 사이 사기 의심도>,\n'
+        '  "decision": "차단" | "추가 인증" | "통과",\n'
+        '  "reasoning_steps": [\n'
+        '    {"step": "<추론 단계>", "attention": <0.0~1.0>},\n'
+        '    ... (3~5개)\n'
+        '  ],\n'
+        '  "judgment": "<최종 판단 자연어 (왜 그렇게 판단했는지 + 권고 조치)>"\n'
+        '}'
+    )
+    payee_str = "신규" if new_payee_bin == 1 else "기존"
+    time_period = "새벽" if (hour <= 6 or hour >= 22) else ("주간" if 9 <= hour <= 18 else "저녁")
+    user_message = (
+        f"[거래 정보]\n"
+        f"- 금액: {amount:.0f}만원\n"
+        f"- 거래 시간: {int(hour):02d}시 ({time_period})\n"
+        f"- 수취인: {payee_str} 수취인\n"
+        f"- 과거 대비 배율: {ratio:.1f}배 (송금인의 평균 거래액 대비)\n\n"
+        f"[1-4세대 모델 사전 판단]\n"
+        f"- 평균 사기 확률: {prior_avg*100:.1f}%\n"
+        f"- 종합 판정: {prior_dec}\n\n"
+        f"위 거래에 대해 FDS 전문가 관점에서 종합 판정해주세요."
+    )
+    return system_prompt, user_message
+
+def call_claude_api(amount, hour, new_payee_bin, ratio, prior_avg, prior_dec):
+    """실제 Claude API 호출. 성공 시 (parsed_dict, raw_meta) 반환, 실패 시 (None, error_msg)"""
+    if not CLAUDE_AVAILABLE:
+        return None, "API 키 미설정 (ANTHROPIC_API_KEY 환경변수 없음)"
+
+    system_prompt, user_message = build_claude_prompt(
+        amount, hour, new_payee_bin, ratio, prior_avg, prior_dec
+    )
+    try:
+        t0 = time.time()
+        response = claude_client.messages.create(
+            model=CLAUDE_MODEL,
+            max_tokens=1024,
+            temperature=0.2,
+            system=system_prompt,
+            messages=[{"role": "user", "content": user_message}]
+        )
+        latency = time.time() - t0
+
+        raw_text = response.content[0].text.strip()
+
+        # JSON 추출: 응답 중 첫 번째 { 부터 마지막 } 까지
+        json_match = re.search(r'\{.*\}', raw_text, re.DOTALL)
+        if not json_match:
+            return None, f"JSON 파싱 실패: {raw_text[:200]}"
+
+        parsed = json.loads(json_match.group(0))
+
+        meta = {
+            "input_tokens": response.usage.input_tokens,
+            "output_tokens": response.usage.output_tokens,
+            "latency": latency,
+            "model": CLAUDE_MODEL,
+            "raw_text": raw_text,
+        }
+        return parsed, meta
+    except Exception as e:
+        return None, f"API 호출 실패: {type(e).__name__}: {str(e)[:200]}"
+
+def render_gen5_card(amount, hour, new_payee_bin, ratio, prior_avg, prior_dec,
+                     prob, decision_text, reasoning_steps, judgment_text,
+                     meta_html, source_label):
+    """5세대 카드 HTML 렌더링 (API/시뮬레이션 공용)"""
     dec, bg, fg = decide(prob)
-    payee_str = "신규 수취인" if new_payee_bin == 1 else "기존 수취인"
-    time_str = "새벽" if (hour <= 6 or hour >= 22) else "일반"
 
     context_html = (
-        f"[SYS] 당신은 FDS 분석가입니다. 거래 패턴을 분석해 사기 여부를 판정하세요.<br>"
+        f"[SYS] 당신은 한국 은행의 FDS 분석 전문가입니다. 거래 정보와 1-4세대 사전 판단을 토대로 종합 판정하세요.<br>"
         f"[INPUT] amount={amount:.0f}만, hour={int(hour):02d}, "
         f"new_payee={'true' if new_payee_bin==1 else 'false'}, ratio={ratio:.1f}×<br>"
-        f"[PRIOR] 1-4세대 합의: {dec} (avg {prior_avg*100:.1f}%)<br>"
-        f"[TASK] 의심 사유와 권고 조치를 한국어로."
+        f"[PRIOR] 1-4세대 평균: {prior_avg*100:.1f}% / 종합: {prior_dec}<br>"
+        f"[TASK] JSON 형식으로 risk_score, decision, reasoning_steps, judgment 출력"
     )
 
-    cot_rows = (
-        f"<tr><td style='padding:6px 4px;'>1</td>"
-        f"<td style='padding:6px 4px; color:#666;'>시간 {int(hour):02d}시 + {payee_str} → 보이스피싱 정형 패턴 매칭</td>"
-        f"<td style='text-align:right; padding:6px 4px; font-family:monospace; color:#666;'>0.42</td></tr>"
-        f"<tr><td style='padding:6px 4px;'>2</td>"
-        f"<td style='padding:6px 4px; color:#666;'>평소 {ratio:.1f}배 금액 → 송금인 평소 행동 이탈도 측정</td>"
-        f"<td style='text-align:right; padding:6px 4px; font-family:monospace; color:#666;'>0.28</td></tr>"
-        f"<tr><td style='padding:6px 4px;'>3</td>"
-        f"<td style='padding:6px 4px; color:#666;'>4세대 GNN 신호 + {amount:.0f}만원 고액 → 위험 가중</td>"
-        f"<td style='text-align:right; padding:6px 4px; font-family:monospace; color:#666;'>0.21</td></tr>"
-        f"<tr><td style='padding:6px 4px;'>4</td>"
-        f"<td style='padding:6px 4px; color:#666;'>최근 보이스피싱 트렌드 사전 지식 활용</td>"
-        f"<td style='text-align:right; padding:6px 4px; font-family:monospace; color:#666;'>0.09</td></tr>"
-    )
-
-    if is_high_risk:
-        nl_judgment = (
-            f"<b style='color:#633806;'>차단 권고 (의심도 {prob*100:.0f}%)</b><br><br>"
-            f"{int(hour):02d}시 {time_str} 시간대에 평소보다 {ratio:.1f}배 급증한 {amount:.0f}만원이 "
-            f"<u>{payee_str}</u>에게 이체되는 것은 전형적인 보이스피싱 패턴입니다. "
-            f"특히 수취인 계좌가 최근 사기 의심 계좌와 2-hop 거리에 있다는 GNN 신호까지 더해져 위험도가 매우 높습니다.<br><br>"
-            f"<b>권고 조치:</b> ① 즉시 거래 보류, ② 등록된 전화번호로 본인 직접 확인, ③ 확인 전 자금 동결 24시간 유지"
-        )
-    elif prob >= 0.5:
-        nl_judgment = (
-            f"<b style='color:#633806;'>추가 인증 권고 (의심도 {prob*100:.0f}%)</b><br><br>"
-            f"{int(hour):02d}시 거래에서 일부 이상 신호({ratio:.1f}배 금액, {payee_str})가 감지되었으나 "
-            f"결정적 위험 패턴은 아닙니다. 차단보다는 추가 인증으로 본인 의사를 확인하는 것이 적절합니다.<br><br>"
-            f"<b>권고 조치:</b> ① ARS 또는 OTP 추가 인증, ② 송금 의도 재확인 메시지 발송"
-        )
-    else:
-        nl_judgment = (
-            f"<b style='color:#3B6D11;'>통과 권고 (의심도 {prob*100:.0f}%)</b><br><br>"
-            f"{int(hour):02d}시 거래의 패턴이 송금인의 평소 행동 범위 내에 있으며, "
-            f"1-4세대 모델 모두 위험 신호를 강하게 보내지 않았습니다. 정상 거래로 판단됩니다.<br><br>"
-            f"<b>권고 조치:</b> 별도 조치 없이 거래 진행"
+    cot_rows = ""
+    for i, step in enumerate(reasoning_steps, 1):
+        step_text = step.get("step", "") if isinstance(step, dict) else str(step)
+        attn = step.get("attention", 0.0) if isinstance(step, dict) else 0.0
+        try:
+            attn = float(attn)
+        except (ValueError, TypeError):
+            attn = 0.0
+        cot_rows += (
+            f"<tr><td style='padding:6px 4px;'>{i}</td>"
+            f"<td style='padding:6px 4px; color:#666;'>{step_text}</td>"
+            f"<td style='text-align:right; padding:6px 4px; font-family:monospace; color:#666;'>{attn:.2f}</td></tr>"
         )
 
+    judg_color = "#3B6D11" if prob < 0.5 else "#633806"
+    judgment_html = (
+        f"<b style='color:{judg_color};'>{decision_text} 권고 (의심도 {prob*100:.0f}%)</b><br><br>"
+        f"{judgment_text}"
+    )
+
     return f"""
     <div style="{CARD_STYLE}">
-      {card_header("GEN 5 · FOUNDATION MODEL (LLM)", "초거대 추론 모델 (시뮬레이션)", dec, bg, fg, f"의심도 {prob*100:.0f}%")}
+      {card_header("GEN 5 · FOUNDATION MODEL (LLM)", f"초거대 추론 모��� ({source_label})", dec, bg, fg, f"의심도 {prob*100:.0f}%")}
       <p style="font-size:13px; color:#666; margin:4px 0 6px;">컨텍스트 토큰화</p>
       <div style="background:#f5f5f0; padding:10px 12px; border-radius:6px; font-family:monospace; font-size:11px; line-height:1.7; margin-bottom:12px;">{context_html}</div>
       <p style="font-size:13px; color:#666; margin:4px 0 6px;">추론 체인 (Chain-of-Thought)</p>
       <table style="width:100%; font-size:13px; border-collapse:collapse; margin-bottom:12px;">
         <thead>
           <tr style="border-bottom:0.5px solid rgba(0,0,0,0.15);">
-            <th style="text-align:left; padding:6px 4px; font-weight:500; color:#666; width:12%;">단계</th>
-            <th style="text-align:left; padding:6px 4px; font-weight:500; color:#666; width:58%;">추론 내용</th>
-            <th style="text-align:right; padding:6px 4px; font-weight:500; color:#666; width:30%;">Attention</th>
+            <th style="text-align:left; padding:6px 4px; font-weight:500; color:#666; width:8%;">단계</th>
+            <th style="text-align:left; padding:6px 4px; font-weight:500; color:#666; width:72%;">추론 내용</th>
+            <th style="text-align:right; padding:6px 4px; font-weight:500; color:#666; width:20%;">Attention</th>
           </tr>
         </thead>
         <tbody>{cot_rows}</tbody>
       </table>
-      <p style="font-size:13px; color:#666; margin:4px 0 6px;">생성된 자연어 판단 (Token 확률)</p>
-      <div style="background:#FAEEDA; padding:12px 14px; border-radius:6px; font-size:13px; line-height:1.7; color:#412402;">{nl_judgment}</div>
+      <p style="font-size:13px; color:#666; margin:4px 0 6px;">생성된 자연어 판단</p>
+      <div style="background:#FAEEDA; padding:12px 14px; border-radius:6px; font-size:13px; line-height:1.7; color:#412402;">{judgment_html}</div>
       <details style="margin-top:10px;">
         <summary style="font-size:12px; color:#666; cursor:pointer;">생성 파라미터 보기</summary>
-        <div style="background:#f5f5f0; padding:8px 12px; border-radius:6px; font-family:monospace; font-size:11px; margin-top:6px; line-height:1.6;">
-          model: claude-sonnet-4-5 / temperature: 0.2 / top_p: 0.9 / max_tokens: 512<br>
-          input_tokens: 142 / output_tokens: 98 / latency: 1.83s (시뮬레이션)
-        </div>
+        <div style="background:#f5f5f0; padding:8px 12px; border-radius:6px; font-family:monospace; font-size:11px; margin-top:6px; line-height:1.6;">{meta_html}</div>
       </details>
       <p style="font-size:12px; color:#888; margin:10px 0 0; font-style:italic;">강점: 사전 학습된 도메인 지식으로 '왜 사기인지' 자연어 설명 + 권고 조치까지 자동 생성</p>
     </div>
     """
 
+def render_gen5_simulation(amount, hour, new_payee_bin, ratio, prior_avg, prior_dec):
+    """API 호출 실패/미설정 시 fallback용 시뮬레이션"""
+    is_high_risk = (new_payee_bin == 1 and (hour <= 6 or hour >= 22) and amount >= 500)
+    prob = 0.95 if is_high_risk else min(prior_avg + 0.02, 0.98)
+    payee_str = "신규 수취인" if new_payee_bin == 1 else "기존 수취인"
+    time_str = "새벽" if (hour <= 6 or hour >= 22) else "일반"
+
+    reasoning_steps = [
+        {"step": f"시간 {int(hour):02d}시 + {payee_str} → 보이스피싱 정형 패턴 매칭", "attention": 0.42},
+        {"step": f"평소 {ratio:.1f}배 금액 → 송금인 평소 행동 이탈도 측정", "attention": 0.28},
+        {"step": f"4세대 GNN 신호 + {amount:.0f}만원 고액 → 위험 가중", "attention": 0.21},
+        {"step": "최근 보이스피싱 트렌드 사전 지식 활용", "attention": 0.09},
+    ]
+
+    if is_high_risk:
+        decision_text = "차단"
+        judgment_text = (
+            f"{int(hour):02d}시 {time_str} 시간대에 평소보다 {ratio:.1f}배 급증한 {amount:.0f}만원이 "
+            f"<u>{payee_str}</u>에게 이체되는 것은 전형적인 보이스피싱 패턴입니다. "
+            f"수취인 계좌가 사기 의심 계좌와 2-hop 거리에 있다는 GNN 신호까지 더해져 위험도가 매우 높습니다.<br><br>"
+            f"<b>권고 조치:</b> ① 즉시 거래 보류, ② 등록된 전화번호로 본인 직접 확인, ③ 확인 전 자금 동결 24시간 유지"
+        )
+    elif prob >= 0.5:
+        decision_text = "추가 인증"
+        judgment_text = (
+            f"{int(hour):02d}시 거래에서 일부 이상 신호({ratio:.1f}배 금액, {payee_str})가 감지되었으나 "
+            f"결정적 위험 패턴은 아닙니다. 차단보다는 추가 인증으로 본인 의사를 확인하는 것이 적절합니다.<br><br>"
+            f"<b>권고 조치:</b> ① ARS 또는 OTP 추가 인증, ② 송금 의도 재확인 메시지 발송"
+        )
+    else:
+        decision_text = "통과"
+        judgment_text = (
+            f"{int(hour):02d}시 거래의 패턴이 송금인의 평소 행동 범위 내에 있으며, "
+            f"1-4세대 모델 모두 위험 신호를 강하게 보내지 않았습니다. 정상 거래로 판단됩니다.<br><br>"
+            f"<b>권고 조치:</b> 별도 조치 없이 거래 진행"
+        )
+
+    meta_html = (
+        "mode: <b>SIMULATION</b> (API 미연결)<br>"
+        "이 결과는 if-else 하드코딩으로 생성된 시뮬레이션입니다.<br>"
+        "실제 Claude 호출을 활성화하려면 ANTHROPIC_API_KEY 환경변수를 설정하세요."
+    )
+
+    return render_gen5_card(
+        amount, hour, new_payee_bin, ratio, prior_avg, prior_dec,
+        prob, decision_text, reasoning_steps, judgment_text,
+        meta_html, source_label="시뮬레이션 모드"
+    )
+
+def render_gen5(amount, hour, new_payee_bin, ratio, prior_avg):
+    """5세대 진입점: API 가능 시 Claude 호출, 실패 시 시뮬레이션 fallback"""
+    prior_dec, _, _ = decide(prior_avg)
+
+    if not CLAUDE_AVAILABLE:
+        return render_gen5_simulation(amount, hour, new_payee_bin, ratio, prior_avg, prior_dec)
+
+    parsed, meta_or_err = call_claude_api(amount, hour, new_payee_bin, ratio, prior_avg, prior_dec)
+
+    if parsed is None:
+        fallback_html = render_gen5_simulation(amount, hour, new_payee_bin, ratio, prior_avg, prior_dec)
+        warning = (
+            f'<div style="background:#FCEBEB; border-left:3px solid #A32D2D; padding:10px 14px; '
+            f'border-radius:6px; margin-bottom:10px; font-size:13px; color:#4A1B0C;">'
+            f'⚠️ Claude API 호출 실패. 시뮬레이션 모드로 대체합니다.<br>'
+            f'<span style="font-family:monospace; font-size:11px; color:#712B13;">사유: {meta_or_err}</span>'
+            f'</div>'
+        )
+        return warning + fallback_html
+
+    try:
+        prob = float(parsed.get("risk_score", 0.5))
+        prob = max(0.0, min(1.0, prob))
+        decision_text = parsed.get("decision", "추가 인증")
+        reasoning_steps = parsed.get("reasoning_steps", [])
+        judgment_text = parsed.get("judgment", "(판단 내용 누락)")
+
+        meta = meta_or_err
+        meta_html = (
+            f"model: <b>{meta['model']}</b> / temperature: 0.2 / max_tokens: 1024<br>"
+            f"input_tokens: {meta['input_tokens']} / output_tokens: {meta['output_tokens']} / "
+            f"latency: {meta['latency']:.2f}s<br>"
+            f"mode: <b style='color:#3B6D11;'>LIVE API CALL ✓</b>"
+        )
+
+        return render_gen5_card(
+            amount, hour, new_payee_bin, ratio, prior_avg, prior_dec,
+            prob, decision_text, reasoning_steps, judgment_text,
+            meta_html, source_label=f"실제 {CLAUDE_MODEL}"
+        )
+    except (KeyError, ValueError, TypeError) as e:
+        fallback_html = render_gen5_simulation(amount, hour, new_payee_bin, ratio, prior_avg, prior_dec)
+        warning = (
+            f'<div style="background:#FCEBEB; border-left:3px solid #A32D2D; padding:10px 14px; '
+            f'border-radius:6px; margin-bottom:10px; font-size:13px; color:#4A1B0C;">'
+            f'⚠️ Claude 응답 파싱 실패. 시뮬레이션 모드로 대체합니다.<br>'
+            f'<span style="font-family:monospace; font-size:11px; color:#712B13;">'
+            f'{type(e).__name__}: {str(e)[:150]}</span></div>'
+        )
+        return warning + fallback_html
+
 # ============================================================
 # 4. 메인 분석 함수
 # ============================================================
@@ -495,10 +669,18 @@ def analyze_transaction(amount, hour, new_payee, ratio):
 # 5. Gradio UI
 # ============================================================
 with gr.Blocks(theme=gr.themes.Default(), title="FDS 1-5세대 비교 데모") as demo:
-    gr.HTML("""
+    _api_badge = (
+        f'<span style="background:#EAF3DE; color:#3B6D11; padding:3px 10px; border-radius:8px; font-size:12px; font-weight:500;">'
+        f'● Claude API 연결됨 ({CLAUDE_MODEL})</span>'
+        if CLAUDE_AVAILABLE else
+        '<span style="background:#FAEEDA; color:#854F0B; padding:3px 10px; border-radius:8px; font-size:12px; font-weight:500;">'
+        '○ API 미연결 (5세대 시뮬레이션 모드)</span>'
+    )
+    gr.HTML(f"""
     <div style="text-align:center; padding:10px 0;">
       <h1 style="margin:0;">🛡️ FDS 1-5세대 비교 데모</h1>
       <p style="color:#666; margin:6px 0 0;">실무 운영 담당자 시연용 · 판단 수식·가중치·근거 전체 노출 모드</p>
+      <div style="margin-top:8px;">{_api_badge}</div>
     </div>
     """)