feat: generate random seed for image generation in usepaintings hook

This commit is contained in:
kangfenmao 2024-11-30 14:18:45 +08:00
parent 494d52ac85
commit 3717ff25bf
2 changed files with 10 additions and 4 deletions

View File

@ -8,6 +8,7 @@ import { uuid } from '@renderer/utils'
export function usePaintings() {
const paintings = useAppSelector((state) => state.paintings.paintings)
const dispatch = useAppDispatch()
const generateRandomSeed = () => Math.floor(Math.random() * 1000000).toString()
return {
paintings,
@ -20,7 +21,7 @@ export function usePaintings() {
negativePrompt: '',
imageSize: '1024x1024',
numImages: 1,
seed: '',
seed: generateRandomSeed(),
steps: 25,
guidanceScale: 4.5,
model: TEXT_TO_IMAGES_MODELS[0].id

View File

@ -1,4 +1,4 @@
import { PlusOutlined, QuestionCircleOutlined } from '@ant-design/icons'
import { PlusOutlined, QuestionCircleOutlined, RedoOutlined } from '@ant-design/icons'
import ImageSize1_1 from '@renderer/assets/images/paintings/image-size-1-1.svg'
import ImageSize1_2 from '@renderer/assets/images/paintings/image-size-1-2.svg'
import ImageSize3_2 from '@renderer/assets/images/paintings/image-size-3-2.svg'
@ -308,7 +308,12 @@ const PaintingsPage: FC = () => {
<Input
value={painting.seed}
onChange={(e) => updatePaintingState({ seed: e.target.value })}
suffix={<RefreshIcon onClick={() => updatePaintingState({ seed: '' })} />}
suffix={
<RedoOutlined
onClick={() => updatePaintingState({ seed: Math.floor(Math.random() * 1000000).toString() })}
style={{ cursor: 'pointer', color: 'var(--color-text-2)' }}
/>
}
/>
<SettingTitle style={{ marginBottom: 5, marginTop: 15 }}>
@ -502,7 +507,7 @@ const InfoIcon = styled(QuestionCircleOutlined)`
}
`
const RefreshIcon = styled.span`
const RefreshIcon = styled(RedoOutlined)`
cursor: pointer;
`