Theme app UI extension block is disappearing when an app embed block is added in the theme editor

[rohith-siddhart]

Hi there,
I've created a Shopify app which includes theme app UI extensions ( a button on a page). I have also created an app embed block which has a JS script for managing the functionality of the theme app UI extensions. This setup worked fine on development stores
Later, I've installed this app on a live store and added the theme app extensions using theme editor and it worked fine. After this I've enabled the app embed in the theme editor.  Both ui extension blocks and app embed blocks were working fine on the theme editor.
But when I open the live store, I see theme extensions appear and then disappear immediately as soon as the page loads. When I disable the app embed and try then they don't disappear
Can someone please help me solve this?
Thanks in advance