from src.data import make_loaders

def main():
    train_loader, val_loader, stoi = make_loaders(batch_size=4, img_height=64)

    batch = next(iter(train_loader))
    print("Vocab size:", len(stoi))
    print("Images:", batch.images.shape)          # [B, 1, H, Wmax]
    print("Image widths:", batch.image_widths)    # [B]
    print("Targets shape:", batch.targets.shape)  # [sum(T)]
    print("Target lengths:", batch.target_lengths)
    print("Sample texts:", batch.texts[:2])

if __name__ == "__main__":
    main()