#!/usr/bin/env python3
"""Initialize the SPARC database schema.

This script creates the necessary tables and indexes for storing
LLM messages for testing and analytics.

Usage:
    python scripts/init_database.py
"""

import sys
import os

# Add parent directory to path
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))

from SPARC import config
from SPARC.database import DatabaseClient


def main():
    """Initialize the database schema."""
    print("Initializing SPARC database...")
    print(f"Database URL: {config.database_url}")

    try:
        db_client = DatabaseClient(config.database_url)
        db_client.initialize_schema()
        print("Database schema initialized successfully!")
        print("\nTables created:")
        print("  - llm_messages: Stores all LLM prompts and responses")
        print("\nIndexes created:")
        print("  - idx_messages_timestamp: For time-based queries")
        print("  - idx_messages_company: For company-specific queries")

    except Exception as e:
        print(f"Error initializing database: {e}")
        sys.exit(1)


if __name__ == "__main__":
    main()