The egg laying health issues are very unusual. In future make certain that is actual problem to prevent culling for wrong reasons.
Feed you are using is fine if birds are free-range but not if confined. Up protein by including a balanced formulation that is higher than targeted amount of protein intended for birds to consume. My males on a maintenance ration I like have consuming a protein level of about 13% which can be realized by mixing a chick grower with a mixture of whole grains. Grains I use include millet, whole oats, BOSS, whole corn and whatever I can scratch up. Hens in lay will need a richer higher protein mix of about 14 to 15% crude protein with a free-choice side of oyster shell. Always make sure birds have access to some greens. You will have to do some math to figure out how much of each feedstuff to mixture.