Update app.py

app.py

@@ -38,6 +38,12 @@ with st.sidebar:
     
     show_subitems = st.checkbox("Show Sub-items (e.g., 1.1, 2.1.3)", value=True)
     plain_text_mode = st.checkbox("Render as Plain Text (Preserve Bold Only)", value=False)
+    
+    num_columns = st.radio(
+        "Number of Columns",
+        options=[1, 2, 3, 4, 5, 6],
+        index=3
+    )
 
 pdfmetrics.registerFont(TTFont(selected_font_name, selected_font_path))
 
@@ -220,7 +226,7 @@ default_markdown = """# 🌟 Deities Guide: Mythology and Moral Lessons 🌟
 3. Saints/Prophets: Virtues (e.g., justice, prophecy).
 """
 
-def markdown_to_pdf_content(markdown_text, show_subitems=True, plain_text_mode=False, font_size=10):
+def markdown_to_pdf_content(markdown_text, show_subitems=True, plain_text_mode=False, font_size=10, num_columns=4):
     lines = markdown_text.strip().split('\n')
     pdf_content = []
     
@@ -287,22 +293,8 @@ def markdown_to_pdf_content(markdown_text, show_subitems=True, plain_text_mode=F
     total_chars = sum(len(item) if isinstance(item, str) else len(item[0]) + sum(len(sub) for sub in item[1]) for item in pdf_content)
     avg_line_length = total_chars / total_lines if total_lines > 0 else 1
     
-    page_height = A4[0] - 72
-    page_width = A4[1] * 2 - 72
-    line_height = font_size * 1.15
-    lines_per_column = page_height / line_height
-    
-    chars_per_line = avg_line_length * (font_size / 10)
-    lines_per_column_width = page_width / (chars_per_line * 0.1)
-    
-    num_columns = max(2, int(total_lines / lines_per_column))
-    num_columns = min(num_columns, int(lines_per_column_width))
-    if num_columns % 2 != 0:
-        num_columns += 1
-    num_columns = min(num_columns, 6)
-    
-    lines_per_column = total_lines / num_columns
     columns = [[] for _ in range(num_columns)]
+    lines_per_column = total_lines / num_columns
     current_line_count = 0
     current_column = 0
     
@@ -316,7 +308,7 @@ def markdown_to_pdf_content(markdown_text, show_subitems=True, plain_text_mode=F
     
     return columns, total_lines, num_columns
 
-def create_main_pdf(markdown_text, base_font_size=10, auto_size=False, show_subitems=True, plain_text_mode=False):
+def create_main_pdf(markdown_text, base_font_size=10, auto_size=False, show_subitems=True, plain_text_mode=False, num_columns=4):
     buffer = io.BytesIO()
     doc = SimpleDocTemplate(
         buffer, 
@@ -330,7 +322,7 @@ def create_main_pdf(markdown_text, base_font_size=10, auto_size=False, show_subi
     styles = getSampleStyleSheet()
     story = []
     spacer_height = 10
-    columns, total_lines, num_columns = markdown_to_pdf_content(markdown_text, show_subitems, plain_text_mode, base_font_size)
+    columns, total_lines, num_columns = markdown_to_pdf_content(markdown_text, show_subitems, plain_text_mode, base_font_size, num_columns)
     
     if auto_size:
         base_font_size = max(6, min(16, base_font_size * 300 / total_lines))
@@ -441,7 +433,7 @@ if 'markdown_content' not in st.session_state:
     st.session_state.markdown_content = default_markdown
 
 with st.spinner("Generating PDF..."):
-    pdf_bytes = create_main_pdf(st.session_state.markdown_content, base_font_size, auto_size, show_subitems, plain_text_mode)
+    pdf_bytes = create_main_pdf(st.session_state.markdown_content, base_font_size, auto_size, show_subitems, plain_text_mode, num_columns)
 
 with st.container():
     pdf_images = pdf_to_image(pdf_bytes)