Update main.py

main.py

@@ -106,8 +106,8 @@ def upload_image_to_imgur(client, image_binary, album=None, name="gemini-image",
     return uploaded_image['link']
 
 # 使用 Gemini 生成圖片
+client = genai.Client(api_key=os.environ["GOOGLE_API_KEY"])
 def generate_image_with_gemini(prompt):
-    client = genai.Client(api_key=os.environ["GOOGLE_API_KEY"])
     response = client.models.generate_content(model="gemini-2.0-flash-exp-image-generation",contents=prompt,config=types.GenerateContentConfig(response_modalities=['Text', 'Image']))
 
     for part in response.candidates[0].content.parts: