1
def videos(
2
    self,
3
    video_ids: List[str],
4
    options: List[Union[str, Literal["visual", "conversation", "text_in_video"]]],
5
    classes: List[models.ClassifyClassParams],
6
    *,
7
    conversation_option: Optional[Union[str, Literal["semantic", "exact_match"]]] = "semantic",
8
    page_limit: Optional[int] = None,
9
    include_clips: Optional[bool] = None,
10
    threshold: Optional[Dict[str, Any]] = None,
11
    show_detailed_score: Optional[bool] = None,
12
    **kwargs
13
) -> models.ClassifyPageResult

1
def index(
2
    self,
3
    index_id: str,
4
    options: List[Union[str, Literal["visual", "conversation", "text_in_video"]]],
5
    classes: List[models.ClassifyClassParams],
6
    *,
7
    conversation_option: Optional[Union[str, Literal["semantic", "exact_match"]]] = "semantic",
8
    page_limit: Optional[int] = None,
9
    include_clips: Optional[bool] = None,
10
    threshold: Optional[Dict[str, Any]] = None,
11
    show_detailed_score: Optional[bool] = None,
12
    **kwargs
13
) -> models.ClassifyPageResult

1
def by_page_token(
2
    self,
3
    page_token: str,
4
    **kwargs
5
) -> models.ClassifyPageResult

1
CLASSES = [
2
    {
3
        "name": "DanceTok",
4
        "prompts": [
5
            "Dance tutorial",
6
            "Dance group",
7
            "Dance competition"
8
        ]
9
    },
10
    {
11
        "name": "CookTok",
12
        "prompts": [
13
            "Cooking tutorial",
14
            "Cooking ustensils review"
15
        ]
16
    }
17
]
18
19
def print_classify_page_result(result):
20
    print("Classification results:")
21
    data = getattr(result, 'data', result)
22
    if isinstance(data, list):
23
        for video_data in data:
24
            print_classify_video_data(video_data)
25
    else:
26
        print_classify_video_data(data)
27
28
def print_classify_video_data(video_data):
29
    print(f"Video ID: {video_data.video_id}")
30
    for class_data in video_data.classes:
31
        print_classify_class(class_data)
32
    print()  # Add a blank line between videos
33
34
def print_classify_class(class_data):
35
    print(f"  Class name: {class_data.name}")
36
    print(f"  Score: {class_data.score}")
37
    print(f"  Duration ratio: {class_data.duration_ratio}")
38
    if class_data.clips:
39
        print("  Clips:")
40
        for clip in class_data.clips:
41
            print_classify_clip(clip)
42
43
def print_classify_clip(clip):
44
    print(f"    Start: {clip.start}")
45
    print(f"    End: {clip.end}")
46
    print(f"    Score: {clip.score}")
47
    print(f"    Option: {clip.option}")
48
    print(f"    Prompt: {clip.prompt}")
49
    print(f"    Thumbnail URL: {clip.thumbnail_url or 'N/A'}")
50
    if clip.detailed_scores:
51
        print_classify_detailed_score(clip.detailed_scores)
52
53
def print_classify_detailed_score(detailed_score):
54
    print(f"      Max score: {detailed_score.max_score}")
55
    print(f"      Avg score: {detailed_score.avg_score}")
56
    print(f"      Normalized score: {detailed_score.normalized_score}")
57
58
59
result = client.classify.index(
60
    index_id="<YOUR_INDEX_ID>",
61
    options=["visual", "conversation"],
62
    classes=CLASSES,
63
    include_clips=True,
64
    show_detailed_score=True,
65
    page_limit=1
66
)
67
68
print_classify_page_result(result)
69
70
page_number = 2
71
while True:
72
    try:
73
        print(f"Page {page_number}")
74
        print_classify_page_result(next(result))
75
        page_number += 1
76
    except StopIteration:
77
        break
78
79
print("No more results.")