#!/usr/bin/env python3
import sys
import os
import argparse
from service import rebuild_index, DEFAULT_GALLERY_DIR, get_status

def main():
    parser = argparse.ArgumentParser(description="Centralized CLIP Vector Search CLI Indexer")
    parser.add_argument(
        "--gallery",
        type=str,
        default=DEFAULT_GALLERY_DIR,
        help=f"Path to gallery image directory (default: {DEFAULT_GALLERY_DIR})"
    )
    args = parser.parse_args()

    print("==================================================")
    print("🚀 Centralized CLIP Vector Search CLI Indexer")
    print(f"📁 Gallery Path: {args.gallery}")
    print("==================================================")

    success, message = rebuild_index(gallery_dir=args.gallery)

    if success:
        print(f"\n✅ Build Complete: {message}")
        stats = get_status()
        print(f"📊 Status Stats: {stats}")
    else:
        print(f"\n❌ Build Failed: {message}")
        sys.exit(1)

if __name__ == "__main__":
    main()