Rotation.swift

 override func viewWillTransition(to size: CGSize,
                                     with coordinator: UIViewControllerTransitionCoordinator) {
        if UIDevice.current.orientation.isLandscape {
            print("Landscape")
            chatButton.removeTarget(self, action: #selector(handleChatButtonClick), for: .touchUpInside)
        } else {
            print("Portrait")
            chatButton.addTarget(self, action: #selector(handleChatButtonClick), for: .touchUpInside)
        }
    }