Update app.py

app.py

@@ -32,32 +32,26 @@ def generate_video():
         i = 0
         while i < len(raw_lines):
             line = raw_lines[i].strip()
-        
-            # Check if it's a heading
-            if (line.startswith('#') or line.startswith('**') or line[0].isnumeric())  and (line.endswith('?**') or line.endswith(':**') or line.endswith('?') or line.endswith(':')):
-                heading = line
+            
+            # Check if current line is a heading
+            if line.endswith('?') or line.endswith(';'):
+                block = line  # Start block with heading
                 i += 1
         
-                # Collect paragraph lines
+                # Add next 3–5 paragraph lines (or until next heading)
                 paragraph_lines = []
-                while i < len(raw_lines) and not (raw_lines[i].strip().endswith('?') and raw_lines[i].strip().endswith(':')):
-                    if raw_lines[i].strip():
-                        paragraph_lines.append(raw_lines[i].strip())
+                count = 0
+                while i < len(raw_lines) and not (raw_lines[i].strip().endswith('?') or raw_lines[i].strip().endswith(';')) and count < 5:
+                    paragraph_lines.append(raw_lines[i])
                     i += 1
+                    count += 1
+        
+                if paragraph_lines:
+                    block += '\n' + '\n'.join(paragraph_lines)
         
-                # Now chunk paragraph (e.g., 4 lines per slide)
-                chunk_size = 4
-                for idx, j in enumerate(range(0, len(paragraph_lines), chunk_size)):
-                    chunk = paragraph_lines[j:j+chunk_size]
-                    if idx == 0:
-                        # First chunk – with heading
-                        block = heading + '\n' + '\n'.join(chunk)
-                    else:
-                        # Following chunks – without heading
-                        block = '\n'.join(chunk)
-                    lines.append(block)
+                lines.append(block)
             else:
-                # If no heading, just group 5 lines normally
+                # If not a heading, just group normal 5 lines
                 block = '\n'.join(raw_lines[i:i+5])
                 lines.append(block)
                 i += 5