#!/usr/bin/env python3
"""
Hugging Face Space app for Second Brain AI Assistant.

Direct implementation for HF Space deployment.
"""

import os
import sys
from pathlib import Path

# Add the current directory and src directory to Python path
sys.path.append('.')
sys.path.append('src')

from second_brain_online.application.agents import get_agent
from second_brain_online.application.ui import CustomGradioUI

def main():
    """Main function for Hugging Face Space deployment."""
    # Set default values for HF Spaces
    retriever_config_path = os.getenv("RETRIEVER_CONFIG_PATH", "configs/compute_rag_vector_index_openai_contextual_simple.yaml")
    
    print("🚀 Starting Second Brain AI Assistant...")
    print(f"📁 Using retriever config: {retriever_config_path}")
    
    try:
        # Initialize agent
        agent = get_agent(retriever_config_path=Path(retriever_config_path))
        
        # Get the actual agent from the wrapper
        actual_agent = agent._AgentWrapper__agent
        
        # Launch custom UI
        CustomGradioUI(actual_agent).launch(
            server_name="0.0.0.0",
            server_port=7860,
            share=False
        )
    except Exception as e:
        print(f"❌ Error starting the application: {e}")
        raise

if __name__ == "__main__":
    main()