diff --git a/src/pages/generate/index.tsx b/src/pages/generate/index.tsx
index 40198c7..6870d78 100644
--- a/src/pages/generate/index.tsx
+++ b/src/pages/generate/index.tsx
@@ -100,18 +100,18 @@ export default function Generate() {
 
   const renderSuccess = () => {
     // 检测是否为视频
-    const isVideo = result?.outputUrl && /\.(mp4|webm|ogg|mov|avi|mkv|flv)$/i.test(result.outputUrl)
+    const isVideo = result?.imageUrl && /\.(mp4|webm|ogg|mov|avi|mkv|flv)$/i.test(result.imageUrl)
     
     return (
       <View className='generate-success'>
         <View className='result-container'>
           <Text className='success-text'>处理完成！</Text>
-          {result?.outputUrl && (
+          {result?.imageUrl && (
             <View className='result-media'>
               {isVideo ? (
                 <Video
                   className='result-video'
-                  src={result.outputUrl}
+                  src={result.imageUrl}
                   autoplay
                   muted
                   loop
@@ -124,7 +124,7 @@ export default function Generate() {
               ) : (
                 <Image 
                   className='result-image'
-                  src={result.outputUrl} 
+                  src={result.imageUrl} 
                   mode='aspectFit' 
                 />
               )}