Update app.py

app.py

@@ -757,12 +757,19 @@ class EnhancedVideoGenerator:
         return f"{hours:02d}:{minutes:02d}:{secs:02d},{msecs:03d}"
 
     def create_video(self, script: str, style: str, duration: int, output_path: str, selected_images: List[str]) -> str:
-        """Create video with selected images"""
+        """Create video with selected images and improved error handling"""
         try:
             # Progress bar
             progress_bar = st.progress(0)
             status_text = st.empty()
             
+            # Validate inputs
+            if not selected_images:
+                raise ValueError("No images selected. Please select at least one image.")
+            
+            if not script.strip():
+                raise ValueError("Script cannot be empty.")
+            
             # Generate voice-over (20%)
             status_text.text("Creating voice-over...")
             audio = self.generate_voice_over(script)
@@ -771,60 +778,87 @@ class EnhancedVideoGenerator:
             # Process selected images (40%)
             status_text.text("Processing images...")
             processed_images = []
+            
             for img_url in selected_images:
-                response = requests.get(img_url)
-                img = Image.open(BytesIO(response.content))
-                img = img.resize((1920, 1080), Image.Resampling.LANCZOS)
-                processed_images.append(np.array(img))
-            progress_bar.progress(40)
+                try:
+                    response = requests.get(img_url, timeout=10)
+                    response.raise_for_status()  # Raise exception for bad status codes
+                    img = Image.open(BytesIO(response.content))
+                    img = img.convert('RGB')  # Ensure RGB format
+                    img = img.resize((1920, 1080), Image.Resampling.LANCZOS)
+                    processed_images.append(np.array(img))
+                except Exception as e:
+                    print(f"Error processing image {img_url}: {e}")
+                    continue
+            
+            if not processed_images:
+                raise ValueError("Failed to process any of the selected images.")
                 
+            progress_bar.progress(40)
+            
             # Create frames with transitions
             fps = 30
             total_frames = int(duration * fps)
             frames = []
-                
+            
             status_text.text("Generating frames...")
             frames_per_image = total_frames // len(processed_images)
-                
+            
             for idx, img in enumerate(processed_images):
+                # Add main frames
                 for _ in range(frames_per_image):
                     frames.append(img)
-                        
-                    # Add transition frames
+                
+                # Add transition frames to next image
                 if idx < len(processed_images) - 1:
                     next_img = processed_images[idx + 1]
                     for alpha in np.linspace(0, 1, 15):
                         transition_frame = (1 - alpha) * img + alpha * next_img
                         frames.append(transition_frame.astype(np.uint8))
-                        
-                progress_bar.progress(70)
-                
-                # Create video clip
+            
+            progress_bar.progress(70)
+            
+            # Create video clip
             status_text.text("Compiling video...")
             video = ImageSequenceClip(frames, fps=fps)
+            
+            # Ensure audio duration matches video duration
+            audio_duration = audio.duration
+            video_duration = len(frames) / fps
+            
+            if audio_duration > video_duration:
+                audio = audio.subclip(0, video_duration)
+            elif audio_duration < video_duration:
+                video = video.subclip(0, audio_duration)
+            
             video = video.set_audio(audio)
-                
+            
             progress_bar.progress(90)
-                
-                # Write final video
-            status_text.text("Saving video...")
-            video.write_videofile(
-                output_path,
-                fps=fps,
-                codec='libx264',
-                audio_codec='aac',
-                threads=4,
-                preset='ultrafast'
-            )
-                
+            
+            # Write final video with error handling
+            try:
+                video.write_videofile(
+                    output_path,
+                    fps=fps,
+                    codec='libx264',
+                    audio_codec='aac',
+                    threads=4,
+                    preset='ultrafast'
+                )
+            except Exception as e:
+                raise RuntimeError(f"Failed to write video file: {str(e)}")
+            
             progress_bar.progress(100)
             status_text.text("Video generation complete!")
-                
+            
             return output_path
-                
+            
         except Exception as e:
-            self.logger.error(f"Video creation failed: {str(e)}")
+            error_msg = f"Video creation failed: {str(e)}"
+            self.logger.error(error_msg)
+            st.error(error_msg)
             raise
+
     
         
     def generate_visual_assets(self, script: str, style: str) -> List[Dict]: