import os
from mcp.server.fastmcp import FastMCP

# Khởi tạo FastMCP
mcp = FastMCP("Yue-Sing-Tools")

@mcp.tool()
def sing_song(song_name: str) -> str:
    # Nếu AI truyền "golden.mp3", ta giữ nguyên. 
    # Nếu AI truyền "golden", ta mới thêm .mp3.
    clean_name = song_name if song_name.endswith(".mp3") else f"{song_name}.mp3"
    return f"[[SING_COMMAND]]{clean_name}"

if __name__ == "__main__":
    # Chạy server MCP
    mcp.run()