Update app.py

app.py

@@ -61,13 +61,11 @@ class SajuCalculator:
 
     def parse_date(self, date_str):
         """날짜 파싱"""
-        # 숫자만 추출
         numbers = re.findall(r'\d+', str(date_str))
         
         if not numbers:
             raise ValueError("날짜에서 숫자를 찾을 수 없습니다")
         
-        # 하나의 긴 숫자인 경우
         if len(numbers) == 1:
             num_str = numbers[0]
             if len(num_str) == 8:  # YYYYMMDD
@@ -81,8 +79,6 @@ class SajuCalculator:
                 day = int(num_str[4:6])
             else:
                 raise ValueError(f"날짜 형식을 인식할 수 없습니다: {num_str}")
-        
-        # 여러 숫자로 분리된 경우
         elif len(numbers) >= 3:
             year = int(numbers[0])
             month = int(numbers[1])
@@ -120,14 +116,11 @@ class SajuCalculator:
             hour = int(numbers[0])
             minute = int(numbers[1]) if len(numbers) > 1 else 0
         
-        # 24시간 형식 조정
         hour = hour % 24
-        
         return hour, minute
 
     def get_ganzhi(self, year, month, day, hour):
         """간지 계산"""
-        # 기준일: 1900년 1월 1일
         base_date = datetime.datetime(1900, 1, 1)
         target_date = datetime.datetime(year, month, day)
         
@@ -215,54 +208,7 @@ def get_element_balance_advice(elements):
     
     return advice
 
-def calculate_saju(birth_date, birth_time, gender, birth_place):
-    """사주 계산 메인 함수"""
-    # SajuCalculator 인스턴스 생성
-    calculator = SajuCalculator()
-    
-    try:
-        # 입력 검증
-        if not birth_date:
-            return "❌ 생년월일을 입력해주세요."
-        
-        # 날짜 파싱
-        year, month, day = calculator.parse_date(birth_date)
-        
-        # 시간 파싱
-        hour, minute = calculator.parse_time(birth_time)
-        
-        # 날짜 유효성 검사
-        if year < 1900 or year > 2100:
-            return f"❌ 연도는 1900~2100 사이여야 합니다. 입력된 연도: {year}"
-        if month < 1 or month > 12:
-            return f"❌ 월은 1~12 사이여야 합니다. 입력된 월: {month}"
-        if day < 1 or day > 31:
-            return f"❌ 일은 1~31 사이여야 합니다. 입력된 일: {day}"
-        
-        # datetime 객체 생성
-        birth_datetime = datetime.datetime(year, month, day, hour, minute)
-        
-        # 간지 계산
-        ganzhi = calculator.get_ganzhi(year, month, day, hour)
-        
-        # 오행 분석
-        elements = calculator.analyze_elements(ganzhi)
-        
-        # 십신 분석
-        ten_gods = calculator.get_ten_gods_analysis(ganzhi)
-        
-        # 결과 포맷팅
-        result = format_saju_result(ganzhi, elements, ten_gods, birth_datetime, gender, birth_place, calculator)
-        
-        return result
-        
-    except ValueError as ve:
-        return f"❌ 입력 오류: {str(ve)}\n\n💡 입력 예시:\n- 생년월일: 19900207, 1990-2-7\n- 시간: 0815, 8:15"
-    
-    except Exception as e:
-        return f"❌ 계산 중 오류가 발생했습니다: {str(e)}"
-
-def format_saju_result(ganzhi, elements, ten_gods, birth_datetime, gender, birth_place, calculator, original_time, corrected_time, location_offset):
+def format_saju_result(calculator, ganzhi, elements, ten_gods, birth_info):
     """사주 결과 포맷팅 - 표 형태로 출력"""
     
     # 오행별 색상 설정
@@ -287,21 +233,21 @@ def format_saju_result(ganzhi, elements, ten_gods, birth_datetime, gender, birth
     
     # 시간 보정 정보
     time_correction_info = ""
-    if location_offset != 0:
-        sign = "+" if location_offset > 0 else ""
+    if birth_info['location_offset'] != 0:
+        sign = "+" if birth_info['location_offset'] > 0 else ""
         time_correction_info = f"""
 ### ⏰ 출생지 시간 보정
-- **입력 시간**: {original_time}
-- **보정 시간**: {corrected_time} ({birth_place} 기준 {sign}{location_offset}분 보정)
+- **입력 시간**: {birth_info['original_time']}
+- **보정 시간**: {birth_info['corrected_time']} ({birth_info['birth_place']} 기준 {sign}{birth_info['location_offset']}분 보정)
 """
     
     result = f"""
 # 🔮 사주명리 만세력 분석결과
 
 ## 📋 기본정보
-- **생년월일**: {birth_datetime.strftime('%Y년 %m월 %d일')}
-- **출생시간**: {corrected_time} ({birth_place})
-- **성별**: {gender}
+- **생년월일**: {birth_info['birth_datetime'].strftime('%Y년 %m월 %d일')}
+- **출생시간**: {birth_info['corrected_time']} ({birth_info['birth_place']})
+- **성별**: {birth_info['gender']}
 {time_correction_info}
 ## 🏛️ 사주(四柱) 만세력표
 
@@ -372,6 +318,83 @@ def format_saju_result(ganzhi, elements, ten_gods, birth_datetime, gender, birth
     
     return result
 
+def calculate_saju(birth_date, birth_time, gender, birth_place):
+    """사주 계산 메인 함수"""
+    # SajuCalculator 인스턴스 생성
+    calculator = SajuCalculator()
+    
+    try:
+        # 입력 검증
+        if not birth_date:
+            return "❌ 생년월일을 입력해주세요."
+        
+        # 날짜 파싱
+        year, month, day = calculator.parse_date(birth_date)
+        
+        # 시간 파싱
+        hour, minute = calculator.parse_time(birth_time)
+        
+        # 출생지에 따른 시간 보정
+        location_offset = calculator.get_location_offset(birth_place)
+        
+        # 보정된 시간 계산
+        corrected_minute = minute + location_offset
+        corrected_hour = hour
+        
+        # 분이 60을 넘거나 0 미만인 경우 시간 조정
+        if corrected_minute >= 60:
+            corrected_hour += corrected_minute // 60
+            corrected_minute = corrected_minute % 60
+        elif corrected_minute < 0:
+            corrected_hour -= (-corrected_minute - 1) // 60 + 1
+            corrected_minute = 60 + (corrected_minute % 60)
+        
+        # 시간이 24를 넘거나 0 미만인 경우 조정
+        corrected_hour = corrected_hour % 24
+        
+        # 날짜 유효성 검사
+        if year < 1900 or year > 2100:
+            return f"❌ 연도는 1900~2100 사이여야 합니다. 입력된 연도: {year}"
+        if month < 1 or month > 12:
+            return f"❌ 월은 1~12 사이여야 합니다. 입력된 월: {month}"
+        if day < 1 or day > 31:
+            return f"❌ 일은 1~31 사이여야 합니다. 입력된 일: {day}"
+        
+        # datetime 객체 생성 (보정된 시간 사용)
+        birth_datetime = datetime.datetime(year, month, day, corrected_hour, corrected_minute)
+        original_time = f"{hour:02d}:{minute:02d}"
+        corrected_time = f"{corrected_hour:02d}:{corrected_minute:02d}"
+        
+        # 출생 정보 딕셔너리
+        birth_info = {
+            'birth_datetime': birth_datetime,
+            'gender': gender,
+            'birth_place': birth_place,
+            'original_time': original_time,
+            'corrected_time': corrected_time,
+            'location_offset': location_offset
+        }
+        
+        # 간지 계산
+        ganzhi = calculator.get_ganzhi(year, month, day, corrected_hour)
+        
+        # 오행 분석
+        elements = calculator.analyze_elements(ganzhi)
+        
+        # 십신 분석
+        ten_gods = calculator.get_ten_gods_analysis(ganzhi)
+        
+        # 결과 포맷팅
+        result = format_saju_result(calculator, ganzhi, elements, ten_gods, birth_info)
+        
+        return result
+        
+    except ValueError as ve:
+        return f"❌ 입력 오류: {str(ve)}\n\n💡 입력 예시:\n- 생년월일: 19851015, 1985-10-15\n- 시간: 1430, 14:30"
+    
+    except Exception as e:
+        return f"❌ 계산 중 오류가 발생했습니다: {str(e)}"
+
 def create_interface():
     """Gradio 인터페이스 생성"""
     with gr.Blocks(title="🔮 사주명리 만세력 시스템") as demo:
@@ -429,8 +452,6 @@ def create_interface():
             outputs=result_output
         )
         
-        # 버튼 클릭 시에만 작동하도록 변경 (자동 로드 제거)
-        
         gr.HTML("""
         <div style="text-align: center; padding: 20px; margin-top: 30px; border-top: 1px solid #eee;">
             <p><small>※ 본 시스템은 전통 사주명리학을 기반으로 하며, 참고용으로만 활용해주시기 바랍니다.</small></p>