from pydantic import BaseModel
from typing import Optional
from datetime import datetime

class TokenConnectRequest(BaseModel):
    hf_token: str

class SpaceModel(BaseModel):
    user_id: str
    hf_space_id: str  # e.g., "username/space-name"
    name: str
    owner: str
    status: str  # e.g., "Running", "Sleeping", "Error"
    direct_url: str
    cron_enabled: bool = False
    cron_interval: int = 3600  # Default 1 hour in seconds
    last_activity: Optional[datetime] = None