Update app.py

app.py

@@ -28,9 +28,32 @@ def generate_video():
         raw_lines = prompts.splitlines(keepends=False)
         lines = []
         
-        for i in range(0, len(raw_lines), 5):
-            block = '\n'.join(raw_lines[i:i+5])
-            lines.append(block)
+        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(';'):
+                block = line  # Start block with heading
+                i += 1
+        
+                # Add next 3–5 paragraph lines (or until next heading)
+                paragraph_lines = []
+                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)
+        
+                lines.append(block)
+            else:
+                # If not a heading, just group normal 5 lines
+                block = '\n'.join(raw_lines[i:i+5])
+                lines.append(block)
+                i += 5
         print(lines)
         image_olst=[]
         for id in range(len(lines)):