Remotion Video Generator

# chuk-motion/src/chuk_motion/components/charts/DonutChart/tool.py """DonutChart MCP tool.""" import asyncio import json from chuk_motion.generator.composition_builder import ComponentInstance from chuk_motion.models import ChartComponentResponse, ErrorResponse def register_tool(mcp, project_manager): """Register the DonutChart tool with the MCP server.""" @mcp.tool async def remotion_add_donut_chart( data: str, title: str | None = None, xlabel: str | None = None, ylabel: str | None = None, duration: float | str = 4.0, track: str = "main", gap_before: float | str | None = None, ) -> str: """ Add an animated donut chart to the composition. Animated donut chart for showing proportions with a center hole. Valid props: data, title, duration, track, gap_before Invalid props: variant, style, color, theme, animation, xlabel, ylabel (these don't exist) Args: data: JSON array of {label, value} objects. Optionally include "color" per slice. Format: [{"label": "Free", "value": 55}, {"label": "Pro", "value": 30}] title: Optional chart title duration: How long to animate (seconds) or time string track: Track name (default: "main") gap_before: Gap before component in seconds or time string Returns: JSON with component info Example: await remotion_add_donut_chart( data='[{"label": "Free", "value": 55}, {"label": "Pro", "value": 30}, {"label": "Enterprise", "value": 15}]', title="Plan Distribution", duration=4.0 ) """ def _add(): if not project_manager.current_timeline: return ErrorResponse( error="No active project. Create a project first." ).model_dump_json() try: data_parsed = json.loads(data) except json.JSONDecodeError as e: return ErrorResponse(error=f"Invalid data JSON: {str(e)}").model_dump_json() try: component = ComponentInstance( component_type="DonutChart", start_frame=0, duration_frames=0, props={ "data": data_parsed, "title": title, "xlabel": xlabel, "ylabel": ylabel, }, layer=0, ) component = project_manager.current_timeline.add_component( component, duration=duration, track=track, gap_before=gap_before ) return ChartComponentResponse( component="DonutChart", data_points=len(data_parsed), title=title, start_time=project_manager.current_timeline.frames_to_seconds( component.start_frame ), duration=duration, ).model_dump_json() except Exception as e: return ErrorResponse(error=str(e)).model_dump_json() return await asyncio.get_event_loop().run_in_executor(None, _add)