From 5ec4b8e5320cf194ff63149a1d8cb44145bfca4a Mon Sep 17 00:00:00 2001
From: Faisal Amir <urmauur@gmail.com>
Date: Sat, 10 Feb 2024 00:51:16 +0700
Subject: [PATCH] feat: revamp ui dropdown list model option (#1977)

* feat: add modal troubleshooting guideline

* resolve inconsistent message hidden

* feat: revamp ui dropdown list model option

* display model id and copy button

* add function copy id model from dropdownlist

* add info concurrently send requests to one active local model
---
 uikit/src/select/styles.scss                 |   1 +
 web/containers/DropdownListSidebar/index.tsx | 170 ++++++++++++++++---
 web/screens/LocalServer/index.tsx            |  22 +++
 3 files changed, 167 insertions(+), 26 deletions(-)

diff --git a/uikit/src/select/styles.scss b/uikit/src/select/styles.scss
index 6f6cd5800b..90485723ab 100644
--- a/uikit/src/select/styles.scss
+++ b/uikit/src/select/styles.scss
@@ -21,6 +21,7 @@
 
   &-item {
     @apply hover:bg-secondary relative my-1 block w-full cursor-pointer select-none items-center rounded-sm px-4 py-2 text-sm data-[disabled]:pointer-events-none data-[disabled]:opacity-50;
+    @apply focus:outline-none focus-visible:outline-0;
   }
 
   &-trigger-viewport {
diff --git a/web/containers/DropdownListSidebar/index.tsx b/web/containers/DropdownListSidebar/index.tsx
index 140a1aba15..2679d6869a 100644
--- a/web/containers/DropdownListSidebar/index.tsx
+++ b/web/containers/DropdownListSidebar/index.tsx
@@ -14,7 +14,14 @@ import {
 
 import { atom, useAtom, useAtomValue, useSetAtom } from 'jotai'
 
-import { MonitorIcon } from 'lucide-react'
+import {
+  MonitorIcon,
+  LayoutGridIcon,
+  FoldersIcon,
+  GlobeIcon,
+  CheckIcon,
+  CopyIcon,
+} from 'lucide-react'
 
 import { twMerge } from 'tailwind-merge'
 
@@ -22,6 +29,7 @@ import { MainViewState } from '@/constants/screens'
 
 import { useActiveModel } from '@/hooks/useActiveModel'
 
+import { useClipboard } from '@/hooks/useClipboard'
 import { useMainViewState } from '@/hooks/useMainViewState'
 
 import useRecommendedModel from '@/hooks/useRecommendedModel'
@@ -42,6 +50,8 @@ import {
 
 export const selectedModelAtom = atom<Model | undefined>(undefined)
 
+const engineOptions = ['Local', 'Remote']
+
 // TODO: Move all of the unscoped logics outside of the component
 const DropdownListSidebar = ({
   strictedThread = true,
@@ -51,13 +61,24 @@ const DropdownListSidebar = ({
   const activeThread = useAtomValue(activeThreadAtom)
   const [selectedModel, setSelectedModel] = useAtom(selectedModelAtom)
   const setThreadModelParams = useSetAtom(setThreadModelParamsAtom)
-
+  const [isTabActive, setIsTabActive] = useState(0)
   const { stateModel } = useActiveModel()
   const [serverEnabled, setServerEnabled] = useAtom(serverEnabledAtom)
   const { setMainViewState } = useMainViewState()
   const [loader, setLoader] = useState(0)
   const { recommendedModel, downloadedModels } = useRecommendedModel()
   const { updateModelParameter } = useUpdateModelParameters()
+  const clipboard = useClipboard({ timeout: 1000 })
+  const [copyId, setCopyId] = useState('')
+
+  const localModel = downloadedModels.filter(
+    (model) => model.engine === InferenceEngine.nitro
+  )
+  const remoteModel = downloadedModels.filter(
+    (model) => model.engine === InferenceEngine.openai
+  )
+
+  const modelOptions = isTabActive === 0 ? localModel : remoteModel
 
   useEffect(() => {
     if (!activeThread) return
@@ -171,48 +192,145 @@ const DropdownListSidebar = ({
           </SelectTrigger>
           <SelectPortal>
             <SelectContent className="right-2  block w-full min-w-[450px] pr-0">
-              <div className="flex w-full items-center space-x-2 px-4 py-2">
-                <MonitorIcon size={20} className="text-muted-foreground" />
-                <span>Local</span>
+              <div className="relative px-2 py-2 dark:bg-secondary/50">
+                <ul className="inline-flex w-full space-x-2 rounded-lg bg-zinc-100 px-1 dark:bg-secondary">
+                  {engineOptions.map((name, i) => {
+                    return (
+                      <li
+                        className={twMerge(
+                          'relative my-1 flex w-full cursor-pointer items-center justify-center space-x-2 px-2 py-2',
+                          isTabActive === i &&
+                            'rounded-md bg-background dark:bg-white'
+                        )}
+                        key={i}
+                        onClick={() => setIsTabActive(i)}
+                      >
+                        {i === 0 ? (
+                          <MonitorIcon
+                            size={20}
+                            className="z-50 text-muted-foreground"
+                          />
+                        ) : (
+                          <GlobeIcon
+                            size={20}
+                            className="z-50 text-muted-foreground"
+                          />
+                        )}
+                        <span
+                          className={twMerge(
+                            'relative z-50 font-medium text-muted-foreground',
+                            isTabActive === i &&
+                              'font-bold text-foreground dark:text-black'
+                          )}
+                        >
+                          {name}
+                        </span>
+                      </li>
+                    )
+                  })}
+                </ul>
               </div>
+
               <div className="border-b border-border" />
               {downloadedModels.length === 0 ? (
                 <div className="px-4 py-2">
                   <p>{`Oops, you don't have a model yet.`}</p>
                 </div>
               ) : (
-                <SelectGroup>
-                  {downloadedModels.map((x, i) => (
-                    <SelectItem
-                      key={i}
-                      value={x.id}
-                      className={twMerge(
-                        x.id === selectedModel?.id && 'bg-secondary'
-                      )}
-                    >
-                      <div className="flex w-full justify-between">
-                        <span className="line-clamp-1 block">{x.name}</span>
-                        <div className="space-x-2">
-                          <span className="font-bold text-muted-foreground">
-                            {toGibibytes(x.metadata.size)}
-                          </span>
-                          {x.engine == InferenceEngine.nitro && (
-                            <ModelLabel size={x.metadata.size} />
+                <SelectGroup className="py-2">
+                  <>
+                    {modelOptions.map((x, i) => (
+                      <div
+                        key={i}
+                        className={twMerge(
+                          x.id === selectedModel?.id && 'bg-secondary',
+                          'hover:bg-secondary'
+                        )}
+                      >
+                        <SelectItem
+                          value={x.id}
+                          className={twMerge(
+                            x.id === selectedModel?.id && 'bg-secondary',
+                            'my-0 pb-8 pt-4'
+                          )}
+                        >
+                          <div className="relative flex w-full justify-between">
+                            {x.engine === InferenceEngine.openai && (
+                              <svg
+                                width="20"
+                                height="20"
+                                viewBox="0 0 20 20"
+                                fill="none"
+                                xmlns="http://www.w3.org/2000/svg"
+                                className="absolute top-1"
+                              >
+                                <path
+                                  d="M18.5681 8.18423C18.7917 7.51079 18.8691 6.79739 18.795 6.09168C18.7209 5.38596 18.497 4.70419 18.1384 4.0919C17.6067 3.16642 16.7948 2.43369 15.8199 1.99936C14.8449 1.56503 13.7572 1.45153 12.7135 1.67523C12.1206 1.0157 11.3646 0.523789 10.5214 0.248906C9.67823 -0.0259764 8.77756 -0.0741542 7.90986 0.109212C7.04216 0.292577 6.23798 0.701031 5.57809 1.29355C4.91821 1.88607 4.42584 2.64179 4.15046 3.48481C3.45518 3.62739 2.79834 3.91672 2.22384 4.33347C1.64933 4.75023 1.1704 5.28481 0.81904 5.90148C0.281569 6.82542 0.0518576 7.89634 0.163116 8.95943C0.274374 10.0225 0.720837 11.0227 1.43796 11.8153C1.21351 12.4884 1.13539 13.2017 1.20883 13.9074C1.28227 14.6132 1.50557 15.2951 1.86379 15.9076C2.39616 16.8334 3.20872 17.5663 4.18438 18.0006C5.16004 18.4349 6.24841 18.5483 7.29262 18.3243C7.76367 18.8548 8.34248 19.2786 8.99038 19.5676C9.63828 19.8566 10.3404 20.004 11.0498 20C12.1195 20.001 13.1618 19.662 14.0263 19.032C14.8909 18.4021 15.5329 17.5137 15.8596 16.4951C16.5548 16.3523 17.2116 16.0629 17.786 15.6461C18.3605 15.2294 18.8395 14.6949 19.191 14.0784C19.7222 13.1558 19.9479 12.0889 19.836 11.0303C19.7242 9.97163 19.2804 8.9754 18.5681 8.18423ZM11.0498 18.691C10.1737 18.6924 9.32512 18.3853 8.65279 17.8236L8.77104 17.7566L12.753 15.4581C12.8521 15.4 12.9343 15.3171 12.9917 15.2176C13.0491 15.118 13.0796 15.0053 13.0802 14.8904V9.27631L14.7635 10.2501C14.7719 10.2544 14.7791 10.2605 14.7846 10.268C14.7901 10.2755 14.7937 10.2843 14.7952 10.2935V14.9456C14.7931 15.9383 14.3978 16.8898 13.6959 17.5917C12.9939 18.2936 12.0425 18.6889 11.0498 18.691ZM2.99921 15.2531C2.55985 14.4945 2.4021 13.6052 2.55371 12.7417L2.67204 12.8127L6.65787 15.1112C6.7565 15.1691 6.86877 15.1996 6.98312 15.1996C7.09747 15.1996 7.20975 15.1691 7.30837 15.1112L12.1774 12.3041V14.2478C12.1769 14.2579 12.1742 14.2677 12.1694 14.2766C12.1646 14.2855 12.1579 14.2932 12.1497 14.2991L8.11654 16.6251C7.25581 17.121 6.2335 17.255 5.27405 16.9978C4.3146 16.7405 3.49644 16.1131 2.99921 15.2531ZM1.95054 6.57965C2.39294 5.81612 3.09123 5.23375 3.92179 4.93565V9.66665C3.92029 9.78094 3.94949 9.89355 4.00635 9.99271C4.06321 10.0919 4.14564 10.174 4.24504 10.2304L9.09037 13.0256L7.40696 13.9994C7.39785 14.0042 7.38769 14.0068 7.37737 14.0068C7.36706 14.0068 7.3569 14.0042 7.34779 13.9994L3.32254 11.6773C2.46343 11.1793 1.83666 10.3612 1.57951 9.40204C1.32236 8.44291 1.45577 7.42095 1.95054 6.55998V6.57965ZM15.7808 9.79281L10.9197 6.96998L12.5992 5.99998C12.6083 5.99514 12.6185 5.99261 12.6288 5.99261C12.6391 5.99261 12.6493 5.99514 12.6584 5.99998L16.6836 8.32606C17.2991 8.68119 17.8008 9.20407 18.1303 9.83365C18.4597 10.4632 18.6032 11.1735 18.5441 11.8816C18.485 12.5898 18.2257 13.2664 17.7964 13.8327C17.3672 14.3989 16.7857 14.8314 16.1199 15.0796V10.3486C16.1164 10.2345 16.0833 10.1232 16.0238 10.0258C15.9644 9.92833 15.8807 9.8481 15.7808 9.79281ZM17.4564 7.27356L17.338 7.20256L13.3601 4.8844C13.2609 4.82617 13.1479 4.79547 13.0329 4.79547C12.9178 4.79547 12.8049 4.82617 12.7056 4.8844L7.84071 7.6914V5.74781C7.83967 5.73793 7.84132 5.72795 7.84549 5.71893C7.84965 5.70991 7.85618 5.70218 7.86437 5.69656L11.8896 3.3744C12.5066 3.01899 13.2119 2.84659 13.9232 2.87736C14.6345 2.90813 15.3224 3.14079 15.9063 3.54813C16.4903 3.95548 16.9461 4.52066 17.2206 5.17759C17.4952 5.83452 17.577 6.55602 17.4565 7.25773L17.4564 7.27356ZM6.92196 10.7191L5.23862 9.74931C5.2302 9.74424 5.223 9.73738 5.21753 9.72921C5.21205 9.72105 5.20845 9.71178 5.20696 9.70206V5.06181C5.20788 4.34996 5.41144 3.65307 5.79383 3.05265C6.17622 2.45222 6.72164 1.97305 7.36632 1.67118C8.011 1.3693 8.7283 1.2572 9.43434 1.34796C10.1404 1.43873 10.806 1.72861 11.3534 2.18373L11.235 2.25081L7.25321 4.54915C7.1541 4.60727 7.07182 4.69017 7.01445 4.78971C6.95707 4.88925 6.92658 5.00201 6.92596 5.1169L6.92196 10.7191ZM7.83662 8.74798L10.005 7.49815L12.1774 8.74798V11.2475L10.0129 12.4972L7.84062 11.2475L7.83662 8.74798Z"
+                                  fill="#18181B"
+                                />
+                              </svg>
+                            )}
+                            <div
+                              className={twMerge(
+                                x.engine === InferenceEngine.openai && 'pl-8'
+                              )}
+                            >
+                              <span className="line-clamp-1 block">
+                                {x.name}
+                              </span>
+                              <div className="absolute right-0 top-2 space-x-2">
+                                <span className="font-bold text-muted-foreground">
+                                  {toGibibytes(x.metadata.size)}
+                                </span>
+                                {x.engine == InferenceEngine.nitro && (
+                                  <ModelLabel size={x.metadata.size} />
+                                )}
+                              </div>
+                            </div>
+                          </div>
+                        </SelectItem>
+                        <div
+                          className={twMerge(
+                            'absolute -mt-6 inline-flex items-center space-x-2 px-4 pb-2 text-muted-foreground',
+                            x.engine === InferenceEngine.openai && 'left-8'
+                          )}
+                        >
+                          <span className="text-xs">{x.id}</span>
+                          {clipboard.copied && copyId === x.id ? (
+                            <CheckIcon size={16} className="text-green-600" />
+                          ) : (
+                            <CopyIcon
+                              size={16}
+                              className="z-20 cursor-pointer"
+                              onClick={() => {
+                                clipboard.copy(x.id)
+                                setCopyId(x.id)
+                              }}
+                            />
                           )}
                         </div>
                       </div>
-                    </SelectItem>
-                  ))}
+                    ))}
+                  </>
                 </SelectGroup>
               )}
               <div className="border-b border-border" />
-              <div className="w-full px-4 py-2">
+              <div className="flex w-full space-x-2 px-4 py-2">
+                <Button
+                  block
+                  themes="secondary"
+                  onClick={() => setMainViewState(MainViewState.Settings)}
+                >
+                  <FoldersIcon size={20} className="mr-2" />
+                  <span>My Models</span>
+                </Button>
                 <Button
                   block
                   className="bg-blue-100 font-bold text-blue-600 hover:bg-blue-100 hover:text-blue-600"
                   onClick={() => setMainViewState(MainViewState.Hub)}
                 >
-                  Explore The Hub
+                  <LayoutGridIcon size={20} className="mr-2" />
+                  <span>Explore The Hub</span>
                 </Button>
               </div>
             </SelectContent>
diff --git a/web/screens/LocalServer/index.tsx b/web/screens/LocalServer/index.tsx
index 65b6c85637..88a9d86bae 100644
--- a/web/screens/LocalServer/index.tsx
+++ b/web/screens/LocalServer/index.tsx
@@ -373,6 +373,28 @@ const LocalServerScreen = () => {
         )}
       >
         <div className="px-4 pt-4">
+          <div className="mb-4 flex items-start space-x-2">
+            <svg
+              width="18"
+              height="18"
+              viewBox="0 0 18 18"
+              className="mt-1 flex-shrink-0"
+              fill="none"
+              xmlns="http://www.w3.org/2000/svg"
+            >
+              <path
+                fill-rule="evenodd"
+                clip-rule="evenodd"
+                d="M9.00033 17.3337C13.6027 17.3337 17.3337 13.6027 17.3337 9.00033C17.3337 4.39795 13.6027 0.666992 9.00033 0.666992C4.39795 0.666992 0.666992 4.39795 0.666992 9.00033C0.666992 10.9978 1.36978 12.8311 2.54157 14.2666L0.910703 15.9111C0.390085 16.436 0.758808 17.3337 1.49507 17.3337H9.00033ZM5.25033 7.33366C5.25033 6.87342 5.62342 6.50033 6.08366 6.50033H11.917C12.3772 6.50033 12.7503 6.87342 12.7503 7.33366C12.7503 7.7939 12.3772 8.16699 11.917 8.16699H6.08366C5.62342 8.16699 5.25033 7.7939 5.25033 7.33366ZM6.08366 9.83366C5.62342 9.83366 5.25033 10.2068 5.25033 10.667C5.25033 11.1272 5.62342 11.5003 6.08366 11.5003H8.58366C9.0439 11.5003 9.41699 11.1272 9.41699 10.667C9.41699 10.2068 9.0439 9.83366 8.58366 9.83366H6.08366Z"
+                fill="#2563EB"
+              />
+            </svg>
+
+            <p>
+              You can concurrently send requests to one active local model and
+              multiple remote models.
+            </p>
+          </div>
           <DropdownListSidebar strictedThread={false} />
           {loadModelError && (
             <div className="mt-3 flex space-x-2 text-xs">