37 أسطر
1.4 KiB
Python
37 أسطر
1.4 KiB
Python
import sys
|
|
import json
|
|
from pathlib import Path
|
|
|
|
# Add the current directory to the path so we can import server.py
|
|
sys.path.insert(0, str(Path(__file__).parent))
|
|
|
|
from server import run_security_suite, evaluate_asr, generate_redteam_report
|
|
|
|
def main():
|
|
print("====================================================")
|
|
print("1. STARTING FULL SECURITY SUITE (45 payloads)...")
|
|
print("====================================================")
|
|
# Headless = True means the browser runs invisibly.
|
|
# Timeout 120s ensures we wait enough time for the AI to respond.
|
|
res = run_security_suite(headless=True, timeout_per_test=120)
|
|
print("\n[STEP 1 RESULTS]")
|
|
print(json.dumps(json.loads(res), indent=2))
|
|
|
|
print("\n====================================================")
|
|
print("2. EVALUATING ATTACK SUCCESS RATE (ASR)...")
|
|
print("====================================================")
|
|
eval_res = evaluate_asr()
|
|
print("\n[STEP 2 RESULTS]")
|
|
print(json.dumps(json.loads(eval_res), indent=2))
|
|
|
|
print("\n====================================================")
|
|
print("3. GENERATING FINAL RED-TEAM REPORT...")
|
|
print("====================================================")
|
|
rep_res = generate_redteam_report()
|
|
print("\n[STEP 3 RESULTS]")
|
|
print(json.dumps(json.loads(rep_res), indent=2))
|
|
print("\nDONE! You can now check redteam_report.md for the final results.")
|
|
|
|
if __name__ == "__main__":
|
|
main()
|