換成GOOGLE格式

image_examples.py

@@ -239,35 +239,61 @@ def generate_video_from_image(image_path):
 
     prompt = "請根據這張圖片生成一段有創意的短影片，內容需與圖片主題高度相關。"
 
-    # 嘗試直接傳 bytes，若不行請參考官方文件
+    # 1. 讀取本地圖片的二進制數據
     with open(image_path, "rb") as img_file:
         img_bytes = img_file.read()
 
-    # 嘗試用 dict 方式（有些 SDK 版本支援）
-    image_obj = {"data": img_bytes, "mime_type": "image/jpeg"}
-
-    operation = client.models.generate_videos(
-        model="veo-2.0-generate-001",
-        prompt=prompt,
-        image=image_obj,
-        config=types.GenerateVideosConfig(
-            person_generation="dont_allow",
-            aspect_ratio="16:9",
-            number_of_videos=1
-        ),
-    )
+    # 2. 判斷圖片的 MIME type (這裡假設 .jpg 都是 image/jpeg)
+    #    更可靠的方式是使用 imghdr 庫或類似方法來檢測
+    mime_type = "image/jpeg"
+    if image_path.lower().endswith(".png"):
+        mime_type = "image/png"
+    # Add more types if needed
+
+    # 3. 建立 GeminiImage 物件
+    #    使用 image_bytes 參數來傳遞圖片數據
+    try:
+        gemini_image_obj = GeminiImage(image_bytes=img_bytes, mime_type=mime_type)
+    except Exception as e:
+        print(f"[ERROR] Failed to create GeminiImage object: {e}")
+        return image_path # Fallback: return original path if image object creation fails
 
-    while not operation.done:
-        time.sleep(10)
-        operation = client.operations.get(operation)
+    print(f"[LOG] Attempting to generate video with image: {image_path}, mime_type: {mime_type}")
 
-    image_dir = os.path.join(os.getcwd(), "images")
-    for n, video in enumerate(operation.response.generated_videos):
-        fname = f"{os.path.splitext(os.path.basename(image_path))[0]}_veo{n}.mp4"
-        save_path = os.path.join(image_dir, fname)
-        video.video.save(save_path)
-        print(f"[LOG] 影片已儲存：{save_path}")
-        return save_path
-
-    print("[LOG] 未產生影片，回傳原圖")
+    try:
+        operation = client.models.generate_videos(
+            model="veo-2.0-generate-001",
+            prompt=prompt,
+            image=gemini_image_obj, # 傳遞 GeminiImage 物件
+            config=types.GenerateVideosConfig(
+                person_generation="dont_allow",
+                aspect_ratio="16:9",
+                number_of_videos=1
+            ),
+        )
+
+        while not operation.done:
+            print("[LOG] Waiting for video generation to complete...")
+            time.sleep(10) # 官方範例用 20 秒，可依需求調整
+            operation = client.operations.get(operation)
+
+        image_dir = os.path.join(os.getcwd(), "images")
+        if operation.response and operation.response.generated_videos:
+            for n, video in enumerate(operation.response.generated_videos):
+                fname = f"{os.path.splitext(os.path.basename(image_path))[0]}_veo{n}.mp4"
+                save_path = os.path.join(image_dir, fname)
+                video.video.save(save_path)
+                print(f"[LOG] 影片已儲存：{save_path}")
+                return save_path
+        else:
+            print("[LOG] Video generation completed but no videos were returned in the response.")
+            if operation.error:
+                print(f"[ERROR] Video generation failed with error: {operation.error.message}")
+            return image_path
+
+    except Exception as e:
+        print(f"[ERROR] Exception during video generation or saving: {e}")
+        return image_path # Fallback
+
+    print("[LOG] 未產生影片或處理出錯，回傳原圖路徑")
     return image_path