codelion commited on
Commit
56772bb
·
verified ·
1 Parent(s): af8afc1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -3
app.py CHANGED
@@ -161,6 +161,7 @@ def generate_item(user_input, ideas, generate_video=False, max_retries=3):
161
  number_of_videos=2,
162
  duration_seconds=8,
163
  negative_prompt="blurry, low quality, text, letters"
 
164
  )
165
  )
166
  # Wait for videos to generate
@@ -168,13 +169,18 @@ def generate_item(user_input, ideas, generate_video=False, max_retries=3):
168
  time.sleep(20)
169
  operation = client.operations.get(operation)
170
 
171
- # Check if operation succeeded and has a valid response
 
 
172
  if operation.response is None:
173
  raise ValueError("Video generation operation failed: No response")
174
- if not hasattr(operation.response, 'generated_videos'):
175
  raise ValueError("Video generation operation failed: No generated_videos in response")
176
 
177
- for n, video in enumerate(operation.response.generated_videos):
 
 
 
178
  fname = f'with_image_input{n}.mp4'
179
  print(f"Generated video: {fname}")
180
  client.files.download(file=video.video)
 
161
  number_of_videos=2,
162
  duration_seconds=8,
163
  negative_prompt="blurry, low quality, text, letters"
164
+ # Removed enhance_prompt=True since it's the default
165
  )
166
  )
167
  # Wait for videos to generate
 
169
  time.sleep(20)
170
  operation = client.operations.get(operation)
171
 
172
+ # Enhanced error handling for video generation response
173
+ if operation.error:
174
+ raise ValueError(f"Video generation operation failed with error: {operation.error.message}")
175
  if operation.response is None:
176
  raise ValueError("Video generation operation failed: No response")
177
+ if not hasattr(operation.response, 'generated_videos') or operation.response.generated_videos is None:
178
  raise ValueError("Video generation operation failed: No generated_videos in response")
179
 
180
+ # Iterate over generated videos
181
+ for n, video in enumerate(operation.response.generated_videos, start=1):
182
+ if video is None or not hasattr(video, 'video'):
183
+ raise ValueError(f"Video {n} is invalid or missing video data")
184
  fname = f'with_image_input{n}.mp4'
185
  print(f"Generated video: {fname}")
186
  client.files.download(file=video.video)