Yes, I think the PAA may prefer a larger design, not just in response to Ukraine, but that the standard was the AH-1Z.
I know an engineer at Bell, and he was saying that the PAA gave top marks to the Viper's performance in hot temperatures, deserts, and high altitudes. Moreover, Bell was also confident of a cross-sell opportunity in the UH-1Y (85% commonality with the AH-1Z) and Bell-412M (building on the Bell-412EP).
IMO, something like the Z-21, T929, or AH249 would not only provide the extra armor protection and payload, but some additional versatility. You can carry more EW/ECM equipment, possibly small cruise missiles, a greater sensor load (mmW), etc.
For anything smaller (5-7-ton-range), one's likely better off investing in a rotary UCAV that can be cheaper to build and easier to replace at scale.