Update app.py

app.py

@@ -249,18 +249,23 @@ def generate_item(user_input, ideas, generate_video=False, max_retries=3):
                     operation = client.operations.get(operation)
                 if operation.error:
                     raise ValueError(f"Video generation failed: {operation.error.message}")
-                if operation.response and len(operation.response.generated_videos) > 0:
-                    video = operation.response.generated_videos[0]
-                    video_data = client.files.download(file=video.video)
-                    video_bytes = video_data if isinstance(video_data, bytes) else BytesIO(video_data).getvalue()
-                    video_base64 = base64.b64encode(video_bytes).decode()
-                    yield {"stage": "finalizing", "message": random.choice(PROGRESS_STAGES["finalizing"]), "progress": 90}
-                    return {
-                        'text': text,
-                        'image_base64': img_str,
-                        'video_base64': video_base64,
-                        'ideas': ideas
-                    }
+                if operation.response and hasattr(operation.response, 'generated_videos') and operation.response.generated_videos:
+                    if len(operation.response.generated_videos) > 0:
+                        video = operation.response.generated_videos[0]
+                        if video and hasattr(video, 'video'):
+                            video_data = client.files.download(file=video.video)
+                            video_bytes = video_data if isinstance(video_data, bytes) else BytesIO(video_data).getvalue()
+                            video_base64 = base64.b64encode(video_bytes).decode()
+                            yield {"stage": "finalizing", "message": random.choice(PROGRESS_STAGES["finalizing"]), "progress": 90}
+                            return {
+                                'text': text,
+                                'image_base64': img_str,
+                                'video_base64': video_base64,
+                                'ideas': ideas
+                            }
+                    raise ValueError("No valid video generated")
+                else:
+                    raise ValueError("Video generation operation failed: No generated_videos in response")
             except Exception as e:
                 print(f"Error generating video: {e}")
                 yield {"stage": "generating_video", "message": random.choice(PROGRESS_STAGES["generating_video"]), "progress": 80}
@@ -278,18 +283,25 @@ def generate_item(user_input, ideas, generate_video=False, max_retries=3):
                     while not operation.done:
                         time.sleep(20)
                         operation = client.operations.get(operation)
-                    if operation.response and len(operation.response.generated_videos) > 0:
-                        video = operation.response.generated_videos[0]
-                        video_data = client.files.download(file=video.video)
-                        video_bytes = video_data if isinstance(video_data, bytes) else BytesIO(video_data).getvalue()
-                        video_base64 = base64.b64encode(video_bytes).decode()
-                        yield {"stage": "finalizing", "message": random.choice(PROGRESS_STAGES["finalizing"]), "progress": 90}
-                        return {
-                            'text': text,
-                            'image_base64': img_str,
-                            'video_base64': video_base64,
-                            'ideas': ideas
-                        }
+                    if operation.error:
+                        raise ValueError(f"Video generation failed: {operation.error.message}")
+                    if operation.response and hasattr(operation.response, 'generated_videos') and operation.response.generated_videos:
+                        if len(operation.response.generated_videos) > 0:
+                            video = operation.response.generated_videos[0]
+                            if video and hasattr(video, 'video'):
+                                video_data = client.files.download(file=video.video)
+                                video_bytes = video_data if isinstance(video_data, bytes) else BytesIO(video_data).getvalue()
+                                video_base64 = base64.b64encode(video_bytes).decode()
+                                yield {"stage": "finalizing", "message": random.choice(PROGRESS_STAGES["finalizing"]), "progress": 90}
+                                return {
+                                    'text': text,
+                                    'image_base64': img_str,
+                                    'video_base64': video_base64,
+                                    'ideas': ideas
+                                }
+                        raise ValueError("No valid video generated")
+                    else:
+                        raise ValueError("Video generation operation failed: No generated_videos in response")
                 except Exception as e:
                     print(f"Error generating text-to-video: {e}")
                     if total_attempts == max_total_attempts:
@@ -525,7 +537,7 @@ def generate_share_links(image_base64, video_base64, caption):
                 font-size: 14px;
                 font-weight: bold;
                 transition: background-color 0.3s;
-            " onmouseover="this стиле.backgroundColor='#00d9d1'" onmouseout="this.style.backgroundColor='#00f2ea'">Share on TikTok</a>
+            " onmouseover="this.style.backgroundColor='#00d9d1'" onmouseout="this.style.backgroundColor='#00f2ea'">Share on TikTok</a>
             <a href="https://www.instagram.com/?caption={caption}" target="_blank" style="
                 background-color: #e1306c;
                 color: white;
@@ -809,7 +821,7 @@ with gr.Blocks(
             label="Generate Video (may take longer)",
             value=False
         )
-        magic_button = gr.Button("✨ Generate Next Item", elem_classes="gr-button")
+        magic_button = gr.Button("✨ Create ✨", elem_classes="gr-button")
 
     feed_html = gr.HTML()
     share_html = gr.HTML(label="Share this item:")