Update Image_generation.py

Image_generation.py

@@ -7,12 +7,12 @@ from google import genai
 from google.genai import types
 from imgurpython import ImgurClient
 
+genai_client = genai.Client(api_key=os.environ["GOOGLE_API_KEY"])
+
 class ImageGenerator:
-    def __init__(self, google_api_key, imgur_client_id, imgur_client_secret, imgur_access_token, imgur_refresh_token):
-        # 初始化 Gemini 客戶端
-        self.genai_client = genai.Client(api_key=google_api_key)
+    def __init__(self, imgur):
         # 初始化 Imgur 客戶端
-        self.imgur_client = ImgurClient(imgur_client_id, imgur_client_secret, imgur_access_token, imgur_refresh_token)
+        self.imgur_client = imgur
 
     def generate_image_with_gemini(self, prompt):
         """
@@ -24,7 +24,7 @@ class ImageGenerator:
         返回:
         bytes: 生成的圖片的二進位資料，如果生成失敗則返回 None。
         """
-        response = self.genai_client.models.generate_content(
+        response = genai_client.models.generate_content(
             model="gemini-2.0-flash-exp-image-generation",
             contents=prompt,
             config=types.GenerateContentConfig(response_modalities=['Text', 'Image'])