From 18846946b50a458b005d04a5bc4cdaa07a09b1fd Mon Sep 17 00:00:00 2001
From: Will Chen <willchen90@gmail.com>
Date: Thu, 1 May 2025 23:18:19 -0700
Subject: [PATCH] De-flake preview iframe error msg removal (#69)

---
 src/components/preview_panel/PreviewIframe.tsx | 3 +++
 src/hooks/useRunApp.ts                         | 6 ++----
 2 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/src/components/preview_panel/PreviewIframe.tsx b/src/components/preview_panel/PreviewIframe.tsx
index ebe34ca..679f137 100644
--- a/src/components/preview_panel/PreviewIframe.tsx
+++ b/src/components/preview_panel/PreviewIframe.tsx
@@ -414,6 +414,9 @@ export const PreviewIframe = ({ loading }: { loading: boolean }) => {
           </div>
         ) : (
           <iframe
+            onLoad={() => {
+              setErrorMessage(undefined);
+            }}
             ref={iframeRef}
             key={reloadKey}
             title={`Preview for App ${selectedAppId}`}
diff --git a/src/hooks/useRunApp.ts b/src/hooks/useRunApp.ts
index 41adf1f..effdae4 100644
--- a/src/hooks/useRunApp.ts
+++ b/src/hooks/useRunApp.ts
@@ -82,8 +82,7 @@ export function useRunApp() {
 
   const onHotModuleReload = useCallback(() => {
     setPreviewPanelKey((prevKey) => prevKey + 1);
-    setPreviewErrorMessage(undefined);
-  }, [setPreviewPanelKey, setPreviewErrorMessage]);
+  }, [setPreviewPanelKey]);
 
   const restartApp = useCallback(
     async ({
@@ -151,8 +150,7 @@ export function useRunApp() {
 
   const refreshAppIframe = useCallback(async () => {
     setPreviewPanelKey((prevKey) => prevKey + 1);
-    setPreviewErrorMessage(undefined);
-  }, [setPreviewPanelKey, setPreviewErrorMessage]);
+  }, [setPreviewPanelKey]);
 
   return {
     loading,