From dfecc62ce3eb211c9c0b843dc04fa3ae9da45e2a Mon Sep 17 00:00:00 2001
From: Will Chen <willchen90@gmail.com>
Date: Tue, 15 Apr 2025 21:32:51 -0700
Subject: [PATCH] Ensure error message is cleared upon refreshing or restarting
 app

---
 src/components/preview_panel/PreviewIframe.tsx | 1 +
 src/components/preview_panel/PreviewPanel.tsx  | 3 +++
 2 files changed, 4 insertions(+)

diff --git a/src/components/preview_panel/PreviewIframe.tsx b/src/components/preview_panel/PreviewIframe.tsx
index ca7815d..9b07007 100644
--- a/src/components/preview_panel/PreviewIframe.tsx
+++ b/src/components/preview_panel/PreviewIframe.tsx
@@ -273,6 +273,7 @@ export const PreviewIframe = ({
   // Function to handle reload
   const handleReload = () => {
     setReloadKey((prevKey) => prevKey + 1);
+    setErrorMessage(undefined);
     // Optionally, add logic here if you need to explicitly stop/start the app again
     // For now, just changing the key should remount the iframe
     console.debug("Reloading iframe preview for app", selectedAppId);
diff --git a/src/components/preview_panel/PreviewPanel.tsx b/src/components/preview_panel/PreviewPanel.tsx
index 61c5bae..ce3da4f 100644
--- a/src/components/preview_panel/PreviewPanel.tsx
+++ b/src/components/preview_panel/PreviewPanel.tsx
@@ -99,10 +99,12 @@ export function PreviewPanel() {
   const [isConsoleOpen, setIsConsoleOpen] = useState(false);
   const { runApp, stopApp, restartApp, error, loading, app } = useRunApp();
   const runningAppIdRef = useRef<number | null>(null);
+  const [key, setKey] = useState(0);
 
   const handleRestart = useCallback(() => {
     if (selectedAppId !== null) {
       restartApp(selectedAppId);
+      setKey((prevKey) => prevKey + 1);
     }
   }, [selectedAppId, restartApp]);
 
@@ -163,6 +165,7 @@ export function PreviewPanel() {
             <div className="h-full overflow-y-auto">
               {previewMode === "preview" ? (
                 <PreviewIframe
+                  key={key}
                   loading={loading}
                   loadingErrorMessage={error?.message}
                 />