Final fixes for PR comments

Author: wenli-cai

workflow-trace-viewer/src/jvmMain/kotlin/com/squareup/workflow1/traceviewer/App.kt

@@ -10,9 +10,9 @@ import androidx.compose.runtime.setValue
 import androidx.compose.ui.Alignment
 import androidx.compose.ui.Modifier
 import com.squareup.workflow1.traceviewer.model.Node
+import com.squareup.workflow1.traceviewer.ui.FrameSelectTab
 import com.squareup.workflow1.traceviewer.ui.RenderDiagram
 import com.squareup.workflow1.traceviewer.ui.RightInfoPanel
-import com.squareup.workflow1.traceviewer.ui.StateSelectTab
 import com.squareup.workflow1.traceviewer.util.SandboxBackground
 import com.squareup.workflow1.traceviewer.util.UploadFile
 import io.github.vinceglb.filekit.PlatformFile
@@ -44,7 +44,7 @@ public fun App(
       }
     }
 
-    StateSelectTab(
+    FrameSelectTab(
       frames = workflowFrames,
       currentIndex = frameIndex,
       onIndexChange = { frameIndex = it },

workflow-trace-viewer/src/jvmMain/kotlin/com/squareup/workflow1/traceviewer/model/WorkflowNode.kt renamed to workflow-trace-viewer/src/jvmMain/kotlin/com/squareup/workflow1/traceviewer/model/Node.kt

@@ -18,7 +18,7 @@ import com.squareup.workflow1.traceviewer.model.Node
  * A trace tab selector that allows devs to switch between different states within the provided trace.
  */
 @Composable
-public fun StateSelectTab(
+public fun FrameSelectTab(
   frames: List<Node>,
   currentIndex: Int,
   onIndexChange: (Int) -> Unit,

workflow-trace-viewer/src/jvmMain/kotlin/com/squareup/workflow1/traceviewer/ui/FrameSelectTab.kt

@@ -43,15 +43,17 @@ public fun RenderDiagram(
   LaunchedEffect(traceFile) {
     val parseResult = parseTrace(traceFile)
 
-    if (parseResult is ParseResult.Failure) {
-      error = parseResult.error
-      return@LaunchedEffect
+    when (parseResult) {
+      is ParseResult.Failure -> {
+        error = parseResult.error
+      }
+      is ParseResult.Success -> {
+        val parsedFrames = parseResult.trace ?: emptyList()
+        frames = parsedFrames
+        onFileParse(parsedFrames)
+        isLoading = false
+      }
     }
-
-    val parsedFrames = (parseResult as ParseResult.Success).trace ?: emptyList()
-    frames = parsedFrames
-    onFileParse(parsedFrames)
-    isLoading = false
   }
 
   if (error != null) {