Fix Android PlotPanelRaw not showing plot after an error message

Author: IKupriyanov-HORIS

lets-plot-compose/src/androidMain/kotlin/org/jetbrains/letsPlot/compose/PlotPanelRaw.kt

@@ -53,10 +53,21 @@ actual fun PlotPanelRaw(
     val processedPlotSpec = remember(rawSpec.hashCode()) {
         processRawSpecs(rawSpec, frontendOnly = false)
     }
-    var errorMessage: String? by remember { mutableStateOf(null) }
 
-    if (PlotConfig.isFailure(processedPlotSpec)) {
-        errorMessage = PlotConfig.getErrorMessage(processedPlotSpec)
+    var errorMessage: String? by remember(processedPlotSpec) { mutableStateOf(null) } // Reset error on spec change
+
+    LaunchedEffect(processedPlotSpec, sizingPolicy, computationMessagesHandler) {
+        runCatching {
+            if (PlotConfig.isFailure(processedPlotSpec)) {
+                errorMessage = PlotConfig.getErrorMessage(processedPlotSpec)
+            } else {
+                plotCanvasFigure?.update(processedPlotSpec, sizingPolicy, computationMessagesHandler)
+                    ?: LOG.info { "Error updating plot figure - plotCanvasFigure is null" }
+            }
+        }.onFailure { e ->
+            errorMessage = e.message ?: "Unknown error: ${e::class.simpleName}"
+            LOG.error(e) { "Error updating plot figure" }
+        }
     }
 
     // Background
@@ -74,17 +85,6 @@ actual fun PlotPanelRaw(
         }
     }
 
-    LaunchedEffect(processedPlotSpec, sizingPolicy, computationMessagesHandler) {
-        runCatching {
-            plotCanvasFigure?.update(processedPlotSpec, sizingPolicy, computationMessagesHandler)
-                ?: LOG.info { "Error updating plot figure - plotCanvasFigure is null" }
-        }.onFailure { e ->
-            errorMessage = e.message ?: "Unknown error: ${e::class.simpleName}"
-            LOG.error(e) { "Error updating plot figure" }
-        }
-
-    }
-
     errorMessage?.let { errMsg ->
         // Reset the figure to resolve the 'Registration already removed' error.
         // On error, the CanvasView is removed and the plotCanvasFigure changes state to 'detached',
@@ -105,7 +105,6 @@ actual fun PlotPanelRaw(
             modifier = finalModifier,
             factory = { ctx ->
                 plotCanvasFigure = plotCanvasFigure ?: PlotCanvasFigure()
-
                 CanvasView(ctx).apply {
                     figure = plotCanvasFigure
                     onError = { e ->