use ListBox for size select

lama_cleaner/app/src/Editor.tsx

@@ -402,9 +402,10 @@ export default function Editor(props: EditorProps) {
         ].join(' ')}
       >
         <SizeSelector
-          value={sizeLimit}
+          value={sizeLimit!}
           onChange={onSizeLimitChange}
-          originalSize={`${original.naturalWidth}x${original.naturalHeight}`}
+          originalWidth={original.naturalWidth}
+          originalHeight={original.naturalHeight}
         />
         <Slider
           label={

lama_cleaner/app/src/components/SizeSelector.tsx

@@ -1,56 +1,88 @@
 import React from 'react'
-import { RadioGroup } from '@headlessui/react'
+import { Listbox } from '@headlessui/react'
+import { CheckIcon, SelectorIcon } from '@heroicons/react/solid'
 
-const sizes = [
+const sizes = ['1080', '2000', 'Original']
-  ['1080', '1080'],
-  ['2000', '2k'],
-  ['Original', 'Original'],
-]
 
 type SizeSelectorProps = {
-  value?: string
+  value: string
-  originalSize: string
+  originalWidth: number
+  originalHeight: number
   onChange: (value: string) => void
 }
 
 export default function SizeSelector(props: SizeSelectorProps) {
-  const { value, originalSize, onChange } = props
+  const { value, originalHeight, originalWidth, onChange } = props
+
+  const getSizeShowName = (size: string) => {
+    if (size === 'Original') {
+      return `${originalWidth}x${originalHeight}(${size})`
+    }
+    const length: number = parseInt(size, 10)
+    const longSide: number =
+      originalWidth > originalHeight ? originalWidth : originalHeight
+    const scale = length / longSide
+
+    if (originalWidth > originalHeight) {
+      const newHeight = Math.ceil(scale * originalHeight)
+      return `${size}x${newHeight}`
+    }
+    const newWidth = Math.ceil(scale * originalWidth)
+    return `${newWidth}x${size}`
+  }
 
   return (
-    <RadioGroup
+    <div className="w-52">
-      className="my-4 flex items-center space-x-2"
+      <Listbox value={value} onChange={onChange}>
-      value={value}
+        <div className="relative">
-      onChange={onChange}
+          <Listbox.Options
-    >
+            style={{ top: '-112px' }}
-      <RadioGroup.Label>Resize</RadioGroup.Label>
+            className="absolute mb-1 w-full overflow-auto text-base bg-black backdrop-blur backdrop-filter md:bg-opacity-10 rounded-md max-h-60 ring-opacity-50 focus:outline-none sm:text-sm"
-      {sizes.map(size => (
+          >
-        <RadioGroup.Option key={size[0]} value={size[0]}>
+            {sizes.map((size, _) => (
-          {({ checked }) => (
+              <Listbox.Option
-            <div
+                key={size}
-              className={[
+                className={({ active }) =>
-                checked ? 'bg-gray-200' : 'border-opacity-10',
+                  `${active ? 'bg-black bg-opacity-10' : 'text-gray-900'}
-                'border-3 px-2 py-2 rounded-md',
+                          cursor-default select-none relative py-2 pl-10 pr-4`
-              ].join(' ')}
+                }
-            >
+                value={size}
-              <div className="flex items-center space-x-4">
+              >
-                <div
+                {({ selected, active }) => (
-                  className={[
+                  <>
-                    'rounded-full w-5 h-5 border-4 ',
+                    <span
-                    checked
+                      className={`${
-                      ? 'border-primary bg-black'
+                        selected ? 'font-medium' : 'font-normal'
-                      : 'border-black border-opacity-10',
+                      } block truncate`}
-                  ].join(' ')}
+                    >
-                />
+                      {getSizeShowName(size)}
-                {size[0] === 'Original' ? (
+                    </span>
-                  <span>{`${size[1]}(${originalSize})`}</span>
+                    {selected ? (
-                ) : (
+                      <span
-                  <span>{size[1]}</span>
+                        className={`${
+                          active ? 'text-amber-600' : 'text-amber-600'
+                        }
+                                absolute inset-y-0 left-0 flex items-center pl-3`}
+                      >
+                        <CheckIcon className="w-5 h-5" aria-hidden="true" />
+                      </span>
+                    ) : null}
+                  </>
                 )}
-              </div>
+              </Listbox.Option>
-            </div>
+            ))}
-          )}
+          </Listbox.Options>
-        </RadioGroup.Option>
+          <Listbox.Button className="relative w-full inline-flex w-full px-4 py-2 text-sm font-medium bg-black rounded-md bg-opacity-10  focus:outline-none focus-visible:ring-2 focus-visible:ring-white focus-visible:ring-opacity-75">
-      ))}
+            <span className="block truncate">{getSizeShowName(value)}</span>
-    </RadioGroup>
+            <span className="absolute inset-y-0 right-0 flex items-center pr-2 pointer-events-none">
+              <SelectorIcon
+                className="w-5 h-5 text-gray-400"
+                aria-hidden="true"
+              />
+            </span>
+          </Listbox.Button>
+        </div>
+      </Listbox>
+    </div>
   )
 }