stax: add UI implementation

src/handle_swap_sign_transaction.c

@@ -67,7 +67,13 @@ void handle_swap_sign_transaction(chain_config_t* config) {
         nvm_write((void*) &N_storage, (void*) &storage, sizeof(internalStorage_t));
     }
 
+#ifdef HAVE_BAGL
     UX_INIT();
+#endif  // HAVE_BAGL
+#ifdef HAVE_NBGL
+    nbgl_objInit();
+#endif  // HAVE_NBGL
+
     USB_power(0);
     USB_power(1);
     // ui_idle();