"""MCP chatbot runner with vulnerability search tools."""

from gradio_mcp import GradioMCP
from scripts.tools_mcp import (
    search_cvedb,
    search_nvd,
    search_kevin,
    search_epss,
    generate_vulnerability_report
)

def main():
    """Main function to run the MCP chatbot."""
    # Initialize the Gradio MCP interface
    mcp = GradioMCP()
    
    # Register tools
    mcp.register_tool(search_cvedb)
    mcp.register_tool(search_nvd)
    mcp.register_tool(search_kevin)
    mcp.register_tool(search_epss)
    mcp.register_tool(generate_vulnerability_report)
    
    # Launch the interface
    mcp.launch()

if __name__ == "__main__":
    main()