Update app.py

app.py

@@ -862,7 +862,7 @@ 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], 
-                    video_effects: dict = None) -> str:
+                    video_effects: dict = None, progress_callback: Callable[[float], None] = None) -> str:
         """Create video with selected images and effects"""
         try:
             # Initialize default effects if none provided
@@ -876,7 +876,9 @@ class EnhancedVideoGenerator:
             
             # Process images with error handling
             processed_images = []
-            for img_url in selected_images:
+            total_images = len(selected_images)
+            
+            for idx, img_url in enumerate(selected_images):
                 try:
                     response = requests.get(img_url, timeout=10)
                     response.raise_for_status()
@@ -884,6 +886,12 @@ class EnhancedVideoGenerator:
                     img = img.convert('RGB')
                     img = img.resize((1920, 1080), Image.LANCZOS)
                     processed_images.append(img)
+                    
+                    # Update progress (20% of total progress is for image processing)
+                    if progress_callback:
+                        progress = (idx + 1) / total_images * 20
+                        progress_callback(progress)
+                        
                 except Exception as e:
                     self.logger.error(f"Error processing image {img_url}: {e}")
                     continue
@@ -893,6 +901,8 @@ class EnhancedVideoGenerator:
     
             # Generate voice-over
             audio = self.generate_fallback_audio(script)
+            if progress_callback:
+                progress_callback(30)  # 30% progress after audio generation
     
             # Calculate frames
             fps = 30
@@ -927,9 +937,14 @@ class EnhancedVideoGenerator:
                     frame = self.apply_video_effects(img_array.copy(), current_effects)
                     frames.append(frame)
                     frame_count += 1
+                    
+                    # Update progress (30% to 70% is for frame generation)
+                    if progress_callback and frame_count % 30 == 0:
+                        progress = 30 + (frame_count / total_frames * 40)
+                        progress_callback(progress)
     
-                # Add transition to next image
-                if idx < len(processed_images) - 1:
+                # Add transition to next image if enabled
+                if idx < len(processed_images) - 1 and video_effects.get('transition_style') != 'None':
                     next_img_array = np.array(processed_images[idx + 1])
                     transition_frames = 15
                     
@@ -945,7 +960,9 @@ class EnhancedVideoGenerator:
     
             # Create video clip
             clip = ImageSequenceClip(frames, fps=fps)
-            
+            if progress_callback:
+                progress_callback(80)  # 80% progress after creating clip
+    
             # Adjust audio duration
             if audio.duration > clip.duration:
                 audio = audio.subclip(0, clip.duration)
@@ -954,6 +971,8 @@ class EnhancedVideoGenerator:
     
             # Combine video and audio
             final_clip = clip.set_audio(audio)
+            if progress_callback:
+                progress_callback(90)  # 90% progress after combining audio
     
             # Ensure output directory exists
             os.makedirs(os.path.dirname(output_path), exist_ok=True)
@@ -968,6 +987,9 @@ class EnhancedVideoGenerator:
                 verbose=False,
                 logger=None
             )
+            
+            if progress_callback:
+                progress_callback(100)  # 100% progress after writing file
     
             return output_path
     
@@ -985,9 +1007,9 @@ class EnhancedVideoGenerator:
                     audio.close()
             except Exception as e:
                 self.logger.error(f"Cleanup error: {e}")
-        
-            
             
+                
+                
     def generate_visual_assets(self, script: str, style: str) -> List[Dict]:
         """Generate relevant visual assets based on script content"""
         try: