Update app.py

app.py

@@ -507,6 +507,38 @@ def create_pdf_with_images(source_pdf_bytes, output_pdf="ImageLinked.pdf"):
 md_files = [f for f in glob.glob("*.md") if os.path.basename(f) != "README.md"]
 md_options = [os.path.splitext(os.path.basename(f))[0] for f in md_files]
 
+# 🖥️ 15. Main PDF generation - wisdom: Spin up the PDF like a weaver at the loom, crafting beauty!
+with st.spinner("Generating PDF..."):
+    # Initialize defaults for PDF generation
+    selected_md = None
+    base_font_size = 8
+    num_columns = 3
+    add_space_before_numbered = True
+    headings_to_fonts = True
+    longest_line_words = 0
+    total_lines = 0
+    
+    # Calculate document stats if markdown exists
+    if 'markdown_content' in st.session_state and st.session_state.markdown_content.strip():
+        current_markdown = st.session_state.markdown_content
+        lines = current_markdown.strip().split('\n')
+        total_lines = len([line for line in lines if line.strip()])
+        for line in lines:
+            if line.strip():
+                word_count = len(line.split())
+                longest_line_words = max(longest_line_words, word_count)
+    
+    pdf_bytes = create_pdf(
+        st.session_state.get('markdown_content', ''),
+        base_font_size,
+        num_columns,
+        add_space_before_numbered,
+        headings_to_fonts,
+        doc_title=selected_md if selected_md else "Untitled",
+        longest_line_words=longest_line_words,
+        total_lines=total_lines
+    )
+
 with st.sidebar:
     # 📚 14.1 Markdown selector - wisdom: Offer choices like a librarian, guiding users to their story!
     st.markdown("### 📄 PDF Options")
@@ -660,8 +692,8 @@ with st.sidebar:
                 pdf_file = create_selflinking_pdf()
                 st.success(f"Generated {pdf_file}")
                 with open(pdf_file, "rb") as f:
-                    pdf_bytes = f.read()
-                images = pdf_to_image(pdf_bytes)
+                    self_linked_pdf_bytes = f.read()
+                images = pdf_to_image(self_linked_pdf_bytes)
                 if images:
                     st.subheader(f"Preview of {pdf_file}")
                     for i, img in enumerate(images):
@@ -691,19 +723,6 @@ with st.sidebar:
                     mime="application/pdf"
                 )
 
-# 🖥️ 15. Main PDF generation - wisdom: Spin up the PDF like a weaver at the loom, crafting beauty!
-with st.spinner("Generating PDF..."):
-    pdf_bytes = create_pdf(
-        st.session_state.markdown_content,
-        base_font_size,
-        num_columns,
-        add_space_before_numbered,
-        headings_to_fonts,
-        doc_title=selected_md if selected_md else "Untitled",
-        longest_line_words=longest_line_words,
-        total_lines=total_lines
-    )
-
 # 🖼️ 16. PDF preview - wisdom: Show the masterpiece before it’s framed, delighting the creator!
 with st.container():
     st.markdown("### 📊 PDF Preview")