[mob][cast] Fix cast progress state

mobile/lib/ui/cast/auto.dart

@@ -79,12 +79,6 @@ class _AutoCastDialogState extends State<AutoCastDialog> {
                       });
                       try {
                         await _connectToYourApp(context, device);
-                        if (mounted) {
-                          setState(() {
-                            _isDeviceTapInProgress.remove(device);
-                          });
-                          Navigator.of(context).pop();
-                        }
                       } catch (e) {
                         if (mounted) {
                           setState(() {
@@ -128,6 +122,11 @@ class _AutoCastDialogState extends State<AutoCastDialog> {
           final code = message[CastMessageType.pairCode]!['code'];
           widget.onConnect(code);
         }
+        if (mounted) {
+          setState(() {
+            _isDeviceTapInProgress.remove(castDevice);
+          });
+        }
       },
     );
   }

mobile/lib/ui/viewer/gallery/gallery_app_bar_widget.dart

@@ -758,6 +758,7 @@ class _GalleryAppBarWidgetState extends State<GalleryAppBarWidget> {
           return AutoCastDialog(
             (device) async {
               await _castPair(bContext, gw, device);
+              Navigator.pop(bContext);
             },
           );
         },