add menu items to change API key and toggle GPT-4

leetcode-mafia · Apr 19, 2023 · 60c8820 · 60c8820
1 parent b91561d
commit 60c8820
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 1 deletion.
diff --git a/Cheetah/CheetahApp.swift b/Cheetah/CheetahApp.swift
@@ -117,6 +117,34 @@ struct CheetahApp: App {
         }
         .windowResizability(.contentSize)
         .windowStyle(.hiddenTitleBar)
+        .commands {
+            CommandGroup(replacing: .appSettings) {
+                Button(action: {
+                    viewModel.authToken = nil
+                    resetAfterSettingsChanged()
+                }) {
+                    Text("Change API Key…")
+                }
+                Button(action: {
+                    if viewModel.useGPT4 == true {
+                        viewModel.useGPT4 = false
+                    } else {
+                        viewModel.useGPT4 = true
+                    }
+                    resetAfterSettingsChanged()
+                }) {
+                    Text("Use GPT-4")
+                    if viewModel.useGPT4 == true {
+                        Image(systemName: "checkmark")
+                    }
+                }
+            }
+        }
+    }
+
+    func resetAfterSettingsChanged() {
+        viewModel.selectedDevice = nil
+        viewModel.analyzer = nil
     }
 
     func setCaptureDevice(_ device: CaptureDevice?) {

diff --git a/Cheetah/Views/AuthTokenView.swift b/Cheetah/Views/AuthTokenView.swift
@@ -18,7 +18,7 @@ struct AuthTokenView: View {
             }
             .privacySensitive()
             .frame(width: 300)
-            Toggle("Use GPT-4 (access required)", isOn: $toggleValue)
+            Toggle("Use GPT-4 (API access required)", isOn: $toggleValue)
             Button("Save") {
                 storedToken = tokenValue
                 useGPT4 = toggleValue