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 && (
  <>