Update main.py

main.py

@@ -77,6 +77,26 @@ async def webhook(
 #==========================
 # 使用者請求生成圖片
 #==========================
+def upload_image_to_imgur_direct(image_binary, client_id, album=None, name="gemini-image", title="Gemini Generated Image"):
+    headers = {
+        "Authorization": f"Client-ID {client_id}",
+    }
+    data = {
+        'image': base64.b64encode(image_binary).decode('utf-8'),
+        'album': album,
+        'type': 'base64',
+        'name': name,
+        'title': title,
+        'description': f'Generated by Gemini - {datetime.now()}'
+    }
+
+    response = httpx.post("https://api.imgur.com/3/image", headers=headers, data=data)
+    if response.status_code == 200:
+        return response.json()["data"]["link"]
+    else:
+        print("Imgur 上傳失敗：", response.text)
+        return None
+
 def upload_image_to_imgur(client, image_binary, album=None, name="gemini-image", title="Gemini Generated Image"):
     config = {
         'album': album,
@@ -96,35 +116,14 @@ def upload_image_to_imgur(client, image_binary, album=None, name="gemini-image",
 # 使用 Gemini 生成圖片
 def generate_image_with_gemini(prompt):
     model_name = "gemini-2.0-flash-exp-image-generation"
+    image_model = genai.GenerativeModel(model_name)
 
-    client = genai.GenerativeModel(model_name)
-
-    contents = [genai.types.Content(
-        role="user",
-        parts=[genai.types.Part.from_text(prompt)]
-    )]
-
-    generate_content_config = genai.types.GenerateContentConfig(
-        temperature=1,
-        top_p=0.95,
-        top_k=40,
-        max_output_tokens=1000,
-        response_modalities=["image", "text"],
-        response_mime_type="text/plain",
-    )
-
-    for chunk in client.generate_content_stream(
-        contents=contents,
-        generation_config=generate_content_config,
-    ):
-        if not chunk.candidates or not chunk.candidates[0].content.parts:
-            continue
-        if chunk.candidates[0].content.parts[0].inline_data:
-            inline_data = chunk.candidates[0].content.parts[0].inline_data
-            file_extension = mimetypes.guess_extension(inline_data.mime_type) or ".png"
-            return inline_data.data, file_extension
-
-    return None, None
+    response = image_model.generate_content(prompt)
+
+    if response.parts and hasattr(response.parts[0], "inline_data"):
+        return response.parts[0].inline_data.data  # 直接回傳圖片binary資料
+
+    return None
 
 #==========================
 # 使用者上傳圖片
@@ -199,7 +198,7 @@ def handle_image_message(event):
 
         if image_binary:
             album = "nvsYwgq"  # 你的相簿ID
-            image_url = upload_image_to_imgur(client, image_binary, album)
+            image_url = upload_image_to_imgur_direct(client, image_binary, album)
             line_bot_api.reply_message(
             event.reply_token,
             [