[assistant_builder] - refactor: streamline PrevNextButtons component …

…and support tab updates (#9075) - Refactor the PrevNextButtons component to simplify the logic for navigating between screens - Extend PrevNextButtons functionality to update current tab when navigating between screens
dust-tt · Dec 3, 2024 · 9a179be · 9a179be
1 parent 49412f1
commit 9a179be
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 15 deletions.
diff --git a/front/components/assistant_builder/AssistantBuilder.tsx b/front/components/assistant_builder/AssistantBuilder.tsx
@@ -480,7 +480,11 @@ export default function AssistantBuilder({
                     assertNever(screen);
                 }
               })()}
-              <PrevNextButtons screen={screen} setScreen={setScreen} />
+              <PrevNextButtons
+                screen={screen}
+                setScreen={setScreen}
+                setCurrentTab={setCurrentTab}
+              />
             </div>
           }
           buttonsRightPanel={

diff --git a/front/components/assistant_builder/PrevNextButtons.tsx b/front/components/assistant_builder/PrevNextButtons.tsx
@@ -3,13 +3,17 @@ import React from "react";
 
 import type { BuilderScreen } from "@app/components/assistant_builder/types";
 
+interface PrevNextButtonsProps {
+  screen: BuilderScreen;
+  setScreen: (screen: BuilderScreen) => void;
+  setCurrentTab: (tab: string) => void;
+}
+
 export function PrevNextButtons({
   screen,
   setScreen,
-}: {
-  screen: BuilderScreen;
-  setScreen: (screen: BuilderScreen) => void;
-}) {
+  setCurrentTab,
+}: PrevNextButtonsProps) {
   return (
     <div className="flex py-6">
       {screen !== "instructions" && (
@@ -18,11 +22,9 @@ export function PrevNextButtons({
           size="md"
           variant="highlight"
           onClick={() => {
-            if (screen === "actions") {
-              setScreen("instructions");
-            } else if (screen === "naming") {
-              setScreen("actions");
-            }
+            const newScreen = screen === "actions" ? "instructions" : "actions";
+            setScreen(newScreen);
+            setCurrentTab(newScreen);
           }}
         />
       )}
@@ -33,11 +35,9 @@ export function PrevNextButtons({
           size="md"
           variant="highlight"
           onClick={() => {
-            if (screen === "instructions") {
-              setScreen("actions");
-            } else if (screen === "actions") {
-              setScreen("naming");
-            }
+            const newScreen = screen === "instructions" ? "actions" : "naming";
+            setScreen(newScreen);
+            setCurrentTab(newScreen);
           }}
         />
       )}