SensitiveContent.swift

import SwiftUI
import SensitiveContentAnalysis

struct ContentView: View {
    
    enum AnalysisState {
        case notStarted
        case analyzing
        case isSensitive
        case notSensitive
        case error(message: String)
    }
    
    @State private var analysisState: AnalysisState = .notStarted
    @State private var errorMessage = ""
    
    var body: some View {
        VStack {
            HStack(spacing: 100) {
                Button {
                    Task {
                        await analyze(imageName: "pitt")
                    }
                } label: {
                    Text("Pitt")
                        .font(.largeTitle)
                }
                
                Button {
                    Task {
                        await analyze(imageName: "lena")
                    }
                } label: {
                    Text("Lena")
                        .font(.largeTitle)
                }
            }
            .padding(.bottom, 50)
            
            Group {
                switch analysisState {
                case .notStarted:
                    EmptyView()
                case .isSensitive:
                    Text("Warning! Image contains nudity ⚠️")
                case .notSensitive:
                    Text("The image is safe ✅")
                case .analyzing:
                    ProgressView()
                case .error(let message):
                    Text("Error analyzing image. \(message)")
                        .foregroundStyle(Color.red)
                }
            }
            .font(.largeTitle)
        }
        .padding()
    }

    func analyze(imageName: String) async {
        analysisState = .analyzing
        let analyzer = SCSensitivityAnalyzer()
        let policy = analyzer.analysisPolicy
        
        if policy == .disabled {
            print("Policy is disabled")
            analysisState = .error(message: "Policy is disabled")
            return
        }
        
        do {
            
            
            guard let image = UIImage(named: imageName)
            else {
                print("Image not found")
                analysisState = .error(message: "Image not found")
                return
            }
            
            let response = try await analyzer.analyzeImage(image.cgImage!)
            
            analysisState = if response.isSensitive {
                .isSensitive
            } else {
                .notSensitive
            }
            
        } catch {
            analysisState = .error(message: error.localizedDescription)
            print("Unable to get a response", error)
        }
    }
}

#Preview {
    ContentView()
}