@hybridlabor-api/aos 4.4.1 → 4.4.2-beta.0

package/mcps/davinci-resolve-mcp/src/granular/gallery.py

@@ -1,306 +0,0 @@
-"""Gallery, still album, and powergrade tools."""
-
-from src.granular.common import *  # noqa: F401,F403
-
-resolve = ResolveProxy()
-
-@mcp.tool(annotations=READ_ONLY_TOOL)
-def get_gallery_album_name() -> Dict[str, Any]:
-    """Get the name of the current gallery album."""
-    resolve = get_resolve()
-    if resolve is None:
-        return {"error": "Not connected to DaVinci Resolve"}
-    project = resolve.GetProjectManager().GetCurrentProject()
-    if not project:
-        return {"error": "No project open"}
-    gallery = project.GetGallery()
-    if not gallery:
-        return {"error": "Failed to get Gallery"}
-    name = gallery.GetAlbumName()
-    return {"album_name": name if name else ""}
-
-
-@mcp.tool()
-def set_gallery_album_name(name: str) -> Dict[str, Any]:
-    """Set the name of the current gallery album.
-
-    Args:
-        name: New album name.
-    """
-    resolve = get_resolve()
-    if resolve is None:
-        return {"error": "Not connected to DaVinci Resolve"}
-    project = resolve.GetProjectManager().GetCurrentProject()
-    if not project:
-        return {"error": "No project open"}
-    gallery = project.GetGallery()
-    if not gallery:
-        return {"error": "Failed to get Gallery"}
-    result = gallery.SetAlbumName(name)
-    return {"success": bool(result)}
-
-
-@mcp.tool()
-def get_gallery_still_albums() -> Dict[str, Any]:
-    """Get list of all gallery still albums."""
-    resolve = get_resolve()
-    if resolve is None:
-        return {"error": "Not connected to DaVinci Resolve"}
-    project = resolve.GetProjectManager().GetCurrentProject()
-    if not project:
-        return {"error": "No project open"}
-    gallery = project.GetGallery()
-    if not gallery:
-        return {"error": "Failed to get Gallery"}
-    albums = gallery.GetGalleryStillAlbums()
-    return {"albums": [str(a) for a in albums] if albums else []}
-
-
-@mcp.tool()
-def get_gallery_power_grade_albums() -> Dict[str, Any]:
-    """Get list of all gallery power grade albums."""
-    resolve = get_resolve()
-    if resolve is None:
-        return {"error": "Not connected to DaVinci Resolve"}
-    project = resolve.GetProjectManager().GetCurrentProject()
-    if not project:
-        return {"error": "No project open"}
-    gallery = project.GetGallery()
-    if not gallery:
-        return {"error": "Failed to get Gallery"}
-    albums = gallery.GetGalleryPowerGradeAlbums()
-    return {"albums": [str(a) for a in albums] if albums else []}
-
-
-@mcp.tool()
-def get_current_still_album() -> Dict[str, Any]:
-    """Get the current still album."""
-    resolve = get_resolve()
-    if resolve is None:
-        return {"error": "Not connected to DaVinci Resolve"}
-    project = resolve.GetProjectManager().GetCurrentProject()
-    if not project:
-        return {"error": "No project open"}
-    gallery = project.GetGallery()
-    if not gallery:
-        return {"error": "Failed to get Gallery"}
-    album = gallery.GetCurrentStillAlbum()
-    return {"has_album": album is not None}
-
-
-@mcp.tool()
-def set_current_still_album(album_index: int) -> Dict[str, Any]:
-    """Set the current still album by index.
-
-    Args:
-        album_index: 0-based index of the album in GetGalleryStillAlbums() list.
-    """
-    resolve = get_resolve()
-    if resolve is None:
-        return {"error": "Not connected to DaVinci Resolve"}
-    project = resolve.GetProjectManager().GetCurrentProject()
-    if not project:
-        return {"error": "No project open"}
-    gallery = project.GetGallery()
-    if not gallery:
-        return {"error": "Failed to get Gallery"}
-    albums = gallery.GetGalleryStillAlbums()
-    if not albums or album_index >= len(albums):
-        return {"error": f"No album at index {album_index}"}
-    result = gallery.SetCurrentStillAlbum(albums[album_index])
-    return {"success": bool(result)}
-
-
-@mcp.tool()
-def create_gallery_still_album(album_name: str = "") -> Dict[str, Any]:
-    """Create a new gallery still album.
-
-    Args:
-        album_name: Optional name for the new album.
-    """
-    resolve = get_resolve()
-    if resolve is None:
-        return {"error": "Not connected to DaVinci Resolve"}
-    project = resolve.GetProjectManager().GetCurrentProject()
-    if not project:
-        return {"error": "No project open"}
-    gallery = project.GetGallery()
-    if not gallery:
-        return {"error": "Failed to get Gallery"}
-    if album_name:
-        album = gallery.CreateGalleryStillAlbum(album_name)
-    else:
-        album = gallery.CreateGalleryStillAlbum()
-    return {"success": album is not None}
-
-
-@mcp.tool()
-def create_gallery_power_grade_album(album_name: str = "") -> Dict[str, Any]:
-    """Create a new gallery power grade album.
-
-    Args:
-        album_name: Optional name for the new album.
-    """
-    resolve = get_resolve()
-    if resolve is None:
-        return {"error": "Not connected to DaVinci Resolve"}
-    project = resolve.GetProjectManager().GetCurrentProject()
-    if not project:
-        return {"error": "No project open"}
-    gallery = project.GetGallery()
-    if not gallery:
-        return {"error": "Failed to get Gallery"}
-    if album_name:
-        album = gallery.CreateGalleryPowerGradeAlbum(album_name)
-    else:
-        album = gallery.CreateGalleryPowerGradeAlbum()
-    return {"success": album is not None}
-
-
-@mcp.tool()
-def get_album_stills(album_index: int = 0) -> Dict[str, Any]:
-    """Get list of stills in a gallery album.
-
-    Args:
-        album_index: 0-based index of the album. Default: 0.
-    """
-    resolve = get_resolve()
-    if resolve is None:
-        return {"error": "Not connected to DaVinci Resolve"}
-    project = resolve.GetProjectManager().GetCurrentProject()
-    if not project:
-        return {"error": "No project open"}
-    gallery = project.GetGallery()
-    if not gallery:
-        return {"error": "Failed to get Gallery"}
-    albums = gallery.GetGalleryStillAlbums()
-    if not albums or album_index >= len(albums):
-        return {"error": f"No album at index {album_index}"}
-    stills = albums[album_index].GetStills()
-    return {"still_count": len(stills) if stills else 0}
-
-
-@mcp.tool()
-def get_still_label(album_index: int, still_index: int) -> Dict[str, Any]:
-    """Get the label of a still in a gallery album.
-
-    Args:
-        album_index: 0-based album index.
-        still_index: 0-based still index.
-    """
-    resolve = get_resolve()
-    if resolve is None:
-        return {"error": "Not connected to DaVinci Resolve"}
-    project = resolve.GetProjectManager().GetCurrentProject()
-    if not project:
-        return {"error": "No project open"}
-    gallery = project.GetGallery()
-    albums = gallery.GetGalleryStillAlbums()
-    if not albums or album_index >= len(albums):
-        return {"error": f"No album at index {album_index}"}
-    stills = albums[album_index].GetStills()
-    if not stills or still_index >= len(stills):
-        return {"error": f"No still at index {still_index}"}
-    label = albums[album_index].GetLabel(stills[still_index])
-    return {"label": label if label else ""}
-
-
-@mcp.tool()
-def set_still_label(album_index: int, still_index: int, label: str) -> Dict[str, Any]:
-    """Set the label of a still in a gallery album.
-
-    Args:
-        album_index: 0-based album index.
-        still_index: 0-based still index.
-        label: New label for the still.
-    """
-    resolve = get_resolve()
-    if resolve is None:
-        return {"error": "Not connected to DaVinci Resolve"}
-    project = resolve.GetProjectManager().GetCurrentProject()
-    if not project:
-        return {"error": "No project open"}
-    gallery = project.GetGallery()
-    albums = gallery.GetGalleryStillAlbums()
-    if not albums or album_index >= len(albums):
-        return {"error": f"No album at index {album_index}"}
-    stills = albums[album_index].GetStills()
-    if not stills or still_index >= len(stills):
-        return {"error": f"No still at index {still_index}"}
-    result = albums[album_index].SetLabel(stills[still_index], label)
-    return {"success": bool(result)}
-
-
-@mcp.tool()
-def import_stills_to_album(album_index: int, file_paths: List[str]) -> Dict[str, Any]:
-    """Import stills from file paths into a gallery album.
-
-    Args:
-        album_index: 0-based album index.
-        file_paths: List of absolute file paths to import.
-    """
-    resolve = get_resolve()
-    if resolve is None:
-        return {"error": "Not connected to DaVinci Resolve"}
-    project = resolve.GetProjectManager().GetCurrentProject()
-    if not project:
-        return {"error": "No project open"}
-    gallery = project.GetGallery()
-    albums = gallery.GetGalleryStillAlbums()
-    if not albums or album_index >= len(albums):
-        return {"error": f"No album at index {album_index}"}
-    result = albums[album_index].ImportStills(file_paths)
-    return {"success": bool(result)}
-
-
-@mcp.tool()
-def export_stills_from_album(album_index: int, folder_path: str, file_prefix: str = "still", format: str = "dpx") -> Dict[str, Any]:
-    """Export stills from a gallery album.
-
-    Args:
-        album_index: 0-based album index.
-        folder_path: Directory to export to.
-        file_prefix: Filename prefix. Default: 'still'.
-        format: File format (dpx, cin, tif, jpg, png, ppm, bmp, xpm, drx). Default: 'dpx'.
-    """
-    resolve = get_resolve()
-    if resolve is None:
-        return {"error": "Not connected to DaVinci Resolve"}
-    project = resolve.GetProjectManager().GetCurrentProject()
-    if not project:
-        return {"error": "No project open"}
-    gallery = project.GetGallery()
-    albums = gallery.GetGalleryStillAlbums()
-    if not albums or album_index >= len(albums):
-        return {"error": f"No album at index {album_index}"}
-    stills = albums[album_index].GetStills()
-    if not stills:
-        return {"error": "No stills in album"}
-    result = albums[album_index].ExportStills(stills, folder_path, file_prefix, format)
-    return {"success": bool(result)}
-
-
-@mcp.tool()
-def delete_stills_from_album(album_index: int, still_indices: List[int]) -> Dict[str, Any]:
-    """Delete stills from a gallery album.
-
-    Args:
-        album_index: 0-based album index.
-        still_indices: List of 0-based still indices to delete.
-    """
-    resolve = get_resolve()
-    if resolve is None:
-        return {"error": "Not connected to DaVinci Resolve"}
-    project = resolve.GetProjectManager().GetCurrentProject()
-    if not project:
-        return {"error": "No project open"}
-    gallery = project.GetGallery()
-    albums = gallery.GetGalleryStillAlbums()
-    if not albums or album_index >= len(albums):
-        return {"error": f"No album at index {album_index}"}
-    stills = albums[album_index].GetStills()
-    if not stills:
-        return {"error": "No stills in album"}
-    to_delete = [stills[i] for i in still_indices if i < len(stills)]
-    result = albums[album_index].DeleteStills(to_delete)
-    return {"success": bool(result)}

package/mcps/davinci-resolve-mcp/src/granular/graph.py

@@ -1,309 +0,0 @@
-"""Color page graph, LUT, and color-group tools."""
-
-from src.granular.common import *  # noqa: F401,F403
-
-resolve = ResolveProxy()
-
-@mcp.tool()
-def graph_get_num_nodes(item_index: int = 0, track_type: str = "video", track_index: int = 1) -> Dict[str, Any]:
-    """Get number of nodes in the color graph for a timeline item.
-
-    Args:
-        item_index: 0-based item index. Default: 0.
-        track_type: 'video' or 'audio'. Default: 'video'.
-        track_index: 1-based track index. Default: 1.
-    """
-    item, err = _get_timeline_item(track_type, track_index, item_index)
-    if err:
-        return err
-    graph = item.GetNodeGraph()
-    if not graph:
-        return {"error": "No node graph available"}
-    return {"num_nodes": graph.GetNumNodes()}
-
-
-@mcp.tool()
-def graph_set_lut(node_index: int, lut_path: str, item_index: int = 0, track_type: str = "video", track_index: int = 1) -> Dict[str, Any]:
-    """Set LUT on a node in the color graph.
-
-    Args:
-        node_index: 1-based node index.
-        lut_path: Absolute or relative LUT path.
-        item_index: 0-based timeline item index. Default: 0.
-        track_type: 'video' or 'audio'. Default: 'video'.
-        track_index: 1-based track index. Default: 1.
-    """
-    item, err = _get_timeline_item(track_type, track_index, item_index)
-    if err:
-        return err
-    graph = item.GetNodeGraph()
-    if not graph:
-        return {"error": "No node graph available"}
-    result = graph.SetLUT(node_index, lut_path)
-    return {"success": bool(result)}
-
-
-@mcp.tool()
-def graph_get_lut(node_index: int, item_index: int = 0, track_type: str = "video", track_index: int = 1) -> Dict[str, Any]:
-    """Get LUT path on a node in the color graph.
-
-    Args:
-        node_index: 1-based node index.
-        item_index: 0-based timeline item index. Default: 0.
-        track_type: 'video' or 'audio'. Default: 'video'.
-        track_index: 1-based track index. Default: 1.
-    """
-    item, err = _get_timeline_item(track_type, track_index, item_index)
-    if err:
-        return err
-    graph = item.GetNodeGraph()
-    if not graph:
-        return {"error": "No node graph available"}
-    lut = graph.GetLUT(node_index)
-    return {"node_index": node_index, "lut_path": lut if lut else ""}
-
-
-@mcp.tool()
-def graph_set_node_cache_mode(node_index: int, cache_value: int, item_index: int = 0, track_type: str = "video", track_index: int = 1) -> Dict[str, Any]:
-    """Set the cache mode on a node.
-
-    Args:
-        node_index: 1-based node index.
-        cache_value: -1=Auto, 0=Disabled, 1=Enabled.
-        item_index: 0-based timeline item index. Default: 0.
-        track_type: 'video' or 'audio'. Default: 'video'.
-        track_index: 1-based track index. Default: 1.
-    """
-    item, err = _get_timeline_item(track_type, track_index, item_index)
-    if err:
-        return err
-    graph = item.GetNodeGraph()
-    if not graph:
-        return {"error": "No node graph available"}
-    result = graph.SetNodeCacheMode(node_index, cache_value)
-    return {"success": bool(result)}
-
-
-@mcp.tool()
-def graph_get_node_cache_mode(node_index: int, item_index: int = 0, track_type: str = "video", track_index: int = 1) -> Dict[str, Any]:
-    """Get the cache mode of a node.
-
-    Args:
-        node_index: 1-based node index.
-        item_index: 0-based timeline item index. Default: 0.
-        track_type: 'video' or 'audio'. Default: 'video'.
-        track_index: 1-based track index. Default: 1.
-    """
-    item, err = _get_timeline_item(track_type, track_index, item_index)
-    if err:
-        return err
-    graph = item.GetNodeGraph()
-    if not graph:
-        return {"error": "No node graph available"}
-    mode = graph.GetNodeCacheMode(node_index)
-    modes = {-1: "Auto", 0: "Disabled", 1: "Enabled"}
-    return {"node_index": node_index, "cache_mode": mode, "mode_name": modes.get(mode, "Unknown")}
-
-
-@mcp.tool()
-def graph_get_node_label(node_index: int, item_index: int = 0, track_type: str = "video", track_index: int = 1) -> Dict[str, Any]:
-    """Get the label of a node.
-
-    Args:
-        node_index: 1-based node index.
-        item_index: 0-based timeline item index. Default: 0.
-        track_type: 'video' or 'audio'. Default: 'video'.
-        track_index: 1-based track index. Default: 1.
-    """
-    item, err = _get_timeline_item(track_type, track_index, item_index)
-    if err:
-        return err
-    graph = item.GetNodeGraph()
-    if not graph:
-        return {"error": "No node graph available"}
-    label = graph.GetNodeLabel(node_index)
-    return {"node_index": node_index, "label": label if label else ""}
-
-
-@mcp.tool()
-def graph_get_tools_in_node(node_index: int, item_index: int = 0, track_type: str = "video", track_index: int = 1) -> Dict[str, Any]:
-    """Get list of tools used in a node.
-
-    Args:
-        node_index: 1-based node index.
-        item_index: 0-based timeline item index. Default: 0.
-        track_type: 'video' or 'audio'. Default: 'video'.
-        track_index: 1-based track index. Default: 1.
-    """
-    item, err = _get_timeline_item(track_type, track_index, item_index)
-    if err:
-        return err
-    graph = item.GetNodeGraph()
-    if not graph:
-        return {"error": "No node graph available"}
-    tools = graph.GetToolsInNode(node_index)
-    return {"node_index": node_index, "tools": tools if tools else []}
-
-
-@mcp.tool()
-def graph_set_node_enabled(node_index: int, is_enabled: bool, item_index: int = 0, track_type: str = "video", track_index: int = 1) -> Dict[str, Any]:
-    """Enable or disable a node.
-
-    Args:
-        node_index: 1-based node index.
-        is_enabled: True to enable, False to disable.
-        item_index: 0-based timeline item index. Default: 0.
-        track_type: 'video' or 'audio'. Default: 'video'.
-        track_index: 1-based track index. Default: 1.
-    """
-    item, err = _get_timeline_item(track_type, track_index, item_index)
-    if err:
-        return err
-    graph = item.GetNodeGraph()
-    if not graph:
-        return {"error": "No node graph available"}
-    result = graph.SetNodeEnabled(node_index, is_enabled)
-    return {"success": bool(result)}
-
-
-@mcp.tool()
-def graph_apply_grade_from_drx(drx_path: str, grade_mode: int = 0, item_index: int = 0, track_type: str = "video", track_index: int = 1) -> Dict[str, Any]:
-    """Apply a grade from a .drx file to a timeline item's graph.
-
-    Args:
-        drx_path: Absolute path to the .drx file.
-        grade_mode: 0=No keyframes, 1=Source Timecode aligned, 2=Start Frames aligned.
-        item_index: 0-based timeline item index. Default: 0.
-        track_type: 'video' or 'audio'. Default: 'video'.
-        track_index: 1-based track index. Default: 1.
-    """
-    item, err = _get_timeline_item(track_type, track_index, item_index)
-    if err:
-        return err
-    graph = item.GetNodeGraph()
-    if not graph:
-        return {"error": "No node graph available"}
-    result = graph.ApplyGradeFromDRX(drx_path, grade_mode)
-    return {"success": bool(result)}
-
-
-@mcp.tool()
-def graph_apply_arri_cdl_lut(item_index: int = 0, track_type: str = "video", track_index: int = 1) -> Dict[str, Any]:
-    """Apply ARRI CDL and LUT to a timeline item's graph.
-
-    Args:
-        item_index: 0-based timeline item index. Default: 0.
-        track_type: 'video' or 'audio'. Default: 'video'.
-        track_index: 1-based track index. Default: 1.
-    """
-    item, err = _get_timeline_item(track_type, track_index, item_index)
-    if err:
-        return err
-    graph = item.GetNodeGraph()
-    if not graph:
-        return {"error": "No node graph available"}
-    result = graph.ApplyArriCdlLut()
-    return {"success": bool(result)}
-
-
-@mcp.tool()
-def graph_reset_all_grades(item_index: int = 0, track_type: str = "video", track_index: int = 1) -> Dict[str, Any]:
-    """Reset all grades on a timeline item's graph.
-
-    Args:
-        item_index: 0-based timeline item index. Default: 0.
-        track_type: 'video' or 'audio'. Default: 'video'.
-        track_index: 1-based track index. Default: 1.
-    """
-    item, err = _get_timeline_item(track_type, track_index, item_index)
-    if err:
-        return err
-    graph = item.GetNodeGraph()
-    if not graph:
-        return {"error": "No node graph available"}
-    result = graph.ResetAllGrades()
-    return {"success": bool(result)}
-
-
-@mcp.tool()
-def get_color_group_clips(group_name: str) -> Dict[str, Any]:
-    """Get clips in a color group for the current timeline.
-
-    Args:
-        group_name: Name of the color group.
-    """
-    resolve = get_resolve()
-    if resolve is None:
-        return {"error": "Not connected to DaVinci Resolve"}
-    project = resolve.GetProjectManager().GetCurrentProject()
-    if not project:
-        return {"error": "No project open"}
-    groups = project.GetColorGroupsList()
-    target = None
-    if groups:
-        for g in groups:
-            if g.GetName() == group_name:
-                target = g
-                break
-    if not target:
-        return {"error": f"Color group '{group_name}' not found"}
-    clips = target.GetClipsInTimeline()
-    if clips:
-        return {"group": group_name, "clips": [{"name": c.GetName()} for c in clips]}
-    return {"group": group_name, "clips": []}
-
-
-@mcp.tool()
-def get_color_group_pre_clip_node_graph(group_name: str) -> Dict[str, Any]:
-    """Get the pre-clip node graph for a color group.
-
-    Args:
-        group_name: Name of the color group.
-    """
-    resolve = get_resolve()
-    if resolve is None:
-        return {"error": "Not connected to DaVinci Resolve"}
-    project = resolve.GetProjectManager().GetCurrentProject()
-    if not project:
-        return {"error": "No project open"}
-    groups = project.GetColorGroupsList()
-    target = None
-    if groups:
-        for g in groups:
-            if g.GetName() == group_name:
-                target = g
-                break
-    if not target:
-        return {"error": f"Color group '{group_name}' not found"}
-    graph = target.GetPreClipNodeGraph()
-    if graph:
-        return {"group": group_name, "graph_type": "pre_clip", "num_nodes": graph.GetNumNodes()}
-    return {"error": "No pre-clip node graph available"}
-
-
-@mcp.tool()
-def get_color_group_post_clip_node_graph(group_name: str) -> Dict[str, Any]:
-    """Get the post-clip node graph for a color group.
-
-    Args:
-        group_name: Name of the color group.
-    """
-    resolve = get_resolve()
-    if resolve is None:
-        return {"error": "Not connected to DaVinci Resolve"}
-    project = resolve.GetProjectManager().GetCurrentProject()
-    if not project:
-        return {"error": "No project open"}
-    groups = project.GetColorGroupsList()
-    target = None
-    if groups:
-        for g in groups:
-            if g.GetName() == group_name:
-                target = g
-                break
-    if not target:
-        return {"error": f"Color group '{group_name}' not found"}
-    graph = target.GetPostClipNodeGraph()
-    if graph:
-        return {"group": group_name, "graph_type": "post_clip", "num_nodes": graph.GetNumNodes()}
-    return {"error": "No post-clip node graph available"}