From fe1a498da083562cc630d8921a167c1b76d5c22f Mon Sep 17 00:00:00 2001
From: Enrico Ros <enrico.ros@gmail.com>
Date: Sat, 25 May 2024 15:51:37 -0700
Subject: [PATCH] Beam: also enable method selection on AutoStart

---
 src/modules/beam/gather/BeamGatherPane.tsx | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/modules/beam/gather/BeamGatherPane.tsx b/src/modules/beam/gather/BeamGatherPane.tsx
index 2b82361fa..496e8175d 100644
--- a/src/modules/beam/gather/BeamGatherPane.tsx
+++ b/src/modules/beam/gather/BeamGatherPane.tsx
@@ -81,8 +81,9 @@ export function BeamGatherPane(props: {
     setCurrentGatherLlmId: state.setCurrentGatherLlmId,
   })));
   const gatherAutoStartAfterScatter = useModuleBeamStore(state => state.gatherAutoStartAfterScatter);
+  const disableUnlessAutoStart = !props.canGather && !gatherAutoStartAfterScatter;
   const [_, gatherLlmComponent/*, gatherLlmIcon*/] = useLLMSelect(
-    currentGatherLlmId, setCurrentGatherLlmId, props.isMobile ? '' : 'Merge Model', true, !props.canGather && !gatherAutoStartAfterScatter,
+    currentGatherLlmId, setCurrentGatherLlmId, props.isMobile ? '' : 'Merge Model', true, disableUnlessAutoStart,
   );
 
   // derived state
@@ -128,7 +129,7 @@ export function BeamGatherPane(props: {
         <ButtonGroup
           variant='outlined'
           size='md'
-          disabled={!props.canGather}
+          disabled={disableUnlessAutoStart}
           // sx={{ boxShadow: isNoFactorySelected ? 'xs' : undefined }}
         >
           {FUSION_FACTORIES.map(factory => {