完成西元農曆轉換

app.py

@@ -3,6 +3,25 @@
 import os
 import openai
 import gradio as gr
+from zhdate import ZhDate
+from datetime import datetime
+
+# 定義地支對應的時間區間
+z_mapping = {
+        "1": "子",
+        "2": "丑",
+        "3": "寅",
+        "4": "卯",
+        "5": "辰",
+        "6": "巳",
+        "7": "午",
+        "8": "未",
+        "9": "申",
+        "10": "酉",
+        "11": "戌",
+        "12": "亥",
+        "0": "夜子"
+    } 
 
 ## AI 建議
 def get_advice(bmi,temp, API_KEY, model="gpt-3.5-turbo"):
@@ -33,21 +52,18 @@ def get_gym(bmi,slide, temp, API_KEY, model="gpt-3.5-turbo"):
     )
     return response.choices[0].message.content
 
-def BMI(year, month) -> int:
-    year = int(year) / 100
-    bmi = int(month) / (year * year)
-    if bmi < 18.5:
-        return str(bmi)[:5], "過輕"
-    elif bmi < 24:
-        return str(bmi)[:5], "正常"
-    elif bmi < 27:
-        return str(bmi)[:5], "過重"
-    elif bmi < 30:
-        return str(bmi)[:5], "輕度肥胖"
-    elif bmi < 35:
-        return str(bmi)[:5], "中度肥胖"
+def BMI(year, month, day, btime) -> datetime:
+    dt_date1 = datetime(int(year), int(month), int(day))
+    date_lunar = ZhDate.from_datetime(dt_date1)
+    bstr = str(date_lunar)
+    
+    hour = int(btime.split(":")[0])
+    if hour >= 23:
+        z = 0
     else:
-        return str(bmi)[:5], "重度肥胖"
+        z = int(hour / 2) + 1
+    result=z_mapping[str(z)]
+    return bstr.replace('农历','農曆') + " " + result + "時"
 
 
 
@@ -67,9 +83,9 @@ day = gr.Textbox(
     info="輸入你的出生日子",
     placeholder="Input your birthday day here...",)
 
-time = gr.Textbox(
+btime = gr.Textbox(
     label="生時",
-    info="輸入你的出生時間",
+    info="輸入你的出生時間 HH:MM (幾點及約略幾分即可，請用24小時制)",
     placeholder="Input your birth time here...",)
 
 sex = gr.Textbox(
@@ -132,15 +148,15 @@ temperature = gr.Slider(
 
 with gr.Blocks() as demo:
     gr.Markdown("""
-    # BMI 計算器
-    簡易測量你的BMI值
+    # AI 紫微
+    起手式 - 轉換西元生日到農曆出生時辰
     """)
     with gr.Column():
         with gr.Row():
             year.render()  # 顯示年
             month.render() # 顯示月
             day.render() # 顯示日
-            time.render() # 顯示時
+            btime.render() # 顯示時
             sex.render() # 顯示性別
         
         with gr.Row():
@@ -148,7 +164,7 @@ with gr.Blocks() as demo:
 
         btn.render() # 顯示計算BMI值按鈕
         btn.click(fn=BMI, 
-                     inputs=[year, month], 
+                     inputs=[year, month, day, btime], 
                      outputs=[output_bmi])
 
         advice.render() # 顯示AI建議結果的文字框