加入星象的測試

app.py

@@ -6,6 +6,16 @@ import gradio as gr
 from zhdate import ZhDate
 from datetime import datetime
 
+
+from dotenv import load_dotenv, find_dotenv
+_ = load_dotenv(find_dotenv()) # read local .env file
+openai.api_key = os.environ['OPENAI_API_KEY']
+print(openai.api_key)
+if openai.api_key:
+   DESKTOP_KEY = openai.api_key 
+print(DESKTOP_KEY)
+
+
 # 定義地支對應的時間區間
 z_mapping = {
         "1": "子",
@@ -25,13 +35,20 @@ z_mapping = {
 
 ## AI 建議
 def get_advice(bmi,temp, API_KEY, model="gpt-3.5-turbo"):
-    openai.api_key = API_KEY
-    messages = [{"role": "system", "content": "You can provide some dietary advice based on \
-                 the user's BMI value. You can only give up to 3 suggestions"},
-                {"role": "user", "content": f'My BMI is {bmi}. What can I do to be healthier?'},]
+    if API_KEY:
+        openai.api_key = API_KEY
+    else:
+        openai.api_key = DESKTOP_KEY
+    print(openai.api_key)
+    bmi = '' #未來再說
+    gw_main_star = '像一個獨裁霸道的君王，無法聽進別人的意見'
+    gw_main_star = '像一個足智多謀的軍師，ㄧ有時候會考慮太多而失去前進的勇氣'
+
+    messages = [{"role": "system", "content": "你是一個心理權威，你會根據用戶的個性，提供至少3個，不超過5個建議。"},
+                {"role": "user", "content": f'我的個性是 {gw_main_star}. 針對這樣的個性，妳可否給我一些待人處事的建議?'},]
     response = openai.chat.completions.create(
         model=model,
-        max_tokens=200,
+        max_tokens=1000,
         messages=messages,
         temperature=temp, # this is the degree of randomness of the model's output
     )
@@ -39,7 +56,11 @@ def get_advice(bmi,temp, API_KEY, model="gpt-3.5-turbo"):
 
 ## 健身計畫
 def get_gym(bmi,slide, temp, API_KEY, model="gpt-3.5-turbo"):
-    openai.api_key = API_KEY
+    if API_KEY:
+        openai.api_key = API_KEY
+    else:
+        openai.api_key = DESKTOP_KEY
+    print(openai.api_key)
     messages = [{"role": "system", "content": "You are a great fitness coach and \
                  you will give users great fitness plans."},
                 {"role": "user", "content": f'My BMI is {bmi}. I want a {slide}-point weight\
@@ -52,7 +73,7 @@ def get_gym(bmi,slide, temp, API_KEY, model="gpt-3.5-turbo"):
     )
     return response.choices[0].message.content
 
-def BMI(year, month, day, btime) -> datetime:
+def GWAI(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)
@@ -93,7 +114,7 @@ sex = gr.Textbox(
     info="輸入你是男生或女生",
     placeholder="Input you are male or female here...",)
 
-output_bmi = gr.Textbox(
+output_gwai = gr.Textbox(
     value="",
     label="你的農曆生日及出生時辰",
     info="這是從西元換算出的農曆生日及出生時辰",
@@ -110,15 +131,15 @@ btn = gr.Button(
     variant="primary", scale=1)
 
 btn_advice  = gr.Button(
-    value="AI 建議",
+    value="AI 紫微的人生建議",
     variant="primary", scale=2)
 
 btn_gym = gr.Button(
-    value="AI 健身計畫",
+    value="AI 紫微聊天室",
     variant="primary", scale=1)
 
 key_box = gr.Textbox(
-    label="Enter your API key",
+    label="輸入你的 API 金鑰",
     info="You have to provide your own OPENAI_API_KEY for this app to function properly",
     placeholder="Type OpenAI API KEY here...",
     type="password")
@@ -160,21 +181,21 @@ with gr.Blocks() as demo:
             sex.render() # 顯示性別
         
         with gr.Row():
-            output_bmi.render() # 顯示BMI值結果
+            output_gwai.render() # 顯示農曆生日欄
 
-        btn.render() # 顯示計算BMI值按鈕
-        btn.click(fn=BMI, 
+        btn.render() # 顯示農曆生日值結果
+        btn.click(fn=GWAI, 
                      inputs=[year, month, day, btime], 
-                     outputs=[output_bmi])
+                     outputs=[output_gwai])
 
         advice.render() # 顯示AI建議結果的文字框
         
         with gr.Row(): 
             key_box.render() # 顯示API金鑰輸入框
             btn_advice.render() # 顯示AI建議按鈕
-            btn_advice.click(fn=get_advice, inputs=[output_bmi,temperature,key_box], outputs=advice)
+            btn_advice.click(fn=get_advice, inputs=[output_gwai,temperature,key_box], outputs=advice)
             btn_gym.render() # 顯示AI健身計畫按鈕
-            btn_gym.click(fn=get_gym, inputs=[output_bmi,slider,temperature, key_box], outputs=advice)
+            btn_gym.click(fn=get_gym, inputs=[output_gwai,slider,temperature, key_box], outputs=advice)
     
         with gr.Accordion("settings", open=True):
             slider.render()