如何在Swift拍摄的图片上插入文本？

import UIKit

func addTextToImage(image: UIImage, text: String) -> UIImage? {
    // 设置文本样式
    let textStyle = NSMutableParagraphStyle()
    textStyle.alignment = .center
    let textColor = UIColor.white
    let textFont = UIFont(name: "Helvetica Bold", size: 30)!
    
    // 创建一个图形上下文
    UIGraphicsBeginImageContext(image.size)
    
    // 将图片绘制到图形上下文中
    image.draw(in: CGRect(x: 0, y: 0, width: image.size.width, height: image.size.height))
    
    // 创建文本属性
    let textAttributes: [NSAttributedString.Key: Any] = [
        .font: textFont,
        .foregroundColor: textColor,
        .paragraphStyle: textStyle
    ]
    
    // 计算文本绘制的位置
    let textSize = text.size(withAttributes: textAttributes)
    let textX = (image.size.width - textSize.width) / 2
    let textY = (image.size.height - textSize.height) / 2
    
    // 在图形上下文中绘制文本
    text.draw(at: CGPoint(x: textX, y: textY), withAttributes: textAttributes)
    
    // 从图形上下文中获取新的图片
    let newImage = UIGraphicsGetImageFromCurrentImageContext()
    
    // 结束图形上下文
    UIGraphicsEndImageContext()
    
    return newImage
}

// 使用示例
let originalImage = UIImage(named: "originalImage.jpg")
let text = "Hello, World!"
let modifiedImage = addTextToImage(image: originalImage, text: text)

// 将修改后的图片显示在UIImageView中
let imageView = UIImageView(image: modifiedImage)