#!/usr/bin/env python3
import os
from typing import Iterable

# Import generators from seeders
from seed_math_models import gen_math_models
from seed_agents import gen_agents
from seed_language import gen_language
from seed_prophecy import gen_prophecy
from seed_marketing import gen_marketing

OUTDIR = os.path.join(os.path.dirname(__file__), 'enumerations')
os.makedirs(OUTDIR, exist_ok=True)

def write_lines(path: str, lines: Iterable[str], n: int):
    with open(path, 'w', encoding='utf-8') as f:
        i = 0
        for line in lines:
            i += 1
            txt = line if line.endswith('.') else f"{line}."
            f.write(txt + "\n")
            if i >= n:
                break

write_lines(os.path.join(OUTDIR, 'math_model_500.txt'), gen_math_models(500), 500)
write_lines(os.path.join(OUTDIR, 'agent_500.txt'), gen_agents(500), 500)
write_lines(os.path.join(OUTDIR, 'language_500.txt'), gen_language(500), 500)
write_lines(os.path.join(OUTDIR, 'prophecy_500.txt'), gen_prophecy(500), 500)
write_lines(os.path.join(OUTDIR, 'marketing_500.txt'), gen_marketing(500), 500)

print("Generated 500-line enumerations in:")
for fn in [
    'math_model_500.txt',
    'agent_500.txt',
    'language_500.txt',
    'prophecy_500.txt',
    'marketing_500.txt',
]:
    print(os.path.join(OUTDIR, fn))