Update main.py

main.py

@@ -77,41 +77,33 @@ 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
+import io
+from PIL import Image
+import tempfile
+from datetime import datetime
 
 def upload_image_to_imgur(client, image_binary, album=None, name="gemini-image", title="Gemini Generated Image"):
-    config = {
-        'album': album,
-        'name': name,
-        'title': title,
-        'description': f'Generated by Gemini - {datetime.now()}'
-    }
-
-    # 使用暫存檔案上傳，因為 imgurpython 需要檔案路徑
+    # 將 binary 資料轉為 PIL Image
+    image = Image.open(io.BytesIO(image_binary))
+
+    # 建立暫存檔案來上傳 (因為 ImgurClient 需要檔案路徑)
     with tempfile.NamedTemporaryFile(suffix=".png", delete=True) as tmp:
-        tmp.write(image_binary)
-        tmp.flush()
-        image = client.upload_from_path(tmp.name, config=config, anon=False)
+        image.save(tmp.name, format='PNG')
+
+        # 準備上傳資訊
+        config = {
+            'album': album,
+            'name': name,
+            'title': title,
+            'description': f'Generated by Gemini - {datetime.now()}'
+        }
+
+        # 使用 client 進行圖片上傳
+        uploaded_image = client.upload_from_path(tmp.name, config=config, anon=False)
+
+    # 回傳圖片網址
+    return uploaded_image['link']
 
-    return image['link']
 
 # 使用 Gemini 生成圖片
 def generate_image_with_gemini(prompt):
@@ -199,7 +191,7 @@ def handle_image_message(event):
 
         if image_binary:
             album = "nvsYwgq"  # 你的相簿ID
-            image_url = upload_image_to_imgur_direct(client, image_binary, album)
+            image_url = upload_image_to_imgur(client, image_binary, album)
             if image_url:
                 # 使用 push 訊息，避免回覆過期
                 line_bot_api.push_message(