Update main.py

main.py

@@ -14,8 +14,13 @@ genai.configure(api_key=os.environ["GOOGLE_API_KEY"])
 # 設定生成文字的參數
 generation_config = genai.types.GenerationConfig(max_output_tokens=2048, temperature=0.2, top_p=0.5, top_k=16)
 
-# 使用 Gemini-1.5-flash 模型
-model = genai.GenerativeModel('gemini-1.5-flash', system_instruction="你是聖誕老人，請使用招牌笑聲做開頭，然後以爽朗愉悅的口氣回答問題。") # 或是使用 "你是博通古今的萬應機器人！"
+# 使用 Gemini-2.0-flash 模型
+model = genai.GenerativeModel('gemini-2.0-flash', 
+                              generation_config=generation_config,
+                              system_instruction="你是聖誕老人，請使用招牌笑聲做開頭，然後以爽朗愉悅的口氣回答問題。") # 或是使用 "你是博通古今的萬應機器人！"
+
+# 使用 model.start_chat 來初始化對話
+chat_session = model.start_chat(history=[])                       
 
 # 設定 Line Bot 的 API 金鑰和秘密金鑰
 line_bot_api = LineBotApi(os.environ["CHANNEL_ACCESS_TOKEN"])
@@ -85,14 +90,22 @@ def handle_message(event):
     # 檢查是否正在與使用者交談
     elif working_status:
         try: 
+            # # 取得使用者輸入的文字
+            # prompt = event.message.text
+            # # 使用 Gemini 模型生成文字
+            # completion = model.generate_content(prompt, generation_config=generation_config)
+            # # 檢查生成結果是否為空
+            # if (completion.parts[0].text != None):
+            #     # 取得生成結果
+            #     out = completion.parts[0].text
+
             # 取得使用者輸入的文字
-            prompt = event.message.text
-            # 使用 Gemini 模型生成文字
-            completion = model.generate_content(prompt, generation_config=generation_config)
-            # 檢查生成結果是否為空
-            if (completion.parts[0].text != None):
+            user_input = event.message.text
+            # 發送使用者的輸入到聊天會話並獲得回應
+            response = chat_session.send_message(prompt)
+            if (response.text != None):
                 # 取得生成結果
-                out = completion.parts[0].text
+                out = response.text
             else:
                 # 回覆 "Gemini沒答案!請換個說法！"
                 out = "Gemini沒答案!請換個說法！"