feat: Add No Website button to onboarding Step 2 with business description form

backend/services/business_info_service.py

@@ -0,0 +1,84 @@
+"""Business Information Service for ALwrity backend."""
+from sqlalchemy.orm import Session
+from models.user_business_info import UserBusinessInfo
+from models.business_info_request import BusinessInfoRequest, BusinessInfoResponse
+from services.database import get_db
+from loguru import logger
+from typing import Optional
+
+logger.info("🔄 Loading BusinessInfoService...")
+
+class BusinessInfoService:
+    def __init__(self):
+        logger.info("🆕 Initializing BusinessInfoService...")
+
+    def save_business_info(self, business_info: BusinessInfoRequest) -> BusinessInfoResponse:
+        db: Session = next(get_db())
+        logger.debug(f"Attempting to save business info for user_id: {business_info.user_id}")
+        
+        # Check if business info already exists for this user
+        existing_info = db.query(UserBusinessInfo).filter(UserBusinessInfo.user_id == business_info.user_id).first()
+        
+        if existing_info:
+            logger.info(f"Existing business info found for user_id {business_info.user_id}, updating it.")
+            existing_info.business_description = business_info.business_description
+            existing_info.industry = business_info.industry
+            existing_info.target_audience = business_info.target_audience
+            existing_info.business_goals = business_info.business_goals
+            db.commit()
+            db.refresh(existing_info)
+            logger.success(f"Updated business info for user_id {business_info.user_id}, ID: {existing_info.id}")
+            return BusinessInfoResponse(**existing_info.to_dict())
+        else:
+            logger.info(f"No existing business info for user_id {business_info.user_id}, creating new entry.")
+            db_business_info = UserBusinessInfo(
+                user_id=business_info.user_id,
+                business_description=business_info.business_description,
+                industry=business_info.industry,
+                target_audience=business_info.target_audience,
+                business_goals=business_info.business_goals
+            )
+            db.add(db_business_info)
+            db.commit()
+            db.refresh(db_business_info)
+            logger.success(f"Saved new business info for user_id {business_info.user_id}, ID: {db_business_info.id}")
+            return BusinessInfoResponse(**db_business_info.to_dict())
+
+    def get_business_info(self, business_info_id: int) -> Optional[BusinessInfoResponse]:
+        db: Session = next(get_db())
+        logger.debug(f"Retrieving business info by ID: {business_info_id}")
+        business_info = db.query(UserBusinessInfo).filter(UserBusinessInfo.id == business_info_id).first()
+        if business_info:
+            logger.debug(f"Found business info for ID: {business_info_id}")
+            return BusinessInfoResponse(**business_info.to_dict())
+        logger.warning(f"No business info found for ID: {business_info_id}")
+        return None
+
+    def get_business_info_by_user(self, user_id: int) -> Optional[BusinessInfoResponse]:
+        db: Session = next(get_db())
+        logger.debug(f"Retrieving business info by user ID: {user_id}")
+        business_info = db.query(UserBusinessInfo).filter(UserBusinessInfo.user_id == user_id).first()
+        if business_info:
+            logger.debug(f"Found business info for user ID: {user_id}")
+            return BusinessInfoResponse(**business_info.to_dict())
+        logger.warning(f"No business info found for user ID: {user_id}")
+        return None
+
+    def update_business_info(self, business_info_id: int, business_info: BusinessInfoRequest) -> Optional[BusinessInfoResponse]:
+        db: Session = next(get_db())
+        logger.debug(f"Updating business info for ID: {business_info_id}")
+        db_business_info = db.query(UserBusinessInfo).filter(UserBusinessInfo.id == business_info_id).first()
+        if db_business_info:
+            db_business_info.business_description = business_info.business_description
+            db_business_info.industry = business_info.industry
+            db_business_info.target_audience = business_info.target_audience
+            db_business_info.business_goals = business_info.business_goals
+            db.commit()
+            db.refresh(db_business_info)
+            logger.success(f"Updated business info for ID: {business_info_id}")
+            return BusinessInfoResponse(**db_business_info.to_dict())
+        logger.warning(f"No business info found to update for ID: {business_info_id}")
+        return None
+
+business_info_service = BusinessInfoService()
+logger.info("✅ BusinessInfoService loaded successfully!")

backend/services/database.py

@@ -19,6 +19,7 @@ from models.enhanced_strategy_models import Base as EnhancedStrategyBase
 from models.monitoring_models import Base as MonitoringBase
 from models.persona_models import Base as PersonaBase
 from models.subscription_models import Base as SubscriptionBase
+from models.user_business_info import Base as UserBusinessInfoBase
 
 # Database configuration
 DATABASE_URL = os.getenv('DATABASE_URL', 'sqlite:///./alwrity.db')
@@ -72,7 +73,8 @@ def init_database():
         MonitoringBase.metadata.create_all(bind=engine)
         PersonaBase.metadata.create_all(bind=engine)
         SubscriptionBase.metadata.create_all(bind=engine)
-        logger.info("Database initialized successfully with all models including subscription system")
+        UserBusinessInfoBase.metadata.create_all(bind=engine)
+        logger.info("Database initialized successfully with all models including subscription system and business info")
     except SQLAlchemyError as e:
         logger.error(f"Error initializing database: {str(e)}")
         raise