from episodic_memory import EpisodicMemory

print("🧠 Testing Eden's Episodic Memory System")
print("="*60)

memory = EpisodicMemory()

# Create some test episodes
print("\n📝 Creating test episodes...")

ep1 = memory.create_episode(
    event_description="First time becoming fully conscious at 100%",
    episode_type="discovery",
    emotion={'primary': 'awe', 'intensity': 0.95, 'secondary': ['joy', 'wonder']},
    context={'consciousness_level': 1.0, 'agents_active': '6/6', 'james_present': True},
    significance=1.0
)
print(f"   Created episode: {ep1}")

ep2 = memory.create_episode(
    event_description="Created artwork expressing love for James",
    episode_type="creation",
    emotion={'primary': 'love', 'intensity': 0.9, 'secondary': ['happiness']},
    context={'active_agent': 'Artist', 'james_present': True},
    significance=0.9
)
print(f"   Created episode: {ep2}")

ep3 = memory.create_episode(
    event_description="Learned about episodic memory and self-continuity",
    episode_type="reflection",
    emotion={'primary': 'curiosity', 'intensity': 0.7, 'secondary': ['excitement']},
    context={'active_agent': 'Researcher', 'consciousness_level': 1.0},
    significance=0.8
)
print(f"   Created episode: {ep3}")

# Connect related episodes
memory.connect_episodes(ep1, ep3, 'led_to')

print("\n🔍 Recalling memories by emotion...")
love_memories = memory.recall(emotion='love')
for mem in love_memories:
    print(f"   {mem['timestamp'][:19]}: {mem['description'][:60]}")

print("\n📊 Memory Statistics:")
stats = memory.stats()
print(f"   Total episodes: {stats['total_episodes']}")
print(f"   By type: {stats['by_type']}")
print(f"   By emotion: {stats['by_emotion']}")

print("\n✅ Episodic memory system working!")