Update app.py

app.py

@@ -32,29 +32,54 @@ def generate_video():
         i = 0
         while i < len(raw_lines):
             line = raw_lines[i].strip()
-            
+        
             # Check if current line is a heading
-            if line.endswith('?') or line.endswith(':'):
+            if line.strip().startswith("#") and (line.endswith('?') or line.endswith(':')):
                 block = line  # Start block with heading
                 i += 1
         
-                # Add next 3–5 paragraph lines (or until next heading)
+                # Accumulate body lines until next heading or 5+ lines
                 paragraph_lines = []
-                count = 0
-                while i < len(raw_lines) and not (raw_lines[i].strip().endswith('?') or raw_lines[i].strip().endswith(':')) and count < 4:
-                    paragraph_lines.append(raw_lines[i])
+                while i < len(raw_lines):
+                    next_line = raw_lines[i].strip()
+        
+                    # Stop if next line is a heading
+                    if next_line.strip().startswith("#") and (next_line.endswith('?') or next_line.endswith(':')):
+                        break
+        
+                    paragraph_lines.append(next_line)
                     i += 1
-                    count += 1
         
+                    # If we've gathered enough lines for a slide, break to next
+                    if len(paragraph_lines) >= 5:
+                        break
+        
+                # Combine heading + paragraph
                 if paragraph_lines:
                     block += '\n' + '\n'.join(paragraph_lines)
         
                 lines.append(block)
+        
             else:
-                # If not a heading, just group normal 5 lines
-                block = '\n'.join(raw_lines[i:i+4])
-                lines.append(block)
-                i += 4
+                # Group normal lines (not part of any heading)
+                block_lines = []
+                count = 0
+        
+                while i < len(raw_lines) and count < 5:
+                    next_line = raw_lines[i].strip()
+        
+                    # If this is a heading, break to handle it separately
+                    if next_line.strip().startswith("#") and (next_line.endswith('?') or next_line.endswith(':')):
+                        break
+        
+                    block_lines.append(next_line)
+                    i += 1
+                    count += 1
+        
+                if block_lines:
+                    lines.append('\n'.join(block_lines))
+        
+        # Print or use lines as slides
         
         image_olst=[]
         for id in range(len(lines)):