This commit is contained in:
ct
2025-07-04 14:55:56 +08:00
parent b5ac848ba2
commit b3ffc261a3
8 changed files with 320 additions and 4 deletions

View File

@@ -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);