Create Image_generation.py

Image_generation.py

@@ -0,0 +1,46 @@
+import os
+import io
+import PIL.Image
+from google import genai
+from google.genai import types
+from datetime import datetime
+
+# 使用 Gemini 生成圖片
+client = genai.Client(api_key=os.environ["GOOGLE_API_KEY"])
+
+def generate_image_with_gemini(prompt):
+    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:
+        if part.text is not None:
+            print(part.text)
+        elif part.inline_data is not None:
+            return part.inline_data.data
+
+    return
+
+def upload_image_to_imgur(client, image_binary, album=None, name="gemini-image", title="gemini Generated Image"):
+    # 將 binary 資料轉為 PIL Image
+    image = PIL.Image.open(io.BytesIO(image_binary))
+
+    # 建立暫存檔案來上傳 (因為 ImgurClient 需要檔案路徑)
+    with tempfile.NamedTemporaryFile(suffix=".png", delete=True) as tmp:
+        image.save(tmp.name, format='PNG')
+
+        # 準備上傳資訊
+        config = {
+            'album': album,
+            'name': name,
+            'title': title,
+            'description': f'Generated by gemini - {datetime.now()}'
+        }
+
+        # 使用 client 進行圖片上傳
+        uploaded_image = client.upload_from_path(tmp.name, config=config, anon=False)
+
+    # 回傳圖片網址
+    return uploaded_image['link']