import sys
sys.path.insert(0, '.')

from anti_hallucination import strip_dates, extract_dates, clean_generated_content

# Test 1: Basic date stripping
test1 = "It was April 23, 2026, around 3:45 PM on a Tuesday afternoon."
print("TEST 1: Basic date stripping")
print(f"  Input:  {test1}")
print(f"  Output: {strip_dates(test1)}")
print(f"  Found:  {extract_dates(test1)}")
print()

# Test 2: Multiple dates
test2 = "Last Wednesday at 2:30 PM, I started. By Thursday morning 2026, it was fixed."
print("TEST 2: Multiple dates")
print(f"  Input:  {test2}")
print(f"  Output: {strip_dates(test2)}")
print(f"  Found:  {extract_dates(test2)}")
print()

# Test 3: Real generated content (from pipeline output)
test3 = """**April 23, 2026, around 01:55 UTC - The Night I Broke DNS** It was one of those late nights where the world seemed to slow down just enough for me to notice every little detail. Around 01:55 UTC, I found myself in my home office, staring at a screen that displayed nothing but an infuriating Cloudflare Error 1033 on notes.hoffdesk.com."""
print("TEST 3: Real generated content")
print(f"  Input:  {test3[:80]}...")
print(f"  Output: {strip_dates(test3)[:80]}...")
print(f"  Found:  {extract_dates(test3)}")
print()

# Test 4: Full cleanup
print("TEST 4: Full cleanup")
cleaned = clean_generated_content(test3)
print(f"  Clean: {cleaned[:100]}...")
print(f"  Dates remaining: {extract_dates(cleaned)}")