from twelvelabs import TwelveLabs
from twelvelabs.models.embed import EmbeddingsTask, SegmentEmbedding

client = TwelveLabs("<YOUR_API_KEY>")

# Create a video embedding task for your video
task = client.embed.task.create(
   model_name="Marengo-retrieval-2.7",
   video_url: "<YOUR_VIDEO_URL>"
)

print(f"Created task: id={task.id} model_name={task.model_name} status={task.status}")

# Wait for embedding task to finish
status = task.wait_for_done()
print(f"Embedding done: {status}")

# Retrieve the video embeddings
task = task.retrieve()

# Print the embeddings
if task.video_embedding is not None and task.video_embedding.segments is not None:
    for segment in task.video_embedding.segments:
            print(
                f"  embedding_scope={segment.embedding_scope} start_offset_sec={segment.start_offset_sec} end_offset_sec={segment.end_offset_sec}"
            )
            print(f"  embeddings: {", ".join(str(segment.embeddings_float))}")