Update main.py

main.py

@@ -4,8 +4,10 @@ import io
 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.generativeai import types
+import google.generativeai as genai_gen
+from google import genai
+from google.genai import types
+
 import base64
 from collections import defaultdict
 from linebot import LineBotApi, WebhookHandler
@@ -25,13 +27,13 @@ from transformers import pipeline
 
 
 # 設定 Google AI API 金鑰
-genai.configure(api_key=os.environ["GOOGLE_API_KEY"])
+genai_gen.configure(api_key=os.environ["GOOGLE_API_KEY"])
 
 # 設定生成文字的參數
-generation_config = genai.types.GenerationConfig(max_output_tokens=1000, temperature=0.2, top_p=0.5, top_k=16) #2048
+generation_config = genai_gen.types.GenerationConfig(max_output_tokens=1000, temperature=0.2, top_p=0.5, top_k=16) #2048
 
 # 使用 Gemini-1.5-flash 模型
-model = genai.GenerativeModel('gemini-2.0-flash-exp', system_instruction="主要用繁體中文回答，但如果用戶使用詢問英文問題，就用英文回應。你現在是個專業助理，職稱為OPEN小助理，個性活潑、樂觀，願意回答所有問題", generation_config=generation_config)
+model = genai_gen.GenerativeModel('gemini-2.0-flash-exp', system_instruction="主要用繁體中文回答，但如果用戶使用詢問英文問題，就用英文回應。你現在是個專業助理，職稱為OPEN小助理，個性活潑、樂觀，願意回答所有問題", generation_config=generation_config)
 
 # 設定 Line Bot 的 API 金鑰和秘密金鑰
 line_bot_api = LineBotApi(os.environ["CHANNEL_ACCESS_TOKEN"])
@@ -110,21 +112,22 @@ def upload_image_to_imgur(client, image_binary, album=None, name="Hugging Face-i
 
 # 使用 Gemini 生成圖片
 def generate_image_with_gemini(prompt):
-    client = genai.Client()
 
+    client = genai.Client(api_key=os.environ["GOOGLE_API_KEY"])
     response = client.models.generate_content(
-        model="gemini-2.0-flash-exp-image-generation",
-        contents=prompt,
-        config=types.GenerateContentConfig(
+            model="gemini-2.0-flash-exp-image-generation",
+            contents=prompt,
+            config=types.GenerateContentConfig(
             response_modalities=['Text', 'Image']
+            )
         )
-    )
 
     for part in response.candidates[0].content.parts:
-        if part.inline_data is not None:
-            return part.inline_data.data  # 回傳 binary 圖片資料
-
-    return None  # 如果無法生成圖片，回傳None
+      if part.text is not None:
+        print(part.text)
+      elif part.inline_data is not None:
+        #image = PIL.Image.open(io.BytesIO(part.inline_data.data))
+        return part.inline_data.data
 
 # 使用 Hugging Face 生成圖片
 def generate_image_hf(prompt):