Update main.py

main.py

@@ -1,10 +1,11 @@
 import json, os
+import io
 import gradio as gr
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi import FastAPI, Request,  Header, BackgroundTasks, HTTPException, status
 import google.generativeai as genai
-import mimetypes
 import base64
+from datetime import datetime
 from collections import defaultdict
 from linebot import LineBotApi, WebhookHandler
 from linebot.exceptions import InvalidSignatureError
@@ -77,11 +78,6 @@ async def webhook(
 #==========================
 # 使用者請求生成圖片
 #==========================
-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"):
     # 將 binary 資料轉為 PIL Image
     image = Image.open(io.BytesIO(image_binary))
@@ -185,24 +181,36 @@ def handle_image_message(event):
     if user_text and user_text.startswith("請幫我生成圖片"):
         prompt = user_text.replace("請幫我生成圖片", "").strip()
 
+         # 先立即回覆避免token過期
+        line_bot_api.reply_message(event.reply_token, TextSendMessage(text="圖片生成中～ 請稍候.....✨"))
+
         client = ImgurClient(client_id, client_secret, access_token, refresh_token)
         image_binary = generate_image_with_gemini(prompt)
-        line_bot_api.reply_message(event.reply_token, TextSendMessage(text="圖片生成中～ 請稍候約10-20秒✨"))
 
         if image_binary:
-            album = "nvsYwgq"  # 你的相簿ID
+            album = "nvsYwgq"
             image_url = upload_image_to_imgur(client, image_binary, album)
+
             if image_url:
-                # 使用 push 訊息，避免回覆過期
+                # 使用 push message 發送圖片，避免 reply token 超時
                 line_bot_api.push_message(
-                event.source.user_id,
-                [TextSendMessage(text="✨ 這是我為你生成的圖片喔～"),ImageSendMessage(original_content_url=image_url,preview_image_url=image_url)]
+                    event.source.user_id,
+                    [
+                        TextSendMessage(text="✨ 這是我為你生成的圖片喔～"),
+                        ImageSendMessage(original_content_url=image_url, preview_image_url=image_url)
+                    ]
                 )
-    else:
-        line_bot_api.push_message(
-            event.source.user_id,
-            TextSendMessage(text="⚠️ 圖片上傳失敗，請稍後再試~")
-        )
+            else:
+                line_bot_api.push_message(
+                    event.source.user_id,
+                    TextSendMessage(text="⚠️ 圖片上傳失敗，請稍後再試~")
+                )
+        else:
+            line_bot_api.push_message(
+                event.source.user_id,
+                TextSendMessage(text="⚠️ 圖片生成失敗，請稍後再試~")
+            )
+        return
 
 
     if event.message.type == "image":