Update app.py

app.py

@@ -738,6 +738,10 @@ def generate_manim_video(python_code, format_type, quality_preset, animation_spe
         full_output = []
         output_file_path = None
         mp4_output_path = None  # Track MP4 output for GIF fallback
+        
+        # Animation tracking variables
+        total_animations = None
+        current_animation = 0
         total_frames = None
         current_frame = 0
         
@@ -749,29 +753,46 @@ def generate_manim_video(python_code, format_type, quality_preset, animation_spe
             full_output.append(line)
             log_placeholder.code("".join(full_output[-10:]))
             
-            # Try to extract total frames information
-            if "Render animations with total frames:" in line:
+            # Try to detect total animations
+            if "Rendering animation number" in line or "Processing animation" in line:
+                try:
+                    # Extract current animation number
+                    anim_match = re.search(r"(?:Rendering animation number|Processing animation) (\d+) (?:out of|/) (\d+)", line)
+                    if anim_match:
+                        current_animation = int(anim_match.group(1))
+                        total_animations = int(anim_match.group(2))
+                        logger.info(f"Animation progress: {current_animation}/{total_animations}")
+                        
+                        # Calculate progress based on animations
+                        animation_progress = current_animation / total_animations
+                        progress_bar.progress(animation_progress)
+                        status_placeholder.info(f"Rendering {scene_class}: Animation {current_animation}/{total_animations} ({int(animation_progress*100)}%)")
+                except Exception as e:
+                    logger.error(f"Error parsing animation progress: {str(e)}")
+            
+            # Try to extract total frames information as fallback
+            elif "Render animations with total frames:" in line and not total_animations:
                 try:
                     total_frames = int(line.split("Render animations with total frames:")[1].strip().split()[0])
                     logger.info(f"Total frames to render: {total_frames}")
-                except:
-                    pass
+                except Exception as e:
+                    logger.error(f"Error parsing total frames: {str(e)}")
             
-            # Update progress bar based on frame information or percentage
-            if "Rendering frame" in line and total_frames:
+            # Update progress bar based on frame information if animation count not available
+            elif "Rendering frame" in line and total_frames and not total_animations:
                 try:
                     # Extract current frame number
                     frame_match = re.search(r"Rendering frame (\d+)", line)
                     if frame_match:
                         current_frame = int(frame_match.group(1))
                         # Calculate progress as current frame / total frames
-                        progress = min(0.99, current_frame / total_frames)
-                        progress_bar.progress(progress)
+                        frame_progress = min(0.99, current_frame / total_frames)
+                        progress_bar.progress(frame_progress)
                         # Update status with frame information
-                        status_placeholder.info(f"Rendering {scene_class}: Frame {current_frame}/{total_frames} ({int(progress*100)}%)")
-                except:
-                    pass
-            elif "%" in line:
+                        status_placeholder.info(f"Rendering {scene_class}: Frame {current_frame}/{total_frames} ({int(frame_progress*100)}%)")
+                except Exception as e:
+                    logger.error(f"Error parsing frame progress: {str(e)}")
+            elif "%" in line and not total_animations and not total_frames:
                 try:
                     # Fallback to percentage if available
                     percent = float(line.split("%")[0].strip().split()[-1])
@@ -816,6 +837,8 @@ def generate_manim_video(python_code, format_type, quality_preset, animation_spe
         # IMPORTANT: Wait a moment for file system to catch up
         time.sleep(3)
         
+        # Rest of the function remains the same
+        
         # Special handling for GIF format - if Manim failed to generate a GIF but we have an MP4
         if format_type == "gif" and (not output_file_path or not os.path.exists(output_file_path)) and mp4_output_path and os.path.exists(mp4_output_path):
             status_placeholder.info("GIF generation via Manim failed. Trying FFmpeg conversion...")