Better error handling in HealthGPTView

StanfordBDHG · Mar 27, 2024 · 56c479e · 56c479e
1 parent 971a1cb
commit 56c479e
Show file tree

Hide file tree

Showing 3 changed files with 39 additions and 3 deletions.
diff --git a/HealthGPT/HealthGPT/HealthDataInterpreter.swift b/HealthGPT/HealthGPT/HealthDataInterpreter.swift
@@ -52,7 +52,6 @@ class HealthDataInterpreter: DefaultInitializable, Module, EnvironmentAccessible
         }
 
         let llm = llmRunner(with: llmSchema)
-
         systemPrompt = try await generateSystemPrompt()
         llm.context.append(systemMessage: systemPrompt)
         self.llm = llm

diff --git a/HealthGPT/HealthGPT/HealthGPTView.swift b/HealthGPT/HealthGPT/HealthGPTView.swift
@@ -20,6 +20,8 @@ struct HealthGPTView: View {
 
     @Environment(HealthDataInterpreter.self) private var healthDataInterpreter
     @State private var showSettings = false
+    @State private var showErrorAlert = false
+    @State private var errorMessage = ""
 
     var body: some View {
         NavigationStack {
@@ -38,7 +40,12 @@ struct HealthGPTView: View {
                     .onChange(of: llm.context, initial: true) { _, _ in
                         Task {
                             if llm.state != .generating {
-                                try? await healthDataInterpreter.queryLLM()
+                                do {
+                                    try await healthDataInterpreter.queryLLM()
+                                } catch {
+                                    showErrorAlert = true
+                                    errorMessage = "Error querying LLM: \(error.localizedDescription)"
+                                }
                             }
                         }
                     }
@@ -52,8 +59,18 @@ struct HealthGPTView: View {
         .sheet(isPresented: $showSettings) {
             SettingsView()
         }
+        .alert("ERROR_ALERT_TITLE", isPresented: $showErrorAlert) {
+            Button("ERROR_ALERT_CANCEL", role: .cancel) {}
+        } message: {
+            Text(errorMessage)
+        }
         .task {
-            try? await healthDataInterpreter.prepareLLM(with: openAIModel)
+            do {
+                try await healthDataInterpreter.prepareLLM(with: openAIModel)
+            } catch {
+                showErrorAlert = true
+                errorMessage = "Error preparing LLM: \(error.localizedDescription)"
+            }
         }
     }
 

diff --git a/HealthGPT/Supporting Files/Localizable.xcstrings b/HealthGPT/Supporting Files/Localizable.xcstrings
@@ -24,6 +24,26 @@
         }
       }
     },
+    "ERROR_ALERT_CANCEL" : {
+      "localizations" : {
+        "en" : {
+          "stringUnit" : {
+            "state" : "translated",
+            "value" : "OK"
+          }
+        }
+      }
+    },
+    "ERROR_ALERT_TITLE" : {
+      "localizations" : {
+        "en" : {
+          "stringUnit" : {
+            "state" : "translated",
+            "value" : "Error"
+          }
+        }
+      }
+    },
     "HEALTHKIT_PERMISSIONS_BUTTON" : {
       "localizations" : {
         "en" : {