Update main.py

main.py

@@ -2,24 +2,17 @@ import json, os
 import gradio as gr
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi import FastAPI, Request,  Header, BackgroundTasks, HTTPException, status
-# import google.generativeai as genai
 from google import genai
 from linebot import LineBotApi, WebhookHandler
 from linebot.exceptions import InvalidSignatureError
 from linebot.models import MessageEvent, TextMessage, TextSendMessage, ImageSendMessage, AudioMessage
 
 # 設定 Google AI API 金鑰
-# genai.configure(api_key=os.environ["GOOGLE_API_KEY"])
 client = genai.Client(api_key=os.getenv("GOOGLE_API_KEY"))
 
 # 設定生成文字的參數
 generation_config = genai.types.GenerateContentConfig(max_output_tokens=2048, temperature=0.2, top_p=0.5, top_k=16)
 
-# 使用 Gemini-2.0-flash 模型
-# model = genai.GenerativeModel('gemini-2.0-flash', 
-#                               generation_config=generation_config,
-#                               system_instruction="你是博通古今的萬應機器人！")
-
 # 設定 Line Bot 的 API 金鑰和秘密金鑰
 line_bot_api = LineBotApi(os.environ["CHANNEL_ACCESS_TOKEN"])
 line_handler = WebhookHandler(os.environ["CHANNEL_SECRET"])
@@ -88,10 +81,9 @@ def handle_message(event):
        
     # 檢查是否正在與使用者交談
     elif working_status:
-        # try: 
-            # 使用 model.start_chat 來初始化 user_id 的對話
+        try: 
             user_id = event.source.user_id
-            chat = chat_sessions.get(user_id) or client.chats.create(model="gemini-2.0-flash", config=generation_config) #model.start_chat(history=[])
+            chat = chat_sessions.get(user_id) or client.chats.create(model="gemini-2.0-flash", config=generation_config)
             chat_sessions[user_id] = chat
             # 取得使用者輸入的文字
             user_input = event.message.text
@@ -102,14 +94,14 @@ def handle_message(event):
                 out = response.text
             else:
                 out = "Gemini沒答案!請換個說法！"
-        # except:
-        #     # 處理錯誤
-        #     out = "Gemini執行出錯!請換個說法！" 
+        except:
+            # 處理錯誤
+            out = "Gemini執行出錯!請換個說法！" 
   
-            # 回覆生成結果
-            line_bot_api.reply_message(
-                event.reply_token,
-                TextSendMessage(text=out))
+        # 回覆生成結果
+        line_bot_api.reply_message(
+            event.reply_token,
+            TextSendMessage(text=out))
 
 if __name__ == "__main__":
     # 啟動 FastAPI 應用程式