File size: 689 Bytes
52b089b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 |
"""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() |