Update app.py

app.py

@@ -35,6 +35,9 @@ with st.sidebar:
     base_font_size = st.slider("Base Font Size (points)", min_value=6, max_value=16, value=10, step=1)
     if auto_size:
         st.info("Font size will adjust between 6-16 points based on content length, starting from your base size.")
+    
+    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)
 
 pdfmetrics.registerFont(TTFont(selected_font_name, selected_font_path))
 
@@ -52,7 +55,12 @@ def apply_emoji_font(text, emoji_font):
         r"\u2600-\u26FF"
         r"\u2700-\u27BF]+)"
     )
-    return emoji_pattern.sub(r'<font face="{}">\1</font>'.format(emoji_font), text)
+    def replace_emoji(match):
+        emoji = match.group(1)
+        if len(emoji) > 1:
+            emoji = emoji[0]
+        return f'<font face="{emoji_font}">{emoji}</font>'
+    return emoji_pattern.sub(replace_emoji, text)
 
 default_markdown = """# 🌟 Deities Guide: Mythology and Moral Lessons 🌟
 
@@ -212,78 +220,93 @@ default_markdown = """# 🌟 Deities Guide: Mythology and Moral Lessons 🌟
 3. Saints/Prophets: Virtues (e.g., justice, prophecy).
 """
 
-def markdown_to_pdf_content(markdown_text):
+def markdown_to_pdf_content(markdown_text, show_subitems=True, plain_text_mode=False):
     lines = markdown_text.strip().split('\n')
     pdf_content = []
-    in_list = False
-    current_item = None
-    sub_items = []
     
-    for line in lines:
-        line = line.strip()
-        if not line:
-            continue
-            
-        if line.startswith('# '):
-            continue
-        elif line.startswith('## '):
-            if current_item and sub_items:
-                pdf_content.append([current_item, sub_items])
-                sub_items = []
-                current_item = None
-            section = line.replace('## ', '').strip()
-            pdf_content.append(f"<b>{section}</b>")
-            in_list = False
-        elif line.startswith('### '):
-            if current_item and sub_items:
-                pdf_content.append([current_item, sub_items])
-                sub_items = []
-                current_item = None
-            subsection = line.replace('### ', '').strip()
-            pdf_content.append(f"<b>{subsection}</b>")
-            in_list = False
-        elif re.match(r'^\d+\.', line):
-            if current_item and sub_items:
-                pdf_content.append([current_item, sub_items])
-                sub_items = []
-            current_item = line.strip()
-            in_list = True
-        elif re.match(r'^\d+\.\d+\.', line):
-            sub_items.append(line.strip())
-        elif line.startswith('*') and not in_list:
-            pdf_content.append(line.strip())
-        else:
-            if in_list:
-                sub_items.append(line.strip())
-            else:
+    if plain_text_mode:
+        for line in lines:
+            line = line.strip()
+            if not line:
+                continue
+            if line.startswith('# '):
+                continue
+            bold_pattern = re.compile(r'\*\*(.*?)\*\*')
+            line = bold_pattern.sub(r'<b>\1</b>', line)
+            pdf_content.append(line)
+    else:
+        in_list = False
+        current_item = None
+        sub_items = []
+        
+        for line in lines:
+            line = line.strip()
+            if not line:
+                continue
+                
+            if line.startswith('# '):
+                continue
+            elif line.startswith('## '):
+                if current_item and sub_items:
+                    pdf_content.append([current_item, sub_items])
+                    sub_items = []
+                    current_item = None
+                section = line.replace('## ', '').strip()
+                pdf_content.append(f"<b>{section}</b>")
+                in_list = False
+            elif line.startswith('### '):
+                if current_item and sub_items:
+                    pdf_content.append([current_item, sub_items])
+                    sub_items = []
+                    current_item = None
+                if show_subitems:
+                    subsection = line.replace('### ', '').strip()
+                    pdf_content.append(f"<b>{subsection}</b>")
+                in_list = False
+            elif re.match(r'^\d+\.', line):
+                if current_item and sub_items:
+                    pdf_content.append([current_item, sub_items])
+                    sub_items = []
+                current_item = line.strip()
+                in_list = True
+            elif re.match(r'^\d+\.\d+\.', line):
+                if show_subitems:
+                    sub_items.append(line.strip())
+            elif line.startswith('*') and not in_list:
                 pdf_content.append(line.strip())
+            else:
+                if in_list:
+                    sub_items.append(line.strip())
+                else:
+                    pdf_content.append(line.strip())
+        
+        if current_item and sub_items:
+            pdf_content.append([current_item, sub_items])
     
-    if current_item and sub_items:
-        pdf_content.append([current_item, sub_items])
-    
-    total_weight = 0
-    for item in pdf_content:
-        if isinstance(item, list):
-            total_weight += 1 + len(item[1])
-        else:
-            total_weight += 1
+    total_lines = sum(1 + len(subs) if isinstance(item, list) else 1 for item in pdf_content)
+    page_height = A4[0] - 72
+    lines_per_column = total_lines / 2
+    num_columns = max(2, int(total_lines / (page_height / (base_font_size * 1.15))))
+    num_columns = min(num_columns, 4)
+    if num_columns % 2 != 0:
+        num_columns += 1
     
-    target_weight = total_weight / 2
-    left_column = []
-    right_column = []
-    current_weight = 0
+    lines_per_column = total_lines / num_columns
+    columns = [[] for _ in range(num_columns)]
+    current_line_count = 0
+    current_column = 0
     
     for item in pdf_content:
-        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_weight += item_weight
-        else:
-            right_column.append(item)
+        item_lines = 1 + (len(item[1]) if isinstance(item, list) else 0)
+        if current_line_count >= lines_per_column and current_column < num_columns - 1:
+            current_column += 1
+            current_line_count = 0
+        columns[current_column].append(item)
+        current_line_count += item_lines
     
-    return left_column, right_column
+    return columns, total_lines, num_columns
 
-def create_main_pdf(markdown_text, base_font_size=10, auto_size=False):
+def create_main_pdf(markdown_text, base_font_size=10, auto_size=False, show_subitems=True, plain_text_mode=False):
     buffer = io.BytesIO()
     doc = SimpleDocTemplate(
         buffer, 
@@ -297,9 +320,8 @@ def create_main_pdf(markdown_text, base_font_size=10, auto_size=False):
     styles = getSampleStyleSheet()
     story = []
     spacer_height = 10
-    left_column, right_column = markdown_to_pdf_content(markdown_text)
+    columns, total_lines, num_columns = markdown_to_pdf_content(markdown_text, show_subitems, plain_text_mode)
     
-    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 * 300 / total_lines))
     
@@ -350,45 +372,36 @@ def create_main_pdf(markdown_text, base_font_size=10, auto_size=False):
     story.append(Paragraph(apply_emoji_font("Deities Guide: Mythology and Moral Lessons", selected_font_name), title_style))
     story.append(Spacer(1, spacer_height))
     
-    left_cells = []
-    for item in left_column:
-        if isinstance(item, str) and item.startswith('<b>'):
-            text = item.replace('<b>', '').replace('</b>', '')
-            left_cells.append(Paragraph(apply_emoji_font(text, selected_font_name), section_style))
-        elif isinstance(item, list):
-            main_item, sub_items = item
-            left_cells.append(Paragraph(apply_emoji_font(main_item, selected_font_name), item_style))
-            for sub_item in sub_items:
-                left_cells.append(Paragraph(apply_emoji_font(sub_item, selected_font_name), subitem_style))
-        else:
-            left_cells.append(Paragraph(apply_emoji_font(item, selected_font_name), item_style))
-    
-    right_cells = []
-    for item in right_column:
-        if isinstance(item, str) and item.startswith('<b>'):
-            text = item.replace('<b>', '').replace('</b>', '')
-            right_cells.append(Paragraph(apply_emoji_font(text, selected_font_name), section_style))
-        elif isinstance(item, list):
-            main_item, sub_items = item
-            right_cells.append(Paragraph(apply_emoji_font(main_item, selected_font_name), item_style))
-            for sub_item in sub_items:
-                right_cells.append(Paragraph(apply_emoji_font(sub_item, selected_font_name), subitem_style))
-        else:
-            right_cells.append(Paragraph(apply_emoji_font(item, selected_font_name), item_style))
+    column_cells = [[] for _ in range(num_columns)]
+    for col_idx, column in enumerate(columns):
+        for item in column:
+            if plain_text_mode:
+                column_cells[col_idx].append(Paragraph(apply_emoji_font(item, selected_font_name), item_style))
+            else:
+                if isinstance(item, str) and item.startswith('<b>'):
+                    text = item.replace('<b>', '').replace('</b>', '')
+                    column_cells[col_idx].append(Paragraph(apply_emoji_font(text, selected_font_name), section_style))
+                elif isinstance(item, list):
+                    main_item, sub_items = item
+                    column_cells[col_idx].append(Paragraph(apply_emoji_font(main_item, selected_font_name), item_style))
+                    for sub_item in sub_items:
+                        column_cells[col_idx].append(Paragraph(apply_emoji_font(sub_item, selected_font_name), subitem_style))
+                else:
+                    column_cells[col_idx].append(Paragraph(apply_emoji_font(item, selected_font_name), item_style))
     
-    max_cells = max(len(left_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)))
+    max_cells = max(len(cells) for cells in column_cells)
+    for cells in column_cells:
+        cells.extend([Paragraph("", item_style)] * (max_cells - len(cells)))
     
-    table_data = list(zip(left_cells, right_cells))
-    col_width = (A4[1] - 36)
-    table = Table(table_data, colWidths=[col_width, col_width], hAlign='CENTER')
+    col_width = (A4[1] * 2 - 72) / num_columns
+    table_data = list(zip(*column_cells))
+    table = Table(table_data, colWidths=[col_width] * num_columns, 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),
+        ('LINEAFTER', (0, 0), (num_columns-1, -1), 0.5, colors.grey),
         ('LEFTPADDING', (0, 0), (-1, -1), 2),
         ('RIGHTPADDING', (0, 0), (-1, -1), 2),
         ('TOPPADDING', (0, 0), (-1, -1), 1),
@@ -418,20 +431,20 @@ 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)
+    pdf_bytes = create_main_pdf(st.session_state.markdown_content, base_font_size, auto_size, show_subitems, plain_text_mode)
 
 with st.container():
     pdf_images = pdf_to_image(pdf_bytes)
     if pdf_images:
         for i, img in enumerate(pdf_images):
-            st.image(img, caption=f"Double-Page Spread", use_container_width=True)
+            st.image(img, caption=f"Multi-Column Spread", use_container_width=True)
     else:
         st.info("Download the PDF to view it locally.")
 
 st.download_button(
     label="Download PDF",
     data=pdf_bytes,
-    file_name="deities_guide_double_page.pdf",
+    file_name="deities_guide_multi_column.pdf",
     mime="application/pdf"
 )