whai/cherry-studio
1
1
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

feat: improved real-time painting generation support

Browse Source
This commit is contained in:
kangfenmao 2024-10-31 13:28:07 +08:00
parent f6aa0dc55a
commit 31fa10f185
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/renderer/src/pages/paintings/PaintingsPage.tsx
View File

@ -13,10 +13,13 @@ import { TEXT_TO_IMAGES_MODELS } from '@renderer/config/models'
import { useTheme } from '@renderer/context/ThemeProvider' import { useTheme } from '@renderer/context/ThemeProvider'
import { usePaintings } from '@renderer/hooks/usePaintings' import { usePaintings } from '@renderer/hooks/usePaintings'
import { useAllProviders } from '@renderer/hooks/useProvider' import { useAllProviders } from '@renderer/hooks/useProvider'
import { useRuntime } from '@renderer/hooks/useRuntime'
import AiProvider from '@renderer/providers/AiProvider' import AiProvider from '@renderer/providers/AiProvider'
import { getProviderByModel } from '@renderer/services/AssistantService' import { getProviderByModel } from '@renderer/services/AssistantService'
import FileManager from '@renderer/services/FileManager' import FileManager from '@renderer/services/FileManager'
import { useAppDispatch } from '@renderer/store'
import { DEFAULT_PAINTING } from '@renderer/store/paintings' import { DEFAULT_PAINTING } from '@renderer/store/paintings'
import { setGenerating } from '@renderer/store/runtime'
import { FileType, Painting } from '@renderer/types' import { FileType, Painting } from '@renderer/types'
import { getErrorMessage } from '@renderer/utils' import { getErrorMessage } from '@renderer/utils'
import { Button, Input, InputNumber, Radio, Select, Slider, Tooltip } from 'antd' import { Button, Input, InputNumber, Radio, Select, Slider, Tooltip } from 'antd'
@ -76,6 +79,8 @@ const PaintingsPage: FC = () => {
const [isLoading, setIsLoading] = useState(false) const [isLoading, setIsLoading] = useState(false)
const [abortController, setAbortController] = useState<AbortController | null>(null) const [abortController, setAbortController] = useState<AbortController | null>(null)
const dispatch = useAppDispatch()
const { generating } = useRuntime()
const modelOptions = TEXT_TO_IMAGES_MODELS.map((model) => ({ const modelOptions = TEXT_TO_IMAGES_MODELS.map((model) => ({
label: model.name, label: model.name,
@ -138,6 +143,7 @@ const PaintingsPage: FC = () => {
const controller = new AbortController() const controller = new AbortController()
setAbortController(controller) setAbortController(controller)
setIsLoading(true) setIsLoading(true)
dispatch(setGenerating(true))
const AI = new AiProvider(provider) const AI = new AiProvider(provider)
try { try {
@ -179,6 +185,7 @@ const PaintingsPage: FC = () => {
} }
} finally { } finally {
setIsLoading(false) setIsLoading(false)
dispatch(setGenerating(false))
setAbortController(null) setAbortController(null)
} }
} }
@ -219,6 +226,7 @@ const PaintingsPage: FC = () => {
} }
const onSelectPainting = (newPainting: Painting) => { const onSelectPainting = (newPainting: Painting) => {
if (generating) return
setPainting(newPainting) setPainting(newPainting)
setCurrentImageIndex(0) setCurrentImageIndex(0)
} }