Swift 3 how to limit user to only taking photos and NOT videos - iOS, Xcode

Question

My photo app crashes when user selects video. The user can select photo and it works fine, but I'd like to remove the option to take a video. How do I disable video? The relevant chunk of code is below...

@IBAction func choosePhoto(_ sender: UIButton) {

    let alertController = UIAlertController(title: "Title", message: "Message", preferredStyle: .actionSheet)
    imagePicker = UIImagePickerController()
    imagePicker.delegate = self

    if (UIImagePickerController.isSourceTypeAvailable(.camera)) {
        let cameraAction = UIAlertAction(title: "Use Camera", style: .default) { (action) in

            let status = AVCaptureDevice.authorizationStatus(forMediaType: AVMediaTypeVideo)

            if (status == .authorized) {
                self.displayPicker(type: .camera)
            }

            if (status == .restricted) {

                self.handleRestricted()
            }

            if (status == .denied) {

                self.handleDenied()
            }

            if (status == .notDetermined) {

                AVCaptureDevice.requestAccess(forMediaType: AVMediaTypeVideo, completionHandler: { (granted) in
                    if (granted) {
                        self.displayPicker(type: .camera)
                    }

                })
            }

        }

    alertController.addAction(cameraAction)
    }

    if (UIImagePickerController.isSourceTypeAvailable(.photoLibrary)) {
        let photoLibraryAction = UIAlertAction(title: "Use Photo Library", style: .default) { (action) in

            let status = PHPhotoLibrary.authorizationStatus()

            if (status == .authorized) {
                self.displayPicker(type: .photoLibrary)
            }

            if (status == .restricted) {

                self.handleRestricted()
            }

            if (status == .denied) {

                self.handleDenied()
            }

            if (status == .notDetermined) {

                PHPhotoLibrary.requestAuthorization({ (status) in
                    if (status == PHAuthorizationStatus.authorized) {
                        self.displayPicker(type: .photoLibrary)
                        }
                    })

                }


            }

        alertController.addAction(photoLibraryAction)
    }

    let cancelAction = UIAlertAction(title: "Cancel", style: .cancel, handler: nil)

    alertController.addAction(cancelAction)

    present(alertController, animated: true, completion: nil)
}

func handleDenied() {

    let alertController = UIAlertController(title: "Camera Access Denied", message: "This app does not have acces to your device's camera. Please update your settings.", preferredStyle: .alert)

    let settingsAction = UIAlertAction(title: "Go To Settings", style: .default) { (action) in
        DispatchQueue.main.async {
            UIApplication.shared.open(NSURL(string: UIApplicationOpenSettingsURLString)! as URL)
        }
    }

    let cancelAction = UIAlertAction(title: "Cancel", style: .cancel, handler: nil)

    alertController.popoverPresentationController?.sourceView = self.photoImage
    alertController.popoverPresentationController?.sourceRect = self.photoImage.bounds

    alertController.addAction(settingsAction)
    alertController.addAction(cancelAction)
    present(alertController, animated: true, completion: nil)
}

func handleRestricted() {

    let alertController = UIAlertController(title: "Camera Access Denied", message: "This device is restricted from accessing the camera at this time", preferredStyle: .alert)

    let defaultAction = UIAlertAction(title: "OK", style: .default, handler: nil)

    alertController.popoverPresentationController?.sourceView = self.photoImage
    alertController.popoverPresentationController?.sourceRect = self.photoImage.bounds

    alertController.addAction(defaultAction)
    present(alertController, animated: true, completion: nil)

}

func displayPicker(type: UIImagePickerControllerSourceType) {
    self.imagePicker.mediaTypes = UIImagePickerController.availableMediaTypes(for: type)!
    self.imagePicker.sourceType = type
    self.imagePicker.allowsEditing = true

    DispatchQueue.main.async {
        self.present(self.imagePicker, animated: true, completion: nil)
    }
}

func imagePickerController(_ picker: UIImagePickerController, didFinishPickingMediaWithInfo info: [String : Any]) {
    let chosenImage = info[UIImagePickerControllerEditedImage] as! UIImage
    photoImage.contentMode = .scaleAspectFill
    photoImage.image = chosenImage
    dismiss(animated: true, completion: nil)
}

func imagePickerControllerDidCancel(_ picker: UIImagePickerController) {
    dismiss(animated: true, completion: nil)
}

Answer 1

Put this line right after imagePicker.delegate = self

imagePicker.mediaTypes = [kUTTypeImage as String] 

Remember to add import MobileCoreServices at the beginning of the file.

Answer 2

What you want to do is after you init imagePicker set the following property

imagePicker.mediaTypes = [UIImagePickerController availableMediaTypesForSourceType: UIImagePickerControllerSourceTypeCamera];

https://developer.apple.com/documentation/uikit/uiimagepickercontroller/1619173-mediatypes