Added onboarding progress tracking & landing page

backend/api/onboarding_utils/onboarding_completion_service.py

@@ -0,0 +1,94 @@
+"""
+Onboarding Completion Service
+Handles the complex logic for completing the onboarding process.
+"""
+
+from typing import Dict, Any, List
+from fastapi import HTTPException
+from loguru import logger
+
+from services.api_key_manager import get_onboarding_progress_for_user, get_api_key_manager, StepStatus
+from services.persona_analysis_service import PersonaAnalysisService
+
+class OnboardingCompletionService:
+    """Service for handling onboarding completion logic."""
+    
+    def __init__(self):
+        self.required_steps = [1, 2, 3, 6]  # Steps 1, 2, 3, and 6 are required
+    
+    async def complete_onboarding(self, current_user: Dict[str, Any]) -> Dict[str, Any]:
+        """Complete the onboarding process with full validation."""
+        try:
+            user_id = str(current_user.get('id'))
+            progress = get_onboarding_progress_for_user(user_id)
+            
+            # Validate required steps are completed
+            missing_steps = self._validate_required_steps(progress)
+            if missing_steps:
+                missing_steps_str = ", ".join(missing_steps)
+                raise HTTPException(
+                    status_code=400, 
+                    detail=f"Cannot complete onboarding. The following steps must be completed first: {missing_steps_str}"
+                )
+            
+            # Validate API keys are configured
+            self._validate_api_keys()
+            
+            # Generate writing persona from onboarding data
+            persona_generated = await self._generate_persona_from_onboarding(user_id)
+            
+            # Complete the onboarding process
+            progress.complete_onboarding()
+            
+            return {
+                "message": "Onboarding completed successfully",
+                "completed_at": progress.completed_at,
+                "completion_percentage": 100.0,
+                "persona_generated": persona_generated
+            }
+            
+        except HTTPException:
+            raise
+        except Exception as e:
+            logger.error(f"Error completing onboarding: {str(e)}")
+            raise HTTPException(status_code=500, detail="Internal server error")
+    
+    def _validate_required_steps(self, progress) -> List[str]:
+        """Validate that all required steps are completed."""
+        missing_steps = []
+        
+        for step_num in self.required_steps:
+            step = progress.get_step_data(step_num)
+            if step and step.status not in [StepStatus.COMPLETED, StepStatus.SKIPPED]:
+                missing_steps.append(step.title)
+        
+        return missing_steps
+    
+    def _validate_api_keys(self):
+        """Validate that API keys are configured."""
+        api_manager = get_api_key_manager()
+        api_keys = api_manager.get_all_keys()
+        if not api_keys:
+            raise HTTPException(
+                status_code=400,
+                detail="Cannot complete onboarding. At least one AI provider API key must be configured."
+            )
+    
+    async def _generate_persona_from_onboarding(self, user_id: str) -> bool:
+        """Generate writing persona from onboarding data."""
+        try:
+            persona_service = PersonaAnalysisService()
+            
+            # Use user_id = 1 for now (assuming single user system)
+            persona_user_id = 1
+            persona_result = persona_service.generate_persona_from_onboarding(persona_user_id)
+            
+            if "error" not in persona_result:
+                logger.info(f"✅ Writing persona generated during onboarding completion: {persona_result.get('persona_id')}")
+                return True
+            else:
+                logger.warning(f"⚠️ Persona generation failed during onboarding: {persona_result['error']}")
+                return False
+        except Exception as e:
+            logger.warning(f"⚠️ Non-critical error generating persona during onboarding: {str(e)}")
+            return False