from fastapi import APIRouter
from pydantic import BaseModel
from .mdm_loader import generate_motion

router = APIRouter()

class MotionRequest(BaseModel):
    prompt: str
    frames: int = 60
    style: str = "default"

@router.post("/generate_motion")
def generate_motion_api(req: MotionRequest):
    print(f"Generating motion for: {req.prompt}")
    motion_data = generate_motion(prompt=req.prompt, num_frames=req.frames, style=req.style)
    return { "motion": motion_data }