|
|
"""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."""
|
|
|
|
|
|
mcp = GradioMCP()
|
|
|
|
|
|
|
|
|
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)
|
|
|
|
|
|
|
|
|
mcp.launch()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
main() |
