Fix connect callback

Author: javizqh

react_frontend/src/components/buttons/PlayPause.tsx

@@ -82,8 +82,9 @@ const PlayPauseButton = ({
     if (manager.getState() === states.IDLE) {
       info("Connecting with the Robotics Backend ...");
       connectManager(states.TOOLS_READY, () => {
-        onAppStateChange();
+        setLoading(false);
         close();
+        onAppStateChange();
       });
       return;
     }

react_frontend/src/components/layouts/ExerciseContainer.tsx

@@ -168,7 +168,7 @@ const ExerciseContainer = ({
       }
     } catch (e: unknown) {
       console.log("Connection failed, trying again!");
-      setTimeout(connectWithRetry, 2000);
+      setTimeout(connectWithRetry, 2000, desiredState, callback);
     }
   };