228 lines
9.8 KiB
Python
228 lines
9.8 KiB
Python
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: 'BusinessInfoRequest'):
|
|
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: 'BusinessInfoRequest'):
|
|
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('_')]
|
|
|
|
|