Fix error in generation dialog not always displayed

HugoGresse · Jul 24, 2024 · f19910b · f19910b
1 parent 8bcf0e9
commit f19910b
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/src/events/actions/sessions/generation/useSessionsGenerationGeneric.ts b/src/events/actions/sessions/generation/useSessionsGenerationGeneric.ts
@@ -56,6 +56,7 @@ export const useSessionsGenerationGeneric = <
         ): Promise<{
             success: boolean
             results: any[]
+            message?: string
         }> => {
             const sessionsCount = sessions.length
 
@@ -82,6 +83,9 @@ export const useSessionsGenerationGeneric = <
                 return {
                     success: false,
                     results: [],
+                    message:
+                        answer.message ||
+                        'Unknown error while generating the video/image, you may want to switch the ShortVid server',
                 }
             }
 

diff --git a/src/events/page/sessions/components/GenerateSessionsVideoDialog.tsx b/src/events/page/sessions/components/GenerateSessionsVideoDialog.tsx
@@ -66,7 +66,7 @@ export const GenerateSessionsVideoDialog = ({
                 ...shortVidSetting,
                 updateSession: updateDoc,
             })
-            .then(({ success, results }: GeneratedSessionVideoAnswer) => {
+            .then(({ success, results, message }: GeneratedSessionVideoAnswer) => {
                 if (success && results.length) {
                     const text = onSuccess ? 'Video generation done!' : 'Video generation done, sessions updated!'
                     if (onSuccess) {
@@ -76,7 +76,7 @@ export const GenerateSessionsVideoDialog = ({
                     onClose()
                 } else {
                     createNotification(
-                        'Error while generating the video/image, you may want to switch the ShortVid server',
+                        message || 'Error while generating the video/image, you may want to switch the ShortVid server',
                         { type: 'error' }
                     )
                 }
@@ -145,6 +145,9 @@ export const GenerateSessionsVideoDialog = ({
                 {generatingState.generationState === GenerationStates.ERROR && (
                     <Typography color="error">{generatingState.message}</Typography>
                 )}
+                {finalGeneration.generatingState.generationState === GenerationStates.ERROR && (
+                    <Typography color="error">{finalGeneration.generatingState.message}</Typography>
+                )}
 
                 {generatingState.generationState === GenerationStates.SUCCESS && (
                     <>