正確寫法

text2image.py

@@ -1,10 +1,8 @@
 import os
 import logging
-from io import BytesIO
 from PIL import Image
 import gradio as gr
-from google import genai
-from google.genai import types
+import google.generativeai as genai
 
 # 設定 logging
 logging.basicConfig(
@@ -15,41 +13,17 @@ logging.basicConfig(
 
 # 初始化 Gemini API
 GEMINI_API_KEY = os.environ.get("GEMINI_API_KEY")
-client = genai.Client(api_key=GEMINI_API_KEY)
+genai.configure(api_key=GEMINI_API_KEY)
+google_client = genai.GenerativeModel(model_name="gemini-2.0-flash")
 
 # 定義「圖解釋文」功能
 def explain_image(image: Image.Image):
-    # 將 PIL Image 轉成 Gemini 所需的格式
-    buffered = BytesIO()
-    image.save(buffered, format="PNG")
-    image_bytes = buffered.getvalue()
-
-    # 直接建立 types.Content
-    contents = [
-        types.Content(
-            parts=[
-                types.Part(
-                    inline_data=types.Blob(
-                        mime_type="image/png",
-                        data=image_bytes
-                    )
-                ),
-                types.Part(
-                    text="請用繁體中文詳細說明這張圖片的內容。"
-                )
-            ]
-        )
-    ]
-
-    # 呼叫 Gemini 模型
-    response = client.generate_content(
-        model="gemini-1.5-flash",
-        contents=contents,
-        generation_config=types.GenerationConfig(response_mime_type="text/plain")
+    # 直接把 PIL image 傳進去
+    response = google_client.generate_content(
+        contents=[image, "請用繁體中文描述這張圖片"]
     )
-
-    # 回傳第一個回答
-    explanation = response.candidates[0].content.parts[0].text
+    # 取出回答
+    explanation = response.text
     logging.info("圖片說明成功取得。")
     return explanation