Update main.py

main.py

@@ -2,7 +2,7 @@ import json, os, glob, pathlib, time, re
 from fastapi import FastAPI, Request, Header, BackgroundTasks, HTTPException
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.responses import HTMLResponse, JSONResponse
-from google import genai
+import google.generativeai as genai
 from linebot import LineBotApi, WebhookHandler
 from linebot.exceptions import InvalidSignatureError
 from linebot.models import MessageEvent, TextMessage, TextSendMessage
@@ -129,7 +129,7 @@ ai_enabled = False
 
 if GOOGLE_API_KEY:
     try:
-        client = genai.Client(api_key=GOOGLE_API_KEY)
+        genai.configure(api_key=GOOGLE_API_KEY)
         ai_enabled = True
         logger.info("✅ Gemini AI 已啟用")
     except Exception as e:
@@ -575,7 +575,10 @@ def handle_message(event):
     if ai_enabled:
         try:
             prompt = f"參考資料：{pdf_content}\n\n問題：{user_message}\n\n簡潔回答。" if pdf_content else user_message
-            response = client.models.generate_content(model="gemini-2.5-flash", contents=prompt)
+            
+            model = genai.GenerativeModel('gemini-2.5-flash')
+            response = model.generate_content(prompt)
+            
             out = response.text if response and response.text else "無法回答"
         except Exception as e:
             logger.error(f"AI錯誤: {str(e)}")