Update utils/calculations.py

utils/calculations.py

@@ -1,77 +1,70 @@
 # utils/calculations.py
 
-def get_ganji(year):
+def get_ganji_full(birth_date, birth_time, calendar_type):
     """
-    연도를 입력받아 천간(Heavenly Stem)과 지지(Earthly Branch), 오행(Element)을 계산
+    생년월일시를 입력받아 사주팔자(네 기둥) 정보를 계산 (약식)
     """
-    cheon = ["갑(Green)", "을(Green)", "병(Red)", "정(Red)", "무(Yellow)", "기(Yellow)", "경(White)", "신(White)", "임(Black)", "계(Black)"]
-    ji = ["자(쥐)", "축(소)", "인(호랑이)", "묘(토끼)", "진(용)", "사(뱀)", "오(말)", "미(양)", "신(원숭이)", "유(닭)", "술(개)", "해(돼지)"]
-    
-    # 오행 매핑 (천간 기준)
-    elements = {
-        "갑": "목(나무)", "을": "목(나무)",
-        "병": "화(불)", "정": "화(불)",
-        "무": "토(흙)", "기": "토(흙)",
-        "경": "금(쇠)", "신": "금(쇠)",
-        "임": "수(물)", "계": "수(물)"
-    }
+    # 천간/지지 데이터
+    cheon = ["갑", "을", "병", "정", "무", "기", "경", "신", "임", "계"]
+    ji = ["자", "축", "인", "묘", "진", "사", "오", "미", "신", "유", "술", "해"]
+    elements = {"갑": "목", "을": "목", "병": "화", "정": "화", "무": "토", "기": "토", "경": "금", "신": "금", "임": "수", "계": "수"}
+    animals = {"자": "쥐", "축": "소", "인": "범", "묘": "토끼", "진": "용", "사": "뱀", "오": "말", "미": "양", "신": "원숭이", "유": "닭", "술": "개", "해": "돼지"}
 
-    # 4년생이 갑자년 기준 (index 0)
-    year_idx = year - 4
-    stem_idx = year_idx % 10
-    branch_idx = year_idx % 12
+    year = birth_date.year
+    
+    # 1. 연주(Year Pillar) 계산
+    y_idx = year - 4
+    y_stem = cheon[y_idx % 10]
+    y_branch = ji[y_idx % 12]
     
-    stem_kor = cheon[stem_idx][0] # 갑, 을..
+    # 2. 월주, 일주, 시주 (정확한 만세력은 라이브러리가 필요하므로 여기선 Mockup 로직 사용)
+    # 실제 앱에선 'korean-lunar-calendar' 라이브러리 사용 권장
+    # 여기선 AI에게 정확한 날짜를 넘겨주는 것에 집중
     
     return {
-        "year_ganji": f"{cheon[stem_idx].split('(')[0]}{ji[branch_idx].split('(')[0]}년", # 갑자년
-        "animal": ji[branch_idx], # 쥐, 소...
-        "color": cheon[stem_idx].split('(')[1].replace(')', ''), # Green, Red...
-        "element": elements[stem_kor], # 목, 화...
-        "stem_desc": f"천간은 {cheon[stem_idx]}",
-        "branch_desc": f"지지는 {ji[branch_idx]}"
+        "text_data": f"{year}년 {birth_date.month}월 {birth_date.day}일 {birth_time.hour}시 {birth_time.minute}분 ({calendar_type})",
+        "year_ganji": f"{y_stem}{y_branch}",
+        "animal": animals[y_branch],
+        "element": elements[y_stem], # 태어난 해의 천간 기준 (띠 색깔 결정)
+        "desc": f"당신은 {animals[y_branch]}띠이며, {elements[y_stem]}({y_stem})의 기운을 가지고 태어났습니다."
     }
 
-def get_fengshui_info(birth_year, gender):
+def get_fengshui_advanced(year, gender, door_dir, head_dir):
     """
-    본명궁(Life Gua) 계산 및 길흉 방위 산출 (가장 중요한 공식)
+    본명궁 + 침대 방향 정밀 분석
     """
-    # 1. 숫자를 다 더해서 한 자리로 만듦 (예: 1988 -> 1+9+8+8=26 -> 2+6=8)
-    # (약식 계산법 사용)
-    digits_sum = sum(int(digit) for digit in str(birth_year))
+    # 본명궁 계산 (간략식)
+    digits_sum = sum(int(digit) for digit in str(year))
     while digits_sum > 9:
         digits_sum = sum(int(digit) for digit in str(digits_sum))
     
-    # 2. 본명궁 상수(Gua Number) 계산
-    if gender == "남성":
-        gua_num = 11 - digits_sum
-    else: # 여성
-        gua_num = digits_sum + 4
+    if gender == "남성" or gender == "Male":
+        gua = 11 - digits_sum
+    else:
+        gua = digits_sum + 4
         
-    # 예외처리 (9를 넘거나 음수 처리)
-    while gua_num > 9: gua_num -= 9
-    while gua_num < 1: gua_num += 9
-    if gua_num == 5: # 5황토성은 남자는 2(곤), 여자는 8(간)으로 치환
-        gua_num = 2 if gender == "남성" else 8
+    while gua > 9: gua -= 9
+    while gua < 1: gua += 9
+    if gua == 5: gua = 2 if (gender == "남성" or gender == "Male") else 8
 
-    # 3. 동사택(East Group) vs 서사택(West Group) 판별
-    # 동사택: 1, 3, 4, 9 / 서사택: 2, 6, 7, 8
-    is_east_group = gua_num in [1, 3, 4, 9]
-    group_name = "동사택(East Group)" if is_east_group else "서사택(West Group)"
+    is_east = gua in [1, 3, 4, 9]
+    group = "동사택 (East Group)" if is_east else "서사택 (West Group)"
     
-    # 4. 길흉 방향 데이터
-    directions = {
-        "동사택": {"길": ["북", "동", "남", "남동"], "흉": ["북동", "남서", "서", "북서"]},
-        "서사택": {"길": ["북동", "남서", "서", "북서"], "흉": ["북", "동", "남", "남동"]}
-    }
+    good_dirs = ["북", "동", "남", "남동"] if is_east else ["북동", "남서", "서", "북서"]
+    
+    # 침대 ���향 진단
+    korean_dirs = {"N": "북", "NE": "북동", "E": "동", "SE": "남동", "S": "남", "SW": "남서", "W": "서", "NW": "북서"}
+    
+    # 입력된 영문 방향을 한글로 변환 (비교용)
+    head_kor = korean_dirs.get(door_dir, door_dir) # 매핑 안 되면 그대로
     
-    good_dirs = directions["동사택" if is_east_group else "서사택"]["길"]
-    bad_dirs = directions["동사택" if is_east_group else "서사택"]["흉"]
+    # 매칭 여부 (단순 문자열 포함 여부로 체크)
+    is_good_head = any(d in head_dir for d in good_dirs) or (head_dir in good_dirs)
 
     return {
-        "gua_number": gua_num,
-        "group": group_name,
-        "good_direction": ", ".join(good_dirs),
-        "bad_direction": ", ".join(bad_dirs),
-        "advice": f"당신은 {group_name}이므로 머리를 {good_dirs[0]}쪽으로 두고 자야 합니다. 절대 {bad_dirs[0]}쪽을 보지 마세요."
+        "gua": gua,
+        "group": group,
+        "good_dirs": ", ".join(good_dirs),
+        "match": is_good_head,
+        "advice": f"귀하는 {group}입니다. {head_dir} 방향은 귀하에게 {'좋은' if is_good_head else '나쁜'} 기운을 줍니다."
     }