#!/usr/bin/env python3
"""
Initialize Database Script
==========================

Author: RSK World (https://rskworld.in)
Year: 2026

Description: Script to initialize the weather chatbot database
"""

import os
import sys

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

from utils.database import WeatherDatabase

def main():
    """Initialize the database."""
    print("Initializing Weather Chatbot Database...")
    print("=" * 60)
    
    db_path = os.getenv('DATABASE_URL', 'weather_chatbot.db')
    if db_path.startswith('sqlite:///'):
        db_path = db_path.replace('sqlite:///', '')
    
    print(f"Database path: {db_path}")
    
    try:
        db = WeatherDatabase(db_path=db_path)
        print("✓ Database initialized successfully")
        
        # Get database stats
        stats = db.get_database_stats()
        
        print("\nDatabase Statistics:")
        print(f"  • Users: {stats.get('users_count', 0)}")
        print(f"  • Weather History Records: {stats.get('weather_history_count', 0)}")
        print(f"  • User Queries: {stats.get('user_queries_count', 0)}")
        print(f"  • Weather Alerts: {stats.get('weather_alerts_count', 0)}")
        print(f"  • Favorite Cities: {stats.get('user_favorites_count', 0)}")
        
        print("\n" + "=" * 60)
        print("Database setup complete!")
        print("=" * 60)
        
    except Exception as e:
        print(f"❌ Error initializing database: {e}")
        sys.exit(1)

if __name__ == '__main__':
    main()