Update app.py

app.py

@@ -3,11 +3,10 @@ import io
 import re
 import streamlit as st
 
-# Must be the very first Streamlit command.
 st.set_page_config(layout="wide", initial_sidebar_state="collapsed")
 
 from PIL import Image
-import fitz  # PyMuPDF
+import fitz
 
 from reportlab.lib.pagesizes import A4
 from reportlab.platypus import SimpleDocTemplate, Paragraph, Spacer, Table, TableStyle
@@ -16,8 +15,6 @@ from reportlab.lib import colors
 from reportlab.pdfbase import pdfmetrics
 from reportlab.pdfbase.ttfonts import TTFont
 
-# ---------------------------------------------------------------
-# Define available NotoEmoji fonts (assumed in the base directory)
 available_fonts = {
     "NotoEmoji Variable": "NotoEmoji-VariableFont_wght.ttf",
     "NotoEmoji Bold": "NotoEmoji-Bold.ttf",
@@ -27,7 +24,6 @@ available_fonts = {
     "NotoEmoji SemiBold": "NotoEmoji-SemiBold.ttf"
 }
 
-# Sidebar: Font selection and size adjustment.
 with st.sidebar:
     selected_font_name = st.selectbox(
         "Select NotoEmoji Font",
@@ -40,11 +36,8 @@ with st.sidebar:
     if auto_size:
         st.info("Font size will adjust between 6-16 points based on content length, starting from your base size.")
 
-# Register the chosen emoji font with ReportLab.
 pdfmetrics.registerFont(TTFont(selected_font_name, selected_font_path))
 
-# ---------------------------------------------------------------
-# Helper function to wrap emoji characters with a font tag.
 def apply_emoji_font(text, emoji_font):
     emoji_pattern = re.compile(
         r"([\U0001F300-\U0001F5FF"
@@ -61,8 +54,6 @@ def apply_emoji_font(text, emoji_font):
     )
     return emoji_pattern.sub(r'<font face="{}">\1</font>'.format(emoji_font), text)
 
-# ---------------------------------------------------------------
-# Default markdown content with your Deities Guide.
 default_markdown = """# 🌟 Deities Guide: Mythology and Moral Lessons 🌟
 
 ## 1. 📜 Introduction
@@ -221,8 +212,6 @@ default_markdown = """# 🌟 Deities Guide: Mythology and Moral Lessons 🌟
 3. Saints/Prophets: Virtues (e.g., justice, prophecy).
 """
 
-# ---------------------------------------------------------------
-# Process markdown into a two-column layout for a double-page spread.
 def markdown_to_pdf_content(markdown_text):
     lines = markdown_text.strip().split('\n')
     pdf_content = []
@@ -236,7 +225,6 @@ def markdown_to_pdf_content(markdown_text):
             continue
             
         if line.startswith('# '):
-            # Skip the main title for now, added separately.
             pass
         elif line.startswith('## '):
             if current_item and sub_items:
@@ -267,31 +255,33 @@ def markdown_to_pdf_content(markdown_text):
     if current_item and sub_items:
         pdf_content.append([current_item, sub_items])
     
-    # Split into two columns (left and right) based on line count.
-    total_lines = sum(1 + len(subs) if isinstance(item, list) else 1 for item in pdf_content)
-    target_lines = total_lines // 2
+    total_weight = 0
+    for item in pdf_content:
+        if isinstance(item, list):
+            total_weight += 1 + len(item[1])
+        else:
+            total_weight += 1
+    
+    target_weight = total_weight / 2
     left_column = []
     right_column = []
-    current_lines = 0
+    current_weight = 0
     
     for item in pdf_content:
-        line_count = 1 + (len(item[1]) if isinstance(item, list) else 0)
-        if current_lines < target_lines or len(left_column) == 0:
+        item_weight = 1 + (len(item[1]) if isinstance(item, list) else 0)
+        if current_weight < target_weight or not left_column:
             left_column.append(item)
-            current_lines += line_count
+            current_weight += item_weight
         else:
             right_column.append(item)
     
     return left_column, right_column
 
-# ---------------------------------------------------------------
-# Create the PDF as a double-page spread.
 def create_main_pdf(markdown_text, base_font_size=10, auto_size=False):
     buffer = io.BytesIO()
-    # Double-page spread: Two A4 pages side-by-side (landscape).
     doc = SimpleDocTemplate(
         buffer, 
-        pagesize=(A4[1] * 2, A4[0]),  # Width = 2 * A4 width, Height = A4 height
+        pagesize=(A4[1] * 2, A4[0]),
         leftMargin=36,
         rightMargin=36,
         topMargin=36,
@@ -303,23 +293,23 @@ def create_main_pdf(markdown_text, base_font_size=10, auto_size=False):
     spacer_height = 10
     left_column, right_column = markdown_to_pdf_content(markdown_text)
     
-    total_items = sum(1 + len(subs) if isinstance(item, list) else 1 for item in left_column + right_column)
+    total_lines = sum(1 + len(subs) if isinstance(item, list) else 1 for item in left_column + right_column)
     if auto_size:
-        base_font_size = max(6, min(16, base_font_size * 200 / total_items))  # Scale based on user’s base size.
+        base_font_size = max(6, min(16, base_font_size * 300 / total_lines))
     
     item_font_size = base_font_size
-    subitem_font_size = base_font_size * 0.9
-    section_font_size = base_font_size * 1.2
-    title_font_size = min(20, base_font_size * 1.5)
+    subitem_font_size = base_font_size * 0.85
+    section_font_size = base_font_size * 1.1
+    title_font_size = min(20, base_font_size * 1.4)
     
-    # Define ParagraphStyles using Helvetica for normal text.
     title_style = ParagraphStyle(
         'Heading1',
         parent=styles['Heading1'],
         fontName="Helvetica-Bold",
         textColor=colors.darkblue,
         alignment=1,
-        fontSize=title_font_size
+        fontSize=title_font_size,
+        leading=title_font_size * 1.2
     )
     
     section_style = ParagraphStyle(
@@ -337,7 +327,7 @@ def create_main_pdf(markdown_text, base_font_size=10, auto_size=False):
         parent=styles['Normal'],
         fontName="Helvetica",
         fontSize=item_font_size,
-        leading=item_font_size * 1.2,
+        leading=item_font_size * 1.15,
         spaceAfter=1
     )
     
@@ -346,16 +336,14 @@ def create_main_pdf(markdown_text, base_font_size=10, auto_size=False):
         parent=styles['Normal'],
         fontName="Helvetica",
         fontSize=subitem_font_size,
-        leading=subitem_font_size * 1.2,
-        leftIndent=10,
+        leading=subitem_font_size * 1.15,
+        leftIndent=8,
         spaceAfter=1
     )
     
-    # Title spanning both columns.
     story.append(Paragraph(apply_emoji_font("Deities Guide: Mythology and Moral Lessons", selected_font_name), title_style))
     story.append(Spacer(1, spacer_height))
     
-    # Prepare two-column layout.
     left_cells = []
     for item in left_column:
         if isinstance(item, str) and item.startswith('<b>'):
@@ -382,21 +370,19 @@ def create_main_pdf(markdown_text, base_font_size=10, auto_size=False):
         else:
             right_cells.append(Paragraph(apply_emoji_font(item, selected_font_name), item_style))
     
-    # Balance the columns by padding with empty strings.
     max_cells = max(len(left_cells), len(right_cells))
-    left_cells.extend([""] * (max_cells - len(left_cells)))
-    right_cells.extend([""] * (max_cells - len(right_cells)))
+    left_cells.extend([Paragraph("", item_style)] * (max_cells - len(left_cells)))
+    right_cells.extend([Paragraph("", item_style)] * (max_cells - len(right_cells)))
     
-    # Create a table for the double-page spread.
     table_data = list(zip(left_cells, right_cells))
-    col_width = (A4[1] - 36)  # Half the spread width minus margins.
+    col_width = (A4[1] - 36)
     table = Table(table_data, colWidths=[col_width, col_width], hAlign='CENTER')
     table.setStyle(TableStyle([
         ('VALIGN', (0, 0), (-1, -1), 'TOP'),
         ('ALIGN', (0, 0), (-1, -1), 'LEFT'),
         ('BACKGROUND', (0, 0), (-1, -1), colors.white),
         ('GRID', (0, 0), (-1, -1), 0, colors.white),
-        ('LINEAFTER', (0, 0), (0, -1), 0.5, colors.grey),  # Divider between columns.
+        ('LINEAFTER', (0, 0), (0, -1), 0.5, colors.grey),
         ('LEFTPADDING', (0, 0), (-1, -1), 2),
         ('RIGHTPADDING', (0, 0), (-1, -1), 2),
         ('TOPPADDING', (0, 0), (-1, -1), 1),
@@ -408,8 +394,6 @@ def create_main_pdf(markdown_text, base_font_size=10, auto_size=False):
     buffer.seek(0)
     return buffer.getvalue()
 
-# ---------------------------------------------------------------
-# Convert PDF bytes to an image for preview using PyMuPDF.
 def pdf_to_image(pdf_bytes):
     try:
         doc = fitz.open(stream=pdf_bytes, filetype="pdf")
@@ -424,17 +408,12 @@ def pdf_to_image(pdf_bytes):
         st.error(f"Failed to render PDF preview: {e}")
         return None
 
-# ---------------------------------------------------------------
-# Persist markdown content in session state.
 if 'markdown_content' not in st.session_state:
     st.session_state.markdown_content = default_markdown
 
-# ---------------------------------------------------------------
-# Generate the PDF.
 with st.spinner("Generating PDF..."):
     pdf_bytes = create_main_pdf(st.session_state.markdown_content, base_font_size, auto_size)
 
-# Display PDF preview.
 with st.container():
     pdf_images = pdf_to_image(pdf_bytes)
     if pdf_images:
@@ -443,7 +422,6 @@ with st.container():
     else:
         st.info("Download the PDF to view it locally.")
 
-# PDF Download button.
 st.download_button(
     label="Download PDF",
     data=pdf_bytes,
@@ -451,19 +429,16 @@ st.download_button(
     mime="application/pdf"
 )
 
-# Markdown editor.
 edited_markdown = st.text_area(
     "Modify the markdown content below:",
     value=st.session_state.markdown_content,
     height=300
 )
 
-# Update PDF on button click.
 if st.button("Update PDF"):
     st.session_state.markdown_content = edited_markdown
     st.rerun()
 
-# Markdown Download button.
 st.download_button(
     label="Save Markdown",
     data=st.session_state.markdown_content,