Run individual simulations in playground

apps/rowboat/app/projects/[projectId]/playground/app.tsx

@@ -1,14 +1,15 @@
 'use client';
 import { Dropdown, DropdownItem, DropdownMenu, DropdownTrigger, Spinner } from "@nextui-org/react";
-import { useEffect, useState, useMemo } from "react";
+import { useEffect, useState, useMemo, useCallback } from "react";
 import { z } from "zod";
-import { PlaygroundChat, SimulationData, Workflow } from "@/app/lib/types";
+import { PlaygroundChat, SimulationData, SimulationScenarioData, Workflow } from "@/app/lib/types";
 import { SimulateScenarioOption, SimulateURLOption } from "./simulation-options";
 import { Chat } from "./chat";
 import { useSearchParams } from "next/navigation";
 import { ActionButton, Pane } from "../workflow/pane";
 import { apiV1 } from "rowboat-shared";
 import { EllipsisVerticalIcon, MessageSquarePlusIcon, PlayIcon } from "lucide-react";
+import { getScenario } from "@/app/actions/simulation_actions";
 
 function SimulateLabel() {
     return <span>Simulate<sup className="pl-1">beta</sup></span>;
@@ -41,9 +42,42 @@ export function App({
         systemMessage: defaultSystemMessage,
     });
 
+    const beginSimulation = useCallback((data: z.infer<typeof SimulationData>) => {
+        setExistingChatId(null);
+        setViewSimulationMenu(false);
+        setCounter(counter + 1);
+        setChat({
+            projectId,
+            createdAt: new Date().toISOString(),
+            messages: [],
+            simulated: true,
+            simulationData: data,
+        });
+    }, [counter, projectId]);
+
+    useEffect(() => {
+        const scenarioId = localStorage.getItem('pendingScenarioId');
+        if (scenarioId && projectId) {
+            console.log('Scenario Effect triggered:', { scenarioId, projectId });
+            getScenario(projectId, scenarioId).then((scenario) => {
+                console.log('Scenario data received:', scenario);
+                beginSimulation(scenario as z.infer<typeof SimulationScenarioData>);
+                localStorage.removeItem('pendingScenarioId');
+            }).catch(error => {
+                console.error('Error fetching scenario:', error);
+                localStorage.removeItem('pendingScenarioId');
+            });
+        }
+    }, [projectId, beginSimulation]);
+
+    if (hidden) {
+        return <></>;
+    }
+
     function handleSimulateButtonClick() {
         setViewSimulationMenu(true);
     }
+
     function handleNewChatButtonClick() {
         setExistingChatId(null);
         setViewSimulationMenu(false);
@@ -56,52 +90,38 @@ export function App({
             systemMessage: defaultSystemMessage,
         });
     }
-    function beginSimulation(data: z.infer<typeof SimulationData>) {
-        setExistingChatId(null);
-        setViewSimulationMenu(false);
-        setCounter(counter + 1);
-        setChat({
-            projectId,
-            createdAt: new Date().toISOString(),
-            messages: [],
-            simulated: true,
-            simulationData: data,
-        });
-    }
 
-    if (hidden) {
-        return <></>;
-    }
-
-    return <Pane title={viewSimulationMenu ? <SimulateLabel /> : "Chat"} actions={[
-        <ActionButton
-            key="new-chat"
-            icon={<MessageSquarePlusIcon size={16} />}
-            onClick={handleNewChatButtonClick}
-        >
-            New chat
-        </ActionButton>,
-        !viewSimulationMenu && <ActionButton
-            key="simulate"
-            icon={<PlayIcon size={16} />}
-            onClick={handleSimulateButtonClick}
-        >
-            Simulate
-        </ActionButton>,
-    ]}>
-        <div className="h-full overflow-auto">
-            {!viewSimulationMenu && loadingChat && <div className="flex justify-center items-center h-full">
-                <Spinner />
-            </div>}
-            {!viewSimulationMenu && !loadingChat && <Chat
-                key={existingChatId || 'chat-' + counter}
-                chat={chat}
-                initialChatId={existingChatId || null}
-                projectId={projectId}
-                workflow={workflow}
-                messageSubscriber={messageSubscriber}
-            />}
-            {viewSimulationMenu && <SimulateScenarioOption beginSimulation={beginSimulation} projectId={projectId} />}
-        </div>
-    </Pane>;
+    return (
+        <Pane title={viewSimulationMenu ? <SimulateLabel /> : "Chat"} actions={[
+            <ActionButton
+                key="new-chat"
+                icon={<MessageSquarePlusIcon size={16} />}
+                onClick={handleNewChatButtonClick}
+            >
+                New chat
+            </ActionButton>,
+            !viewSimulationMenu && <ActionButton
+                key="simulate"
+                icon={<PlayIcon size={16} />}
+                onClick={handleSimulateButtonClick}
+            >
+                Simulate
+            </ActionButton>,
+        ]}>
+            <div className="h-full overflow-auto">
+                {!viewSimulationMenu && loadingChat && <div className="flex justify-center items-center h-full">
+                    <Spinner />
+                </div>}
+                {!viewSimulationMenu && !loadingChat && <Chat
+                    key={existingChatId || 'chat-' + counter}
+                    chat={chat}
+                    initialChatId={existingChatId || null}
+                    projectId={projectId}
+                    workflow={workflow}
+                    messageSubscriber={messageSubscriber}
+                />}
+                {viewSimulationMenu && <SimulateScenarioOption beginSimulation={beginSimulation} projectId={projectId} />}
+            </div>
+        </Pane>
+    );
 }

apps/rowboat/app/projects/[projectId]/simulation/app.tsx

@@ -156,8 +156,10 @@ export default function SimulationApp() {
   };
 
   const runSingleScenario = (scenario: ScenarioType) => {
-    // Navigate to the workflow playground with the scenario
-    router.push(`/projects/${projectId}/workflow/playground?scenarioId=${scenario._id}`);
+    // Store scenario ID in localStorage instead of URL parameter
+    localStorage.setItem('pendingScenarioId', scenario._id);
+    // Navigate to the playground without query parameter
+    router.push(`/projects/${projectId}/workflow`);
     setMenuOpenScenarioId(null);
   };