Update app.py

app.py

@@ -327,75 +327,75 @@ class EnhancedVideoGenerator:
             self.logger.error(f"Voice-over generation failed: {str(e)}")
             return AudioFileClip(duration=len(script.split()) * 0.3)
 
-     def create_video(self, script: str, style: str, duration: int, output_path: str, selected_images: List[str]) -> str:
-            """Create video with selected images"""
-            try:
-                # Progress bar
-                progress_bar = st.progress(0)
-                status_text = st.empty()
-                
-                # Generate voice-over (20%)
-                status_text.text("Creating voice-over...")
-                audio = self.generate_voice_over(script)
-                progress_bar.progress(20)
+    def create_video(self, script: str, style: str, duration: int, output_path: str, selected_images: List[str]) -> str:
+        """Create video with selected images"""
+        try:
+            # Progress bar
+            progress_bar = st.progress(0)
+            status_text = st.empty()
             
-                # 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)
+            # Generate voice-over (20%)
+            status_text.text("Creating voice-over...")
+            audio = self.generate_voice_over(script)
+            progress_bar.progress(20)
+            
+            # 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)
                 
-                # Create frames with transitions
-                fps = 30
-                total_frames = int(duration * fps)
-                frames = []
+            # 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)
+            status_text.text("Generating frames...")
+            frames_per_image = total_frames // len(processed_images)
                 
-                for idx, img in enumerate(processed_images):
-                    for _ in range(frames_per_image):
-                        frames.append(img)
+            for idx, img in enumerate(processed_images):
+                for _ in range(frames_per_image):
+                    frames.append(img)
                         
                     # Add transition frames
-                    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))
-                
+                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
-                status_text.text("Compiling video...")
-                video = ImageSequenceClip(frames, fps=fps)
-                video = video.set_audio(audio)
+            status_text.text("Compiling video...")
+            video = ImageSequenceClip(frames, fps=fps)
+            video = video.set_audio(audio)
                 
-                progress_bar.progress(90)
+            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'
-                )
+            status_text.text("Saving video...")
+            video.write_videofile(
+                output_path,
+                fps=fps,
+                codec='libx264',
+                audio_codec='aac',
+                threads=4,
+                preset='ultrafast'
+            )
                 
-                progress_bar.progress(100)
-                status_text.text("Video generation complete!")
+            progress_bar.progress(100)
+            status_text.text("Video generation complete!")
                 
-                return output_path
+            return output_path
                 
-            except Exception as e:
-                self.logger.error(f"Video creation failed: {str(e)}")
-                raise
+        except Exception as e:
+            self.logger.error(f"Video creation failed: {str(e)}")
+            raise
     
         
     def generate_visual_assets(self, script: str, style: str) -> List[Dict]: