From 439ca8e5edfebee1548fc4b6c63451c4621b4cff Mon Sep 17 00:00:00 2001 From: Mark Daoust Date: Mon, 8 Jul 2024 13:18:23 -0700 Subject: [PATCH] Check video status. (#442) --- samples/text_generation.py | 17 ++++++++++++++--- 1 file changed, 14 insertions(+), 3 deletions(-) diff --git a/samples/text_generation.py b/samples/text_generation.py index 742b96715..a5e800c75 100644 --- a/samples/text_generation.py +++ b/samples/text_generation.py @@ -99,10 +99,21 @@ def test_text_gen_multimodal_audio(self): def test_text_gen_multimodal_video_prompt(self): # [START text_gen_multimodal_video_prompt] + import time + + # Video clip (CC BY 3.0) from https://peach.blender.org/download/ + myfile = genai.upload_file(media / "Big_Buck_Bunny.mp4") + print(f"{myfile=}") + + # Videos need to be processed before you can use them. + while myfile.state.name == "PROCESSING": + print("processing video...") + time.sleep(5) + myfile = genai.get_file(myfile.name) + model = genai.GenerativeModel("gemini-1.5-flash") - video = genai.upload_file(media / "Big_Buck_Bunny.mp4") - response = model.generate_content(["Describe this video clip.", video]) - print(response.text) + result = model.generate_content([myfile, "Describe this video clip"]) + print(f"{result.text=}") # [END text_gen_multimodal_video_prompt] def test_text_gen_multimodal_video_prompt_streaming(self):