import asyncio
from services.ollama_service import OllamaService

async def test():
    service = OllamaService()
    text = '''Once upon a time, in a tiny village surrounded by lush green forests, there lived three best friends: Lily the rabbit, Benny the bear, and Sammy the squirrel.

As they sat down on a blanket, Lily asked, "Have you guys heard about the new stream that just appeared in the woods?"

Benny's eyes lit up. "Ooh, I love streams! Do we get to go see it now?"

Sammy nodded enthusiastically. "I've been wanting to explore the forest more since the snow melted. Let's go!"'''
    
    print("Testing Ollama speaker detection...")
    speakers = await service.detect_speakers(text)
    print(f'Detected speakers: {speakers}')

asyncio.run(test())