Update main.py

main.py

@@ -107,13 +107,14 @@ def upload_image_to_imgur(client, image_binary, album=None, name="gemini-image",
 
 # 使用 Gemini 生成圖片
 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'])
-            )
+                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:
         if part.text is not None:
@@ -121,7 +122,7 @@ def generate_image_with_gemini(prompt):
         elif part.inline_data is not None:
             #image = PIL.Image.open(io.BytesIO(part.inline_data.data))
             return part.inline_data.data
-
+    return 
 
 
 #==========================