Modified the code

Author: Karan-SF4772

Word-document/Insert-PageBreak-Before-Heading-Paragraphs/.NET/Insert-PageBreak-Before-Heading-Paragraphs/Program.cs

@@ -13,41 +13,45 @@ public static void Main(string[] args)
 
             if (entities == null)
             {
-                Console.WriteLine("No paragraphs with the style 'Heading 1' found.");
-                return;
+                Console.WriteLine("No paragraphs with the style 'Heading 1' found.");                
             }
-            foreach (Entity entity in entities)
+            else
             {
-                // Check if the entity is a paragraph
-                WParagraph paragraph = entity as WParagraph;
-                // Get the index of the current paragraph in its parent text body
-                int index = paragraph.OwnerTextBody.ChildEntities.IndexOf(paragraph);
-                // Continue only if there is a previous entity
-                if (index > 0)
+                foreach (Entity entity in entities)
                 {
-                    // Get the previous entity and cast it to a paragraph
-                    WParagraph previousParagraph = paragraph.OwnerTextBody.ChildEntities[index - 1] as WParagraph;
-                    bool hasPageBreak = false;
-                    // Check if the previous paragraph ends with a page break
-                    if (previousParagraph != null && previousParagraph.ChildEntities.Count > 0)
+                    // Check if the entity is a paragraph
+                    WParagraph paragraph = entity as WParagraph;
+                    // Get the index of the current paragraph in its parent text body
+                    int index = paragraph.OwnerTextBody.ChildEntities.IndexOf(paragraph);
+                    // Continue only if there is a previous entity
+                    if (index > 0)
                     {
-                        // Get the last item in the previous paragraph
-                        ParagraphItem lastItem = previousParagraph.ChildEntities[previousParagraph.ChildEntities.Count - 1] as ParagraphItem;
-                        // Enable the boolean if the last item is a page break
-                        if (lastItem != null && lastItem is Break && (lastItem as Break).BreakType == BreakType.PageBreak)
-                            hasPageBreak = true;
-                    }
-                    // If no page break is found, insert the page break before the current paragraph
-                    if (!hasPageBreak)
-                    {
-                        WParagraph newPara = new WParagraph(document);
-                        newPara.AppendBreak(BreakType.PageBreak);
-                        // Insert the new paragraph with page break at the correct position
-                        paragraph.OwnerTextBody.ChildEntities.Insert(index, newPara);
+                        // Get the previous entity and cast it to a paragraph
+                        WParagraph previousParagraph = paragraph.OwnerTextBody.ChildEntities[index - 1] as WParagraph;
+                        bool hasPageBreak = false;
+                        // Check if the previous paragraph ends with a page break
+                        if (previousParagraph != null && previousParagraph.ChildEntities.Count > 0)
+                        {
+                            // Get the last item in the previous paragraph
+                            ParagraphItem lastItem = previousParagraph.ChildEntities[previousParagraph.ChildEntities.Count - 1] as ParagraphItem;
+                            // Enable the boolean if the last item is a page break
+                            if (lastItem != null && lastItem is Break && (lastItem as Break).BreakType == BreakType.PageBreak)
+                                hasPageBreak = true;
+                        }
+                        // If no page break is found, insert the page break before the current paragraph
+                        if (!hasPageBreak)
+                        {
+                            WParagraph newPara = new WParagraph(document);
+                            newPara.AppendBreak(BreakType.PageBreak);
+                            // Insert the new paragraph with page break at the correct position
+                            paragraph.OwnerTextBody.ChildEntities.Insert(index, newPara);
+                        }
                     }
                 }
             }
+             // Save the Word document.   
             document.Save(Path.GetFullPath("../../../Output/Output.docx"));
+            // Close the Word document.
             document.Close();
         }
     }