Hello,
There is a high view to implement VNet integration to restrict traffic in your case :
- Create the virtual Network
- Create a NSG to define the rules to restrict traffic as you need
- Asssociate the NSG to the subnet
- Implement VNet integration or private endpoint features on API, Function and Storage Services on
More info on how to ingrate Azure Services with VNet :
https://learn.microsoft.com/en-us/azure/virtual-network/vnet-integration-for-azure-services
You can also consider implement Azure Policy to further enhance security.
If this answer helped, please mark it as accepted for other users with similar question to be able to find this topic.
Regards