#!/usr/bin/env python3
"""
Client script to send video to the Ray Serve API.

Usage:
    python -m client.send_video --video s3://bucket/path/to/video.mp4
    python -m client.send_video --video s3://bucket/video.mp4 --chunk-duration 5.0
    python -m client.send_video --video s3://bucket/video.mp4 --token YOUR_TOKEN
"""

import argparse
import time
import uuid

import httpx


def main():
    parser = argparse.ArgumentParser(description="Send video to Ray Serve API")
    parser.add_argument("--video", type=str, required=True, help="S3 URI: s3://bucket/key")
    parser.add_argument("--stream-id", type=str, default=None, help="Stream ID (random if not provided)")
    parser.add_argument("--num-frames", type=int, default=16, help="Frames per chunk")
    parser.add_argument("--chunk-duration", type=float, default=10.0, help="Chunk duration in seconds")
    parser.add_argument("--url", type=str, default="http://127.0.0.1:8000", help="Server URL")
    parser.add_argument("--token", type=str, default=None, help="Bearer token for Authorization header")
    args = parser.parse_args()
    
    # Generate random stream ID if not provided
    stream_id = args.stream_id or uuid.uuid4().hex[:8]
    
    payload = {
        "stream_id": stream_id,
        "video_path": args.video,
        "num_frames": args.num_frames,
        "chunk_duration": args.chunk_duration,
        "use_batching": True
    }
    
    print(f"📹 Processing video: {args.video}")
    print(f"   Stream ID: {stream_id}")
    print(f"   Chunk duration: {args.chunk_duration}s, Frames/chunk: {args.num_frames}")
    print()
    
    start = time.perf_counter()
    
    headers = {}
    if args.token:
        headers["Authorization"] = f"Bearer {args.token}"
    
    with httpx.Client(timeout=300.0) as client:
        response = client.post(f"{args.url}/analyze", json=payload, headers=headers)
    
    latency_ms = (time.perf_counter() - start) * 1000
    
    if response.status_code != 200:
        print(f"❌ Error {response.status_code}: {response.text}")
        return
    
    result = response.json()
    
    print("=" * 60)
    print("✅ Response")
    print("=" * 60)
    print(f"Stream ID: {result['stream_id']}")
    print(f"Video duration: {result['video_duration']:.1f}s")
    print(f"Chunks processed: {result['num_chunks']}")
    print()
    
    print("🏷️  Top Tags (aggregated):")
    for tag in result["tags"]:
        print(f"   {tag['score']:.3f}  {tag['text']}")
    print()
    
    print("📝 Best Caption:")
    caption = result["retrieval_caption"]
    print(f"   {caption['score']:.3f}  {caption['text']}")
    print()
    
    # Scene changes
    scene_changes = result["scene_changes"]
    print(f"🎬 Scene Changes Detected: {result['num_scene_changes']}")
    if scene_changes:
        for sc in scene_changes:
            print(f"   {sc['timestamp']:6.2f}s  score={sc['score']:.3f}  (chunk {sc['chunk_index']}, frame {sc['frame_index']})")
    else:
        print("   (none detected)")
    print()
    
    # Show per-chunk results
    print("📊 Per-Chunk Results:")
    print("-" * 60)
    for chunk in result["chunks"]:
        print(f"  Chunk {chunk['chunk_index']}: {chunk['start_time']:.1f}s - {chunk['start_time'] + chunk['duration']:.1f}s")
        print(f"    Top tag: {chunk['tags'][0]['text']} ({chunk['tags'][0]['score']:.3f})")
        print(f"    Caption: {chunk['retrieval_caption']['text'][:50]}...")
        num_changes = len(chunk["scene_changes"])
        print(f"    Scene changes: {num_changes}")
    print()
    
    timing = result["timing_ms"]
    print("⏱️  Timing:")
    print(f"   S3 download:  {timing['s3_download_ms']:.1f} ms")
    print(f"   Video decode: {timing['decode_video_ms']:.1f} ms")
    print(f"   Encode (GPU): {timing['encode_ms']:.1f} ms")
    print(f"   Decode (CPU): {timing['decode_ms']:.1f} ms")
    print(f"   Total server: {timing['total_ms']:.1f} ms")
    print(f"   Round-trip:   {latency_ms:.1f} ms")
    
    if result['num_chunks'] > 1:
        avg_per_chunk = timing['total_ms'] / result['num_chunks']
        print(f"   Avg/chunk:    {avg_per_chunk:.1f} ms")


if __name__ == "__main__":
    main()