diff --git a/src/components/AIVideoGenerator.tsx b/src/components/AIVideoGenerator.tsx
index 45a2fe2..45390a3 100644
--- a/src/components/AIVideoGenerator.tsx
+++ b/src/components/AIVideoGenerator.tsx
@@ -1,6 +1,6 @@
 import React, { useState } from 'react'
 import { Play } from 'lucide-react'
-import { useAIVideoJobs, useAIVideoProcessing, useAIVideoSettings } from '../stores/useAIVideoStore'
+import { useAIVideoStore, useAIVideoJobs, useAIVideoProcessing, useAIVideoSettings } from '../stores/useAIVideoStore'
 import AIVideoResultPreview from './AIVideoResultPreview'
 import VideoModeSelector from './ai-video/VideoModeSelector'
 import SingleVideoForm from './ai-video/SingleVideoForm'
@@ -36,10 +36,15 @@ const AIVideoGenerator: React.FC<AIVideoGeneratorProps> = ({ className = '' }) =
   })
   
   // Store hooks
-  const { jobs, isProcessing, clearCompletedJobs } = useAIVideoJobs()
-  const { generateSingleVideo, generateBatchVideos } = useAIVideoProcessing()
+  const jobs = useAIVideoJobs()
+  const isProcessing = useAIVideoProcessing()
   const { defaultDuration, defaultModelType } = useAIVideoSettings()
 
+  // Store actions
+  const generateSingleVideo = useAIVideoStore(state => state.generateSingleVideo)
+  const batchGenerateVideos = useAIVideoStore(state => state.batchGenerateVideos)
+  const clearCompletedJobs = useAIVideoStore(state => state.clearCompletedJobs)
+
   // Initialize settings
   React.useEffect(() => {
     setDuration(defaultDuration)
@@ -98,7 +103,7 @@ const AIVideoGenerator: React.FC<AIVideoGeneratorProps> = ({ className = '' }) =
           ? batchPrompts.filter(p => p.trim())
           : ['默认提示词']
 
-        await generateBatchVideos({
+        await batchGenerateVideos({
           image_folder: selectedFolder,
           prompts,
           output_folder: outputFolder,