#!/usr/bin/env python3

import subprocess

# Try some common CTF flag formats and simple words
flags = [
    "flag{baby}",
    "flag{baby_checker}",
    "baby",
    "baby_checker",
    "checker",
    "flag{checker}",
    "check",
    "flag{check}",
    "rev",
    "flag{rev}",
    "reverse",
    "flag{reverse}",
    "flag{baby_rev}",
    "baby_rev",
]

print("Testing flags...")
for flag in flags:
    result = subprocess.run(
        ["python", "baby.py"],
        input=flag,
        capture_output=True,
        text=True,
        timeout=2
    )
    output = result.stdout + result.stderr
    
    has_not = "not the flag" in output.lower()
    if not has_not:
        print(f"✓ POTENTIAL MATCH: {flag}")
        print(f"  Output: {output[:200]}")
    else:
        print(f"✗ {flag}")