| @@ -3,8 +3,9 @@ | ||
| 3 | 3 | import { useRouter } from '@draft/hooks/useRouter'; |
| 4 | 4 | import { pageState } from '@draft/state/factory'; |
| 5 | 5 | import { backArrow } from '@draft/svg/BackArrow'; |
| 6 | 6 | import { generateImage } from '@shared/api/DataApi'; |
| 7 | +import { useImageCreditsSync } from '@shared/hooks/useImageCreditsSync'; | |
| 7 | 8 | import { useImageGenerationStore } from '@shared/state/generate-images'; |
| 8 | 9 | import { |
| 9 | 10 | BaseControl, |
| 10 | 11 | __experimentalHeading as Heading, |
| @@ -34,8 +35,10 @@ | ||
| 34 | 35 | const abortController = useRef(null); |
| 35 | 36 | const noCredits = curCredits.remaining === 0; |
| 36 | 37 | const { imageDetails, setImageDetails } = usePageState(); |
| 37 | 38 | const { goBack } = useRouter(); |
| 39 | + | |
| 40 | + useImageCreditsSync(); | |
| 38 | 41 | |
| 39 | 42 | const clearImageResponse = () => setImageDetails({ src: '', id: undefined }); |
| 40 | 43 | const handleSubmit = async (event) => { |
| 41 | 44 | event.preventDefault(); |
