Moderation demo view

jamesrochabrun · Oct 24, 2023 · 27a2c31 · 27a2c31
1 parent 7c1277d
commit 27a2c31
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 5 deletions.
diff --git a/...le/SwiftOpenAIExample.xcodeproj/project.xcworkspace/xcshareddata/swiftpm/Package.resolved b/...le/SwiftOpenAIExample.xcodeproj/project.xcworkspace/xcshareddata/swiftpm/Package.resolved
@@ -6,7 +6,7 @@
       "location" : "https://github.com/jamesrochabrun/SwiftOpenAI",
       "state" : {
         "branch" : "main",
-        "revision" : "75cb57017618400ccde01834777f201358839a45"
+        "revision" : "7c1277d50e05521c7fe61e5ea3c58bbaa4a91aea"
       }
     }
   ],

diff --git a/Examples/SwiftOpenAIExample/SwiftOpenAIExample/ModerationsDemo/ModerationDemoView.swift b/Examples/SwiftOpenAIExample/SwiftOpenAIExample/ModerationsDemo/ModerationDemoView.swift
@@ -13,16 +13,32 @@ struct ModerationDemoView: View {
    @State private var moderationProvider: ModerationProvider
    @State private var isLoading = false
    @State private var prompt = ""
-
+   @State private var errorMessage = ""
+
    init(service: OpenAIService) {
       _moderationProvider = State(initialValue: ModerationProvider(service: service))
    }
 
    var body: some View {
       VStack {
          textArea
+         if moderationProvider.isFlagged {
+            Text("That is not a nice thing to say.")
+         }
+         if !errorMessage.isEmpty {
+            Text("Error \(errorMessage)")
+               .bold()
+         }
       }
-
+      .overlay(
+         Group {
+            if isLoading {
+               ProgressView()
+            } else {
+               EmptyView()
+            }
+         }
+      )
    }
 
    var textArea: some View {
@@ -36,6 +52,8 @@ struct ModerationDemoView: View {
                defer { isLoading = false }  // ensure isLoading is set to false when the
                do {
                   try await moderationProvider.createModerationFromText(parameters: .init(input: prompt))
+               } catch {
+                  errorMessage = "\(error)"
                }
 
             }

diff --git a/Examples/SwiftOpenAIExample/SwiftOpenAIExample/OptionsListView.swift b/Examples/SwiftOpenAIExample/SwiftOpenAIExample/OptionsListView.swift
@@ -53,8 +53,8 @@ struct OptionsListView: View {
                      ImagesDemoView(service: openAIService)
                   case .models:
                      ModelsDemoView(service: openAIService)
-                  default:
-                     EmptyView()
+                  case .moderations:
+                     ModerationDemoView(service: openAIService)
                   }
                }
             }