Update main.py

main.py

@@ -19,7 +19,8 @@ from Uploading_images import get_image_url, store_user_message, analyze_with_gem
 #==========================
 
 # 設定 Google AI API 金鑰
-genai_gen.configure(api_key=os.environ["GOOGLE_API_KEY"])
+google_api = os.environ["GOOGLE_API_KEY"]
+genai_gen.configure(api_key=google_api)
 
 # 設定生成文字的參數
 generation_config = genai_gen.types.GenerationConfig(max_output_tokens=1000, temperature=0.2, top_p=0.5, top_k=16) #2048
@@ -37,7 +38,6 @@ 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"
 
@@ -99,7 +99,7 @@ def handle_image_message(event):
 
          # 先立即回覆避免token過期
         line_bot_api.reply_message(event.reply_token, TextSendMessage(text="圖片生成中～ 請稍候.....✨"))
-
+        image_generator = ImageGenerator( google_api, client_id, client_secret, access_token, refresh_token)
         # 生成圖片
         image_binary = image_generator.generate_image_with_gemini(prompt)