Spaces:
Sleeping
Sleeping
File size: 3,028 Bytes
8aa8f36 c322df2 8aa8f36 |
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 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 |
import gradio as gr
from .tools import log_alert, get_open_alerts, mark_alert_resolved, summarize_alerts
# Define the Gradio interface
# We can use a TabbedInterface to organize the tools for the UI,
# which also registers them as MCP tools.
def create_demo():
with gr.Blocks(title="CredentialWatch Alert MCP") as demo:
gr.Markdown("# CredentialWatch Alert MCP Server")
gr.Markdown("This server exposes tools for managing alerts via MCP. It also provides a simple UI.")
with gr.Tab("Log Alert"):
gr.Markdown("## Log a new alert")
with gr.Row():
t1_provider_id = gr.Number(label="Provider ID", value=1, precision=0)
t1_credential_id = gr.Number(label="Credential ID (Optional)", value=None, precision=0)
with gr.Row():
t1_severity = gr.Dropdown(choices=["info", "warning", "critical"], label="Severity", value="info")
t1_window_days = gr.Number(label="Window Days", value=30, precision=0)
t1_message = gr.Textbox(label="Message")
t1_channel = gr.Textbox(label="Channel", value="ui")
t1_output = gr.JSON(label="Response")
t1_btn = gr.Button("Log Alert")
t1_btn.click(
fn=log_alert,
inputs=[t1_provider_id, t1_severity, t1_window_days, t1_message, t1_credential_id, t1_channel],
outputs=t1_output
)
with gr.Tab("Get Open Alerts"):
gr.Markdown("## List open alerts")
t2_provider_id = gr.Number(label="Provider ID (Optional)", value=None, precision=0)
t2_severity = gr.Dropdown(choices=["info", "warning", "critical", None], label="Severity (Optional)", value=None)
t2_output = gr.JSON(label="Open Alerts")
t2_btn = gr.Button("Get Alerts")
t2_btn.click(
fn=get_open_alerts,
inputs=[t2_provider_id, t2_severity],
outputs=t2_output
)
with gr.Tab("Resolve Alert"):
gr.Markdown("## Mark alert as resolved")
t3_alert_id = gr.Number(label="Alert ID", precision=0)
t3_note = gr.Textbox(label="Resolution Note")
t3_output = gr.JSON(label="Updated Alert")
t3_btn = gr.Button("Resolve")
t3_btn.click(
fn=mark_alert_resolved,
inputs=[t3_alert_id, t3_note],
outputs=t3_output
)
with gr.Tab("Summarize Alerts"):
gr.Markdown("## Summary of alerts")
t4_window = gr.Number(label="Window Days (Optional)", value=None, precision=0)
t4_output = gr.JSON(label="Summary")
t4_btn = gr.Button("Summarize")
t4_btn.click(
fn=summarize_alerts,
inputs=[t4_window],
outputs=t4_output
)
return demo
if __name__ == "__main__":
demo = create_demo()
demo.launch(mcp_server=True)
|