Breast compressions (gently squeezing the breast) while bub is attached can keep her interested in what is going on. I used to rub baby's cheek with one finger to keep them focussed.

I think you are sometimes feeding by bottle? Some bubs find the bottle a bit easier, and so need a bit more encouragement on the breast. The ABA has a guide to the caring for the breastfed baby that describes how you can feed by bottle and still allow bub to take breaks and control the feed to make it easier for them to switch between the two. https://www.breastfeeding.asn.au/bfinfo/caregivers.html