from fastapi import APIRouter, Depends
from pydantic import BaseModel
from backend.services.hf_inference import get_voice_clone_status, set_colab_url
from backend.routers.auth import get_current_user
from backend.database import get_dashboard_stats

router = APIRouter(prefix="/admin", tags=["Admin"])

class ColabRequest(BaseModel):
    url: str

@router.get("/status")
def status():
    return get_voice_clone_status()

@router.get("/stats")
async def stats(current_user: str = Depends(get_current_user)):
    return await get_dashboard_stats(current_user)

@router.post("/colab")
def set_colab(req: ColabRequest):
    set_colab_url(req.url)
    return {"message": "Colab URL updated"}