Update app.py
Browse files
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 |
-
#
|
|
|
|
|
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 |
-
|
|
|
|
|
|
|
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)
|