Update
This commit is contained in:
@@ -17,7 +17,7 @@ const EditorAISheet = () => {
|
||||
const [isOpen, setIsOpen] = useState(false);
|
||||
const [prompt, setPrompt] = useState('');
|
||||
const emitter = useMitt();
|
||||
const { generateMeme, isGeneratingMeme, keywords, isLoadingAIHints, fetchAIHints, checkMemeJobStatus, updateMemeResult, setGeneratingMeme } = useMediaStore();
|
||||
const { generateMeme, isGeneratingMeme, keywords, isLoadingAIHints, fetchAIHints, checkMemeJobStatus, updateMemeResult, setGeneratingMeme, checkActiveJob } = useMediaStore();
|
||||
|
||||
const pollingIntervalRef = useRef(null);
|
||||
const currentJobIdRef = useRef(null);
|
||||
@@ -38,6 +38,36 @@ const EditorAISheet = () => {
|
||||
};
|
||||
}, [emitter, fetchAIHints]);
|
||||
|
||||
// Check for active job on component mount
|
||||
useEffect(() => {
|
||||
if (auth.user) {
|
||||
checkForActiveJob();
|
||||
}
|
||||
}, [auth.user]);
|
||||
|
||||
const checkForActiveJob = async () => {
|
||||
try {
|
||||
const response = await checkActiveJob();
|
||||
if (response?.success?.data) {
|
||||
const { job_id, status, result } = response.success.data;
|
||||
|
||||
if (status === 'pending' || status === 'processing') {
|
||||
// Resume polling for active job
|
||||
setGeneratingMeme(true);
|
||||
currentJobIdRef.current = job_id;
|
||||
startPolling(job_id);
|
||||
} else if (status === 'completed' && result) {
|
||||
// Show completed result
|
||||
updateMemeResult(result);
|
||||
toast.success('Your previous meme generation completed!');
|
||||
}
|
||||
}
|
||||
} catch (error) {
|
||||
// No active job or error - continue normally
|
||||
console.log('No active job found or error checking:', error);
|
||||
}
|
||||
};
|
||||
|
||||
const handleOpenChange = (open) => {
|
||||
setIsOpen(open);
|
||||
|
||||
|
||||
Reference in New Issue
Block a user