Swiftui-ReviewDetail-NLP.swift

struct ReviewDetail: View {
    var review: String
    
    @State var mlPrediction = ""
    @State var nlpResult = ""
    let tagger = NLTagger(tagSchemes: [.sentimentScore])
    
    var reviewPredictor : NLModel?
    
    var body: some View {
        
        return VStack{
            Text(review)
            Button(action:{
                
                self.tagger.string = self.review
                if let string = self.tagger.string{
                    let (sentiment,_) = self.tagger.tag(at: string.startIndex, unit: .paragraph, scheme: .sentimentScore)
                    
                    self.nlpResult = "\(sentiment?.rawValue ?? "")"
                }
                
            }){
                Text("Run Sentiment Analysis")
            }
            Text(self.nlpResult)
            Button(action:{
                self.runPrediction()
                
            }){
                Text("Run Core ML Prediction")
            }
            Text(self.mlPrediction)
        }
    }
    
    func runPrediction(){
        do{
            let reviewPredictor = try NLModel(mlModel: ReviewTextClassifier().model)
            let label = reviewPredictor.predictedLabel(for: self.review)
            
            self.mlPrediction = label ?? ""
            
        }catch(let error){
            print("error is \(error.localizedDescription)")
        }
    }
}