Remotion Video Generator

# chuk-motion/src/chuk_motion/components/charts/PieChart/tool.py """PieChart 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 PieChart tool with the MCP server.""" @mcp.tool async def remotion_add_pie_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 pie chart to the composition. Animated pie chart for showing proportions and percentages. 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": "North", "value": 40}, {"label": "South", "value": 25}] 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_pie_chart( data='[{"label": "North", "value": 40}, {"label": "South", "value": 25}, {"label": "East", "value": 20}]', title="Regional Sales", 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="PieChart", 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="PieChart", 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)