In your CreateWidget node there is an ‘Owning Player’ input, try providing your PlayerController(0) here. Then, save the widget’s reference into a variable. Variable that you will use for the ‘In Widget To Focus’ parameter of the SetInputModeUIOnly function.
Also, if you are testing your game in PIE mode (Play In Editor), you may try to launch your game in standalone mode instead.