Update app.py

app.py

@@ -538,30 +538,47 @@ class GeneratedMathScene(Scene):
             
             # Split by # separator
             segments = content.split("#")
-            mobjects = []
+            all_lines = []  # Store all lines
+            current_line = []  # Current line being built
+            current_width = 0
             
             for segment in segments:
                 segment = segment.strip()
                 if not segment:
                     continue
                 
-                # Check if it's a LaTeX equation (starts with %)
+                # Create the mobject (Text or MathTex)
                 if segment.startswith("%"):
-                    # Remove % from start
                     latex_content = segment[1:]
                     mob = MathTex(latex_content, color=color, font_size=equation_size)
                 else:
-                    # Regular text
                     mob = Text(segment, color=color, font=font, font_size=text_size)
                 
-                mobjects.append(mob)
+                # Calculate what width would be if we add this segment
+                test_line = current_line + [mob]
+                test_group = VGroup(*test_line).arrange(RIGHT, buff=0.05)
+                
+                # Check if adding this segment exceeds wrap_width
+                if test_group.width > wrap_width and len(current_line) > 0:
+                    # Save current line and start new line
+                    line_group = VGroup(*current_line).arrange(RIGHT, buff=0.05)
+                    all_lines.append(line_group)
+                    current_line = [mob]  # Start new line with current segment
+                else:
+                    # Add to current line
+                    current_line.append(mob)
+            
+            # Add the last line
+            if current_line:
+                line_group = VGroup(*current_line).arrange(RIGHT, buff=0.05)
+                all_lines.append(line_group)
             
-            if not mobjects:
+            if not all_lines:
                 return VGroup()
             
-            # Arrange horizontally with minimal spacing
-            inline_group = VGroup(*mobjects).arrange(RIGHT, buff=0.05)
-            return inline_group
+            # Stack all lines vertically
+            final_group = VGroup(*all_lines).arrange(DOWN, aligned_edge=LEFT, buff=0.2)
+            return final_group
         
         def make_wrapped_paragraph(content, color, font, font_size, line_spacing=0.2):
             lines = []