生成影片

image_examples.py

@@ -136,6 +136,15 @@ def handle_image_message(event):
                         "label": "吉卜力風格圖片",
                         "data": f"ghibli,{filename}"
                     }
+                },
+                {
+                    "type": "button",
+                    "style": "primary",
+                    "action": {
+                        "type": "postback",
+                        "label": "生成影片",
+                        "data": f"veo,{filename}"
+                    }
                 }
             ]
         }
@@ -180,6 +189,23 @@ def handle_postback(event):
                     ]
                 )
             )
+    elif data.startswith("veo,"):
+        _, filename = data.split(",", 1)
+        image_dir = os.path.join(os.getcwd(), "images")
+        image_path = os.path.join(image_dir, filename)
+        # 呼叫 Veo 影片生成
+        video_path = generate_video_from_image(image_path)
+        video_url = f"https://{base_url}/images/{os.path.basename(video_path)}"
+        with ApiClient(configuration) as api_client:
+            MessagingApi(api_client).reply_message(
+                ReplyMessageRequest(
+                    reply_token=event.reply_token,
+                    messages=[
+                        TextMessage(text="這是根據圖片生成的影片："),
+                        TextMessage(text=video_url)
+                    ]
+                )
+            )
 
 def convert_to_ghibli_style(image_path):
     from PIL import Image
@@ -204,4 +230,32 @@ def convert_to_ghibli_style(image_path):
             out_path = image_path.replace(".jpg", "_ghibli.jpg")
             out_img.save(out_path)
             return out_path
-    return image_path  # 若失敗則回傳原圖
+    return image_path  # 若失敗則回傳原圖
+
+def generate_video_from_image(image_path):
+    import io
+    from PIL import Image
+
+    # 讀取圖片
+    with open(image_path, "rb") as f:
+        img_bytes = f.read()
+    image = Image.open(io.BytesIO(img_bytes))
+
+    prompt = "請根據這張圖片生成一段有創意的短影片，內容需與圖片主題高度相關，回傳影片檔案。"
+    response = client.models.generate_content(
+        model="veo-2.0-generate-001",
+        contents=[image, prompt],
+        config=types.GenerateContentConfig(
+            response_modalities=["VIDEO", "TEXT"]
+        ),
+    )
+    # 取得 Gemini Veo 回傳的影片
+    for part in response.candidates[0].content.parts:
+        if part.inline_data is not None:
+            video_bytes = part.inline_data.data
+            out_path = image_path.replace(".jpg", "_veo.mp4")
+            with open(out_path, "wb") as f:
+                f.write(video_bytes)
+            return out_path
+    # 若失敗則回傳原圖
+    return image_path