#!/usr/bin/env python3
"""Final fixes for stat7_entity.py and verify the fixes work"""

# Fix the stat7_entity.py bug
with open("warbler_cda/stat7_entity.py", "r", encoding="utf-8") as f:
    content = f.read()

# Fix the description reference bug
content = content.replace('"description": description,', '"description": self.description,')

# Write back the fixed content
with open("warbler_cda/stat7_entity.py", "w", encoding="utf-8") as f:
    f.write(content)

print("Fixed stat7_entity.py description bug")

# Test import to make sure everything works
try:
    print("✅ stat7_entity imports successfully")
except Exception as e:
    print(f"❌ stat7_entity import failed: {e}")

try:
    print("✅ stat7_rag_bridge imports successfully")
except Exception as e:
    print(f"❌ stat7_rag_bridge import failed: {e}")

print("All fixes applied!")