ALwrity/backend/api/onboarding_utils/endpoints_config_data.py

from typing import Dict, Any
from loguru import logger
from fastapi import HTTPException

from .endpoint_models import APIKeyRequest

# Import persona generation functions
from .step4_persona_routes import (
    generate_writing_personas,
    generate_writing_personas_async,
    get_persona_task_status,
    assess_persona_quality,
    regenerate_persona,
    get_persona_generation_options
)


async def get_api_keys():
    try:
        from api.onboarding_utils.api_key_management_service import APIKeyManagementService
        api_service = APIKeyManagementService()
        return await api_service.get_api_keys()
    except Exception as e:
        logger.error(f"Error getting API keys: {str(e)}")
        raise HTTPException(status_code=500, detail="Internal server error")


async def get_api_keys_for_onboarding(current_user: dict = None):
    try:
        from api.onboarding_utils.api_key_management_service import APIKeyManagementService
        api_service = APIKeyManagementService()
        user_id = str(current_user.get('id')) if current_user and current_user.get('id') else None
        return await api_service.get_api_keys_for_onboarding(user_id)
    except Exception as e:
        logger.error(f"Error getting API keys for onboarding: {str(e)}")
        raise HTTPException(status_code=500, detail="Internal server error")


async def save_api_key(request: APIKeyRequest, current_user: dict = None):
    try:
        from api.onboarding_utils.api_key_management_service import APIKeyManagementService
        api_service = APIKeyManagementService()
        return await api_service.save_api_key(request.provider, request.api_key, request.description, current_user)
    except Exception as e:
        logger.error(f"Error saving API key: {str(e)}")
        raise HTTPException(status_code=500, detail="Internal server error")


async def validate_api_keys():
    try:
        from api.onboarding_utils.api_key_management_service import APIKeyManagementService
        api_service = APIKeyManagementService()
        return await api_service.validate_api_keys()
    except Exception as e:
        logger.error(f"Error validating API keys: {str(e)}")
        raise HTTPException(status_code=500, detail="Internal server error")


def get_onboarding_config():
    try:
        from api.onboarding_utils.onboarding_config_service import OnboardingConfigService
        config_service = OnboardingConfigService()
        return config_service.get_onboarding_config()
    except Exception as e:
        logger.error(f"Error getting onboarding config: {str(e)}")
        raise HTTPException(status_code=500, detail="Internal server error")


async def get_provider_setup_info(provider: str):
    try:
        from api.onboarding_utils.onboarding_config_service import OnboardingConfigService
        config_service = OnboardingConfigService()
        return await config_service.get_provider_setup_info(provider)
    except Exception as e:
        logger.error(f"Error getting provider setup info: {str(e)}")
        raise HTTPException(status_code=500, detail="Internal server error")


async def get_all_providers_info():
    try:
        from api.onboarding_utils.onboarding_config_service import OnboardingConfigService
        config_service = OnboardingConfigService()
        return config_service.get_all_providers_info()
    except Exception as e:
        logger.error(f"Error getting all providers info: {str(e)}")
        raise HTTPException(status_code=500, detail="Internal server error")


async def validate_provider_key(provider: str, request: APIKeyRequest):
    try:
        from api.onboarding_utils.onboarding_config_service import OnboardingConfigService
        config_service = OnboardingConfigService()
        return await config_service.validate_provider_key(provider, request.api_key)
    except Exception as e:
        logger.error(f"Error validating provider key: {str(e)}")
        raise HTTPException(status_code=500, detail="Internal server error")


async def get_enhanced_validation_status():
    try:
        from api.onboarding_utils.onboarding_config_service import OnboardingConfigService
        config_service = OnboardingConfigService()
        return await config_service.get_enhanced_validation_status()
    except Exception as e:
        logger.error(f"Error getting enhanced validation status: {str(e)}")
        raise HTTPException(status_code=500, detail="Internal server error")


async def get_onboarding_summary(current_user: Dict[str, Any]):
    try:
        from api.onboarding_utils.onboarding_summary_service import OnboardingSummaryService
        user_id = str(current_user.get('id'))
        summary_service = OnboardingSummaryService(user_id)
        logger.info(f"Getting onboarding summary for user {user_id}")
        return await summary_service.get_onboarding_summary()
    except Exception as e:
        logger.error(f"Error getting onboarding summary: {str(e)}")
        raise HTTPException(status_code=500, detail="Internal server error")


async def get_website_analysis_data(current_user: Dict[str, Any]):
    try:
        from api.onboarding_utils.onboarding_summary_service import OnboardingSummaryService
        user_id = str(current_user.get('id'))
        summary_service = OnboardingSummaryService(user_id)
        logger.info(f"Getting website analysis data for user {user_id}")
        return await summary_service.get_website_analysis_data()
    except Exception as e:
        logger.error(f"Error getting website analysis data: {str(e)}")
        raise HTTPException(status_code=500, detail="Internal server error")


async def get_research_preferences_data(current_user: Dict[str, Any]):
    try:
        from api.onboarding_utils.onboarding_summary_service import OnboardingSummaryService
        user_id = str(current_user.get('id'))
        summary_service = OnboardingSummaryService(user_id)
        logger.info(f"Getting research preferences data for user {user_id}")
        return await summary_service.get_research_preferences_data()
    except Exception as e:
        logger.error(f"Error getting research preferences data: {str(e)}")
        raise HTTPException(status_code=500, detail="Internal server error")


async def check_persona_generation_readiness(user_id: int = 1):
    try:
        from api.onboarding_utils.persona_management_service import PersonaManagementService
        persona_service = PersonaManagementService()
        return await persona_service.check_persona_generation_readiness(user_id)
    except Exception as e:
        logger.error(f"Error checking persona readiness: {str(e)}")
        raise HTTPException(status_code=500, detail="Internal server error")


async def generate_persona_preview(user_id: int = 1):
    try:
        from api.onboarding_utils.persona_management_service import PersonaManagementService
        persona_service = PersonaManagementService()
        return await persona_service.generate_persona_preview(user_id)
    except Exception as e:
        logger.error(f"Error generating persona preview: {str(e)}")
        raise HTTPException(status_code=500, detail="Internal server error")


async def generate_writing_persona(user_id: int = 1):
    try:
        from api.onboarding_utils.persona_management_service import PersonaManagementService
        persona_service = PersonaManagementService()
        return await persona_service.generate_writing_persona(user_id)
    except Exception as e:
        logger.error(f"Error generating writing persona: {str(e)}")
        raise HTTPException(status_code=500, detail="Internal server error")


async def get_user_writing_personas(user_id: int = 1):
    try:
        from api.onboarding_utils.persona_management_service import PersonaManagementService
        persona_service = PersonaManagementService()
        return await persona_service.get_user_writing_personas(user_id)
    except Exception as e:
        logger.error(f"Error getting user personas: {str(e)}")
        raise HTTPException(status_code=500, detail="Internal server error")


async def save_business_info(business_info: dict):
    try:
        from api.onboarding_utils.business_info_service import BusinessInfoService
        business_service = BusinessInfoService()
        return await business_service.save_business_info(business_info)
    except Exception as e:
        logger.error(f"❌ Error saving business info: {str(e)}")
        raise HTTPException(status_code=500, detail=f"Failed to save business info: {str(e)}")


async def get_business_info(business_info_id: int):
    try:
        from api.onboarding_utils.business_info_service import BusinessInfoService
        business_service = BusinessInfoService()
        return await business_service.get_business_info(business_info_id)
    except Exception as e:
        logger.error(f"❌ Error getting business info: {str(e)}")
        raise HTTPException(status_code=500, detail=f"Failed to get business info: {str(e)}")


async def get_business_info_by_user(user_id: int):
    try:
        from api.onboarding_utils.business_info_service import BusinessInfoService
        business_service = BusinessInfoService()
        return await business_service.get_business_info_by_user(user_id)
    except Exception as e:
        logger.error(f"❌ Error getting business info: {str(e)}")
        raise HTTPException(status_code=500, detail=f"Failed to get business info: {str(e)}")


async def update_business_info(business_info_id: int, business_info: dict):
    try:
        from api.onboarding_utils.business_info_service import BusinessInfoService
        business_service = BusinessInfoService()
        return await business_service.update_business_info(business_info_id, business_info)
    except Exception as e:
        logger.error(f"❌ Error updating business info: {str(e)}")
        raise HTTPException(status_code=500, detail=f"Failed to update business info: {str(e)}")


__all__ = [name for name in globals().keys() if not name.startswith('_')]