From 336e6e0c19ad991fa4fb43f3e94f47e473bc42af Mon Sep 17 00:00:00 2001
From: Mary Hipp <maryhipp@Marys-MacBook-Air.local>
Date: Fri, 11 Oct 2024 14:54:48 -0400
Subject: [PATCH] only show Add Model button if not adding models

---
 .../modelManagerV2/subpanels/ModelManager.tsx   | 17 +++++++++++------
 1 file changed, 11 insertions(+), 6 deletions(-)

diff --git a/invokeai/frontend/web/src/features/modelManagerV2/subpanels/ModelManager.tsx b/invokeai/frontend/web/src/features/modelManagerV2/subpanels/ModelManager.tsx
index a07cb8c10b..8133960c51 100644
--- a/invokeai/frontend/web/src/features/modelManagerV2/subpanels/ModelManager.tsx
+++ b/invokeai/frontend/web/src/features/modelManagerV2/subpanels/ModelManager.tsx
@@ -1,6 +1,6 @@
 import { Button, Flex, Heading } from '@invoke-ai/ui-library';
-import { useAppDispatch } from 'app/store/storeHooks';
-import { setSelectedModelKey } from 'features/modelManagerV2/store/modelManagerV2Slice';
+import { useAppDispatch, useAppSelector } from 'app/store/storeHooks';
+import { selectSelectedModelKey, setSelectedModelKey } from 'features/modelManagerV2/store/modelManagerV2Slice';
 import { memo, useCallback } from 'react';
 import { useTranslation } from 'react-i18next';
 import { PiPlusBold } from 'react-icons/pi';
@@ -14,14 +14,19 @@ export const ModelManager = memo(() => {
   const handleClickAddModel = useCallback(() => {
     dispatch(setSelectedModelKey(null));
   }, [dispatch]);
+  const selectedModelKey = useAppSelector(selectSelectedModelKey);
 
   return (
     <Flex flexDir="column" layerStyle="first" p={4} gap={4} borderRadius="base" w="50%" h="full">
       <Flex w="full" gap={4} justifyContent="space-between" alignItems="center">
-        <Heading fontSize="xl">{t('common.modelManager')}</Heading>
-        <Button size="sm" colorScheme="invokeYellow" leftIcon={<PiPlusBold />} onClick={handleClickAddModel}>
-          {t('modelManager.addModels')}
-        </Button>
+        <Heading fontSize="xl" py={1}>
+          {t('common.modelManager')}
+        </Heading>
+        {!!selectedModelKey && (
+          <Button size="sm" colorScheme="invokeYellow" leftIcon={<PiPlusBold />} onClick={handleClickAddModel}>
+            {t('modelManager.addModels')}
+          </Button>
+        )}
       </Flex>
       <Flex flexDir="column" layerStyle="second" p={4} gap={4} borderRadius="base" w="full" h="full">
         <ModelListNavigation />