Update main.py

main.py

@@ -12,6 +12,9 @@ import PIL.Image
 import tempfile
 import httpx
 
+from imgurpython import ImgurClient
+
+
 # 設定 Google AI API 金鑰
 genai.configure(api_key=os.environ["GOOGLE_API_KEY"])
 
@@ -25,6 +28,12 @@ model = genai.GenerativeModel('gemini-2.0-flash-exp', system_instruction="主要
 line_bot_api = LineBotApi(os.environ["CHANNEL_ACCESS_TOKEN"])
 line_handler = WebhookHandler(os.environ["CHANNEL_SECRET"])
 
+# 設定 Imgur 的 API
+client_id = os.environ.get("IMGUR_CLIENT_ID")
+client_secret = os.environ.get("IMGUR_CLIENT_SECRET")
+access_token = os.environ.get("IMGUR_ACCESS_TOKEN")
+refresh_token = os.environ.get("IMGUR_REFRESH_TOKEN")
+
 # 設定是否正在與使用者交談
 working_status = os.getenv("DEFALUT_TALKING", default = "true").lower() == "true"
 
@@ -67,6 +76,30 @@ async def webhook(
 #==========================
 # 使用者請求生成圖片
 #==========================
+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 需要檔案路徑
+    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)
+
+    return image['link']
+
+# 使用 Gemini 生成圖片
+def generate_image_with_gemini(prompt):
+    image_model = genai.GenerativeModel("gemini-2.0-flash-exp-image-generation")
+    response = image_model.generate_content(prompt)
+    if response.parts and hasattr(response.parts[0], "inline_data"):
+        return response.parts[0].inline_data.data
+    return None
+
 def upload_image_to_imgur_with_token(image_binary, access_token):
     """
     使用 Imgur OAuth2 Access Token 上傳圖片（直接接收 binary，不需存檔）
@@ -158,25 +191,15 @@ def handle_image_message(event):
     if user_text and user_text.startswith("請幫我生成圖片"):
         prompt = user_text.replace("請幫我生成圖片", "").strip()
 
-        image_model = genai.GenerativeModel("gemini-2.0-flash-exp-image-generation")
-        response = image_model.generate_content(prompt)
-
-        if response.parts and hasattr(response.parts[0], "inline_data"):
-            image_data = response.parts[0].inline_data.data
-            access_token = os.environ.get("IMGUR_ACCESS_TOKEN")
-            image_url = upload_image_to_imgur_with_token(image_data,access_token)
-            print("Imgur 回傳 URL：", image_url)
+        client = ImgurClient(client_id, client_secret, access_token, refresh_token)
+        image_binary = generate_image_with_gemini(prompt)
 
-        if image_url:
-            line_bot_api.reply_message(
-                event.reply_token,
-                [
-                    TextSendMessage(text="這是我為你生成的圖片喔～ ✨"),
-                    ImageSendMessage(original_content_url=image_url, preview_image_url=image_url)
-                ]
-            )
+        if image_binary:
+            album = "nvsYwgq"  # 你的相簿ID
+            image_url = upload_image_to_imgur(client, image_binary, album)
+            print(f"圖片網址: {image_url}")
         else:
-            line_bot_api.reply_message(event.reply_token, TextSendMessage(text="圖片生成失敗，請稍後再試~"))
+            print("Gemini 圖片生成失敗")
         return
 
     if event.message.type == "image":