Improve the demo feature of the project

services/demo_session/app/models/__init__.py

@@ -1,5 +1,5 @@
 """Demo Session Service Models"""
 
-from .demo_session import DemoSession, DemoSessionStatus
+from .demo_session import DemoSession, DemoSessionStatus, CloningStatus
 
-__all__ = ["DemoSession", "DemoSessionStatus"]
+__all__ = ["DemoSession", "DemoSessionStatus", "CloningStatus"]

services/demo_session/app/models/demo_session.py

@@ -14,9 +14,21 @@ from shared.database.base import Base
 
 class DemoSessionStatus(enum.Enum):
     """Demo session status"""
-    ACTIVE = "active"
-    EXPIRED = "expired"
-    DESTROYED = "destroyed"
+    PENDING = "pending"      # Data cloning in progress
+    READY = "ready"          # All data loaded, safe to use
+    FAILED = "failed"        # One or more services failed completely
+    PARTIAL = "partial"      # Some services failed, others succeeded
+    ACTIVE = "active"        # User is actively using the session (deprecated, use READY)
+    EXPIRED = "expired"      # Session TTL exceeded
+    DESTROYED = "destroyed"  # Session terminated
+
+
+class CloningStatus(enum.Enum):
+    """Individual service cloning status"""
+    NOT_STARTED = "not_started"
+    IN_PROGRESS = "in_progress"
+    COMPLETED = "completed"
+    FAILED = "failed"
 
 
 class DemoSession(Base):
@@ -37,16 +49,24 @@ class DemoSession(Base):
     demo_account_type = Column(String(50), nullable=False)  # 'individual_bakery', 'central_baker'
 
     # Session lifecycle
-    status = Column(SQLEnum(DemoSessionStatus, values_callable=lambda obj: [e.value for e in obj]), default=DemoSessionStatus.ACTIVE, index=True)
+    status = Column(SQLEnum(DemoSessionStatus, values_callable=lambda obj: [e.value for e in obj]), default=DemoSessionStatus.PENDING, index=True)
     created_at = Column(DateTime(timezone=True), default=lambda: datetime.now(timezone.utc), index=True)
     expires_at = Column(DateTime(timezone=True), nullable=False, index=True)
     last_activity_at = Column(DateTime(timezone=True), default=lambda: datetime.now(timezone.utc))
     destroyed_at = Column(DateTime(timezone=True), nullable=True)
 
+    # Cloning progress tracking
+    cloning_started_at = Column(DateTime(timezone=True), nullable=True)
+    cloning_completed_at = Column(DateTime(timezone=True), nullable=True)
+    total_records_cloned = Column(Integer, default=0)
+
+    # Per-service cloning status
+    cloning_progress = Column(JSONB, default=dict)  # {service_name: {status, records, started_at, completed_at, error}}
+
     # Session metrics
     request_count = Column(Integer, default=0)
-    data_cloned = Column(Boolean, default=False)
-    redis_populated = Column(Boolean, default=False)
+    data_cloned = Column(Boolean, default=False)  # Deprecated: use status instead
+    redis_populated = Column(Boolean, default=False)  # Deprecated: use status instead
 
     # Session metadata
     session_metadata = Column(JSONB, default=dict)