Update main.py

main.py

@@ -12,8 +12,7 @@ import io
 import PIL.Image
 
 from imgurpython import ImgurClient
-from Image_generation import generate_image_with_gemini, upload_image_to_imgur
-
+from Image_generation import ImageGenerator
 from Uploading_images import get_image_url, store_user_message, analyze_with_gemini, get_previous_message
 
 #==========================
@@ -101,14 +100,13 @@ def handle_image_message(event):
 
          # 先立即回覆避免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)
+        image_binary = image_generator.generate_image_with_gemini(prompt)
 
         if image_binary:
             album = "nvsYwgq"
-            image_url = upload_image_to_imgur(client, image_binary, album)
+            image_url = image_generator.upload_image_to_imgur(image_binary,album)
 
             if image_url:
                 # 使用 push message 發送圖片，避免 reply token 超時