Update main.py

main.py

@@ -108,15 +108,21 @@ def upload_image_to_imgur(client, image_binary, album=None, name="Hugging Face-i
 
 # 使用 Gemini 生成圖片
 def generate_image_with_gemini(prompt):
-    model_name = "gemini-2.0-flash-exp-image-generation"
-    image_model = genai.GenerativeModel(model_name)
+    client = genai.Client()
 
-    response = image_model.generate_content(prompt)
+    response = client.models.generate_content(
+        model="gemini-2.0-flash-exp-image-generation",
+        contents=prompt,
+        config=types.GenerateContentConfig(
+            response_modalities=['Text', 'Image']
+        )
+    )
 
-    if response.parts and hasattr(response.parts[0], "inline_data"):
-        return response.parts[0].inline_data.data  # 直接回傳圖片binary資料
+    for part in response.candidates[0].content.parts:
+        if part.inline_data is not None:
+            return part.inline_data.data  # 回傳 binary 圖片資料
 
-    return None
+    return None  # 如果無法生成圖片，回傳None
 
 # 使用 Hugging Face 生成圖片
 def generate_image_hf(prompt):
@@ -215,7 +221,7 @@ def handle_image_message(event):
 
         prompt_en = translate_zh_to_en(prompt)
         # 生成圖片
-        image_binary = generate_image_hf(prompt_en)
+        image_binary = generate_image_with_gemini(prompt_en)
 
         if image_binary:
             album = "nvsYwgq"