Update apps/saju.py

apps/saju.py

@@ -1,5 +1,7 @@
 import os
 import requests
+import re       # 정규식 파싱용
+import random
 from datetime import datetime, timedelta
 from typing import Union
 from fastapi import APIRouter, Request
@@ -67,19 +69,19 @@ def analyze_saju(req: SajuRequest):
     1. **중요: 모든 강조(Bold) 처리는 마크다운(**)이 아닌 HTML `<b>` 태그를 사용하세요.**
     2. 인사말: 반드시 `<div class="greeting">` 태그로 감싸서 작성하세요. 
     3. 소개 생략: 자기소개는 하지 마세요.
-    4. 형식: 모든 섹션('오늘의 사주 분석', '오늘의 행운 컬러', '오늘의 추천 코디', '오늘의 마음⼼가짐')은 <details><summary>... [보기]</summary></details> 태그로 감싸서 접어두세요.
-    5. 톤앤매너: 전문적이고 우아한 어조.
+    4. 형식: 모든 섹션('오늘의 사주 분석', '오늘의 행운 컬러', '오늘의 행운 번호', '오늘의 추천 코디', '오늘의 마음⼼가짐')은 <details><summary>... [보기]</summary></details> 태그로 감싸서 접어두세요.
+    5. **(필수)** '오늘의 행운 번호' 섹션을 추가하여 1부터 45 사이의 겹치지 않는 숫자 6개를 추천해 주세요.
+    6. **(시스템 파싱용 - 매우 중요)** 백엔드 시스템 파싱을 위해 응답 텍스트 맨 마지막 줄에 반드시 아래와 같은 형식의 HTML 주석을 추가하세요. (숫자는 쉼표로 구분)
+    7. 톤앤매너: 전문적이고 우아한 어조.
     """
 
     last_error = None
     
-    # [수정 2] api_keys -> GEMINI_API_KEYS 로 변수명 수정
     if not GEMINI_API_KEYS:
         return JSONResponse(content={'result': "<div class='greeting'>서버에 API 키가 설정되지 않았습니다.</div>"})
 
     send_slack_message(f"🔮 [Theo] 분석 요청: {birth_info}")
 
-    # [수정 3] api_keys -> GEMINI_API_KEYS 로 변수명 수정
     for key_idx, current_key in enumerate(GEMINI_API_KEYS):
         key_alias = f"KEY_{key_idx+1}"
         
@@ -97,13 +99,50 @@ def analyze_saju(req: SajuRequest):
                 
                 if response.text:
                     send_slack_message(f"✅ [성공] {key_alias} / {model_name}")
-                    return {"result": response.text}
+                    
+                    # ==========================================
+                    # 🚨 [추가된 로직] 행운 번호 추출 (파싱)
+                    # ==========================================
+                    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={