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

View File

@@ -6,7 +6,9 @@ const Home = () => {
return (
<div className="min-h-screen bg-neutral-50 dark:bg-black">
<Editor />
{/* <div className="dark:bg-neutral-800">What is MEMEAIGEN?</div> */}
<div className="dark:bg-neutral-800">MEMEAIGEN HERO</div>
<div className="dark:bg-neutral-800">MEMEAIGEN features</div>
<div className="dark:bg-neutral-800">MEMEAIGEN FAQ</div>
<FlashMessages />
<AuthUser />
</div>

View File

@@ -118,6 +118,26 @@ const useMediaStore = create(
}
},
checkActiveJob: async () => {
try {
const response = await axiosInstance.post(route('api.user.get_active_job'));
return response.data;
} catch (error) {
console.error('Error checking active job:', error);
throw error;
}
},
getMemeHistory: async () => {
try {
const response = await axiosInstance.post(route('api.user.get_meme_history'));
return response.data;
} catch (error) {
console.error('Error getting meme history:', error);
throw error;
}
},
updateMemeResult: (result) => {
set({
currentCaption: result.generate.caption,