apps/saju.py

import os
import requests
import re       # 정규식 파싱용
import random
from datetime import datetime, timedelta
from typing import Union
from fastapi import APIRouter, Request
from fastapi.responses import JSONResponse, HTMLResponse
from fastapi.templating import Jinja2Templates
from pydantic import BaseModel
from google import genai

router = APIRouter()
templates = Jinja2Templates(directory="templates")

# [수정 1] API 키 저장 변수명 정의
GEMINI_API_KEYS = []
default_key = os.environ.get("GEMINI_API_KEY")
if default_key: GEMINI_API_KEYS.append(default_key)
for i in range(1, 11):
    key = os.environ.get(f"GEMINI_API_KEY{i}")
    if key and key not in GEMINI_API_KEYS: GEMINI_API_KEYS.append(key)

SLACK_WEBHOOK_URL = os.environ.get("SLACK_WEBHOOK_URL")

def send_slack_message(message):
    if not SLACK_WEBHOOK_URL: return 
    try: requests.post(SLACK_WEBHOOK_URL, json={"text": message})
    except: pass

class SajuRequest(BaseModel):
    year: Union[str, int]
    month: Union[str, int]
    day: Union[str, int]
    time: Union[str, int] = "시간 모름"

@router.get("/saju", response_class=HTMLResponse)
async def main_saju(request: Request):
    return templates.TemplateResponse("main_saju.html", {"request": request})

@router.post("/analyze")
def analyze_saju(req: SajuRequest):
    req_year = str(req.year)
    req_month = str(req.month)
    req_day = str(req.day)
    req_time = str(req.time)

    candidate_models = [
        "gemini-3-flash-preview",
        "gemini-3-pro-preview",
        "gemini-flash-latest"      
    ]
    
    birth_info = f"{req_year}년 {req_month}월 {req_day}일 ({req_time})"
    korea_now = datetime.now() + timedelta(hours=9)
    today_date = korea_now.strftime("%Y년 %m월 %d일")
    
    prompt = f"""
    당신은 트렌디한 '퍼스널 사주 패션 디렉터' Theo입니다. 
    
    [사용자 정보]
    - 생년월일: {birth_info} (양력/Solar Calendar 기준)
    - 요청사항: 위 양력 날짜를 바탕으로 정확한 사주를 분석하세요.
    
    [현재 시점]
    - 오늘 날짜: {today_date}

    [출력 가이드]
    1. **중요: 모든 강조(Bold) 처리는 마크다운(**)이 아닌 HTML `<b>` 태그를 사용하세요.**
    2. 인사말: 반드시 `<div class="greeting">` 태그로 감싸서 작성하세요. 
    3. 소개 생략: 자기소개는 하지 마세요.
    4. 형식: 모든 섹션('오늘의 사주 분석', '오늘의 행운 컬러', '오늘의 행운 번호', '오늘의 추천 코디', '오늘의 마음⼼가짐')은 <details><summary>... [보기]</summary></details> 태그로 감싸서 접어두세요.
    5. **(필수)** '오늘의 행운 번호' 섹션을 추가하여 1부터 45 사이의 겹치지 않는 숫자 6개를 추천해 주세요.
    6. **(시스템 파싱용 - 매우 중요)** 백엔드 시스템 파싱을 위해 응답 텍스트 맨 마지막 줄에 반드시 아래와 같은 형식의 HTML 주석을 추가하세요. (숫자는 쉼표로 구분)
    7. 톤앤매너: 전문적이고 우아한 어조.
    """

    last_error = None
    
    if not GEMINI_API_KEYS:
        return JSONResponse(content={'result': "<div class='greeting'>서버에 API 키가 설정되지 않았습니다.</div>"})

    send_slack_message(f"🔮 [Theo] 분석 요청: {birth_info}")

    for key_idx, current_key in enumerate(GEMINI_API_KEYS):
        key_alias = f"KEY_{key_idx+1}"
        
        try:
            client = genai.Client(api_key=current_key)
        except Exception:
            continue

        for model_name in candidate_models:
            try:
                response = client.models.generate_content(
                    model=model_name, 
                    contents=prompt
                )
                
                if response.text:
                    send_slack_message(f"✅ [성공] {key_alias} / {model_name}")
                    
                    # ==========================================
                    # 🚨 [추가된 로직] 행운 번호 추출 (파싱)
                    # ==========================================
                    text_result = response.text
                    lucky_numbers = []
                    
                    # 제미나이가 숨겨둔 주석()에서 번호만 빼오기
                    match = re.search(r'', text_result)
                    if match:
                        try:
                            num_str = match.group(1)
                            # "7, 14, 21" 문자를 숫자 리스트 [7, 14, 21]로 변환 및 1~45 범위 체크
                            lucky_numbers = [int(n.strip()) for n in num_str.split(',') if n.strip().isdigit()]
                            lucky_numbers = [n for n in lucky_numbers if 1 <= n <= 45]
                            lucky_numbers = list(set(lucky_numbers)) # 중복 제거
                        except Exception:
                            pass
                            
                    # 💡 [안전장치] 제미나이가 실수로 번호를 안 줬거나 6개가 안 될 경우, 
                    # 빈자리는 파이썬이 알아서 무작위 행운번호로 채워줍니다! (에러 방지)
                    while len(lucky_numbers) < 6:
                        rand_num = random.randint(1, 45)
                        if rand_num not in lucky_numbers:
                            lucky_numbers.append(rand_num)
                    
                    # 딱 6개만 자르고 오름차순 정렬
                    lucky_numbers = lucky_numbers[:6]
                    lucky_numbers.sort()

                    # ==========================================
                    # 프론트엔드로 텍스트를 넘기기 전에 주석()을 아예 지워버림!
                    text_result = re.sub(r'', '', text_result).strip()
                    # ==========================================

                    # 프론트엔드(main_saju.html)에서 약속했던 luckyNumbers 키값으로 쏴줍니다!
                    return {"result": text_result, "luckyNumbers": lucky_numbers}
                
            except Exception as e:
                last_error = e
                continue

    error_msg = str(last_error)

    send_slack_message(f"🚨 [전체 실패] 에러: {error_msg}")
    
    return JSONResponse(content={
        'result': f"<div class='greeting'>죄송합니다. 접속이 원활하지 않습니다.<br><span style='font-size:0.8rem; color:#999'>({error_msg})</span></div>"
    })