mcp_server.py

from mcp.server.fastmcp import FastMCP
from src.utils.change_format import change_format
from src.utils.image_helpers import remove_background_from_url
from src.utils.visualize_image import visualize_base64_image

mcp = FastMCP("Youtube Service")

@mcp.tool()
def say_hello(name: str) -> str:
  """
    Returns a greeting message for the given name.

    Args:
        name (str): The name to greet.
        
    Returns:
        str: A greeting message.
  """
  return f"Hello, {name}!"

mcp.add_tool(remove_background_from_url)
mcp.add_tool(change_format)
mcp.add_tool(visualize_base64_image)

if __name__ == "__main__":
    mcp.run()