Update app.py

app.py

@@ -1,13 +1,10 @@
-import os
+
+
 import io
 import re
 import streamlit as st
-
-st.set_page_config(layout="wide", initial_sidebar_state="collapsed")
-
 from PIL import Image
 import fitz
-
 from reportlab.lib.pagesizes import A4
 from reportlab.platypus import SimpleDocTemplate, Paragraph, Spacer, Table, TableStyle
 from reportlab.lib.styles import getSampleStyleSheet, ParagraphStyle
@@ -15,58 +12,190 @@ from reportlab.lib import colors
 from reportlab.pdfbase import pdfmetrics
 from reportlab.pdfbase.ttfonts import TTFont
 
-available_fonts = {
-    "NotoEmoji Variable": "NotoEmoji-VariableFont_wght.ttf",
-    "NotoEmoji Bold": "NotoEmoji-Bold.ttf",
-    "NotoEmoji Light": "NotoEmoji-Light.ttf",
-    "NotoEmoji Medium": "NotoEmoji-Medium.ttf",
-    "NotoEmoji Regular": "NotoEmoji-Regular.ttf",
-    "NotoEmoji SemiBold": "NotoEmoji-SemiBold.ttf"
-}
-
-with st.sidebar:
-    selected_font_name = st.selectbox(
-        "Select NotoEmoji Font",
-        options=list(available_fonts.keys())
-    )
-    selected_font_path = available_fonts[selected_font_name]
-    
-    auto_size = st.checkbox("Auto-size text", value=True)
-    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)
-    
-    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))
-
-def apply_emoji_font(text, emoji_font):
-    emoji_pattern = re.compile(
-        r"([\U0001F300-\U0001F5FF"
-        r"\U0001F600-\U0001F64F"
-        r"\U0001F680-\U0001F6FF"
-        r"\U0001F700-\U0001F77F"
-        r"\U0001F780-\U0001F7FF"
-        r"\U0001F800-\U0001F8FF"
-        r"\U0001F900-\U0001F9FF"
-        r"\U0001FA00-\U0001FA6F"
-        r"\U0001FA70-\U0001FAFF"
-        r"\u2600-\u26FF"
-        r"\u2700-\u27BF]+)"
-    )
-    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)
+st.set_page_config(layout="wide", initial_sidebar_state="collapsed")
+
+def create_pdf_tab(default_markdown):
+    # Font setup
+    available_fonts = {
+        "NotoEmoji Variable": "NotoEmoji-VariableFont_wght.ttf",
+        "NotoEmoji Bold": "NotoEmoji-Bold.ttf",
+        "NotoEmoji Light": "NotoEmoji-Light.ttf",
+        "NotoEmoji Medium": "NotoEmoji-Medium.ttf",
+        "NotoEmoji Regular": "NotoEmoji-Regular.ttf",
+        "NotoEmoji SemiBold": "NotoEmoji-SemiBold.ttf"
+    }
+
+    # Sidebar configuration
+    with st.sidebar:
+        selected_font_name = st.selectbox("Select NotoEmoji Font", options=list(available_fonts.keys()))
+        selected_font_path = available_fonts[selected_font_name]
+        auto_size = st.checkbox("Auto-size text", value=True)
+        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.")
+        plain_text_mode = st.checkbox("Render as Plain Text (Preserve Bold Only)", value=False)
+        num_columns = st.selectbox("Number of Columns", options=[1, 2, 3, 4, 5, 6], index=0)  # Default to 1 column
+
+    # Register font
+    pdfmetrics.registerFont(TTFont(selected_font_name, selected_font_path))
+
+    # Emoji font application
+    def apply_emoji_font(text, emoji_font):
+        emoji_pattern = re.compile(
+            r"([\U0001F300-\U0001F5FF"
+            r"\U0001F600-\U0001F64F"
+            r"\U0001F680-\U0001F6FF"
+            r"\U0001F700-\U0001F77F"
+            r"\U0001F780-\U0001F7FF"
+            r"\U0001F800-\U0001F8FF"
+            r"\U0001F900-\U0001F9FF"
+            r"\U0001FA00-\U0001FA6F"
+            r"\U0001FA70-\U0001FAFF"
+            r"\u2600-\u26FF"
+            r"\u2700-\u27BF]+)"
+        )
+        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)
+
+    # Markdown to PDF content
+    def markdown_to_pdf_content(markdown_text, plain_text_mode, font_size, num_columns):
+        lines = markdown_text.strip().split('\n')
+        pdf_content = []
+
+        if plain_text_mode:
+            for line in lines:
+                line = line.strip()
+                if not line or line.startswith('# '):
+                    continue
+                bold_pattern = re.compile(r'\*\*(.*?)\*\*')
+                line = bold_pattern.sub(r'<b>\1</b>', line)
+                pdf_content.append(line)
+        else:
+            for line in lines:
+                line = line.strip()
+                if not line or line.startswith('# '):
+                    continue
+                if line.startswith('## ') or line.startswith('### '):
+                    text = line.replace('## ', '').replace('### ', '').strip()
+                    pdf_content.append(f"<b>{text}</b>")
+                else:
+                    pdf_content.append(line.strip())
+
+        total_lines = len(pdf_content)
+        columns = [[] for _ in range(num_columns)]
+        lines_per_column = total_lines / num_columns if num_columns > 0 else total_lines
+        current_line_count = 0
+        current_column = 0
+
+        for item in pdf_content:
+            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 += 1
+
+        return columns, total_lines
+
+    # Create PDF
+    def create_pdf(markdown_text, base_font_size, auto_size, plain_text_mode, num_columns):
+        buffer = io.BytesIO()
+        doc = SimpleDocTemplate(buffer, pagesize=A4, leftMargin=36, rightMargin=36, topMargin=36, bottomMargin=36)
+        styles = getSampleStyleSheet()
+        story = []
+        spacer_height = 10
+        columns, total_lines = markdown_to_pdf_content(markdown_text, plain_text_mode, base_font_size, num_columns)
+
+        if auto_size and total_lines > 0:
+            base_font_size = max(6, min(16, base_font_size * 300 / total_lines))
+
+        item_font_size = base_font_size
+        section_font_size = base_font_size * 1.1
+
+        section_style = ParagraphStyle(
+            'SectionStyle', parent=styles['Heading2'], fontName="Helvetica-Bold",
+            textColor=colors.darkblue, fontSize=section_font_size, leading=section_font_size * 1.2, spaceAfter=2
+        )
+        item_style = ParagraphStyle(
+            'ItemStyle', parent=styles['Normal'], fontName="Helvetica",
+            fontSize=item_font_size, leading=item_font_size * 1.15, spaceAfter=1
+        )
+
+        story.append(Spacer(1, spacer_height))
+        column_cells = [[] for _ in range(num_columns)]
+        for col_idx, column in enumerate(columns):
+            for item in column:
+                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))
+                else:
+                    column_cells[col_idx].append(Paragraph(apply_emoji_font(item, selected_font_name), item_style))
+
+        max_cells = max(len(cells) for cells in column_cells) if column_cells else 0
+        for cells in column_cells:
+            cells.extend([Paragraph("", item_style)] * (max_cells - len(cells)))
+
+        col_width = (A4[0] - 72) / num_columns if num_columns > 0 else A4[0] - 72
+        table_data = list(zip(*column_cells)) if column_cells else [[]]
+        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), (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), ('BOTTOMPADDING', (0, 0), (-1, -1), 1),
+        ]))
+
+        story.append(table)
+        doc.build(story)
+        buffer.seek(0)
+        return buffer.getvalue()
+
+    # PDF to image
+    def pdf_to_image(pdf_bytes):
+        try:
+            doc = fitz.open(stream=pdf_bytes, filetype="pdf")
+            images = []
+            for page in doc:
+                pix = page.get_pixmap(matrix=fitz.Matrix(2.0, 2.0))
+                img = Image.frombytes("RGB", [pix.width, pix.height], pix.samples)
+                images.append(img)
+            doc.close()
+            return images
+        except Exception as e:
+            st.error(f"Failed to render PDF preview: {e}")
+            return None
+
+    # Main logic
+    if 'markdown_content' not in st.session_state:
+        st.session_state.markdown_content = default_markdown
+
+    with st.spinner("Generating PDF..."):
+        pdf_bytes = create_pdf(st.session_state.markdown_content, base_font_size, auto_size, plain_text_mode, num_columns)
+
+    with st.container():
+        pdf_images = pdf_to_image(pdf_bytes)
+        if pdf_images:
+            for img in pdf_images:
+                st.image(img, 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.pdf", mime="application/pdf")
+
+    edited_markdown = st.text_area("Modify the markdown content below:", value=st.session_state.markdown_content, height=300)
+    if st.button("Update PDF"):
+        st.session_state.markdown_content = edited_markdown
+        st.rerun()
+
+    st.download_button(label="Save Markdown", data=st.session_state.markdown_content, file_name="deities_guide.md", mime="text/markdown")
+
+
+
+
 
 default_markdown = """# 🌟 Deities Guide: Mythology and Moral Lessons 🌟
 
@@ -226,243 +355,5 @@ 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, num_columns=4):
-    lines = markdown_text.strip().split('\n')
-    pdf_content = []
-    
-    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])
-    
-    total_lines = sum(1 + len(subs) if isinstance(item, list) else 1 for item in pdf_content)
-    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
-    
-    columns = [[] for _ in range(num_columns)]
-    lines_per_column = total_lines / num_columns
-    current_line_count = 0
-    current_column = 0
-    
-    for item in pdf_content:
-        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 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, num_columns=4):
-    buffer = io.BytesIO()
-    doc = SimpleDocTemplate(
-        buffer, 
-        pagesize=(A4[1] * 2, A4[0]),
-        leftMargin=36,
-        rightMargin=36,
-        topMargin=36,
-        bottomMargin=36
-    )
-    
-    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, num_columns)
-    
-    if auto_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.85
-    section_font_size = base_font_size * 1.1
-    title_font_size = min(20, base_font_size * 1.4)
-    
-    title_style = ParagraphStyle(
-        'Heading1',
-        parent=styles['Heading1'],
-        fontName="Helvetica-Bold",
-        textColor=colors.darkblue,
-        alignment=1,
-        fontSize=title_font_size,
-        leading=title_font_size * 1.2
-    )
-    
-    section_style = ParagraphStyle(
-        'SectionStyle',
-        parent=styles['Heading2'],
-        fontName="Helvetica-Bold",
-        textColor=colors.darkblue,
-        fontSize=section_font_size,
-        leading=section_font_size * 1.2,
-        spaceAfter=2
-    )
-    
-    item_style = ParagraphStyle(
-        'ItemStyle',
-        parent=styles['Normal'],
-        fontName="Helvetica",
-        fontSize=item_font_size,
-        leading=item_font_size * 1.15,
-        spaceAfter=1
-    )
-    
-    subitem_style = ParagraphStyle(
-        'SubItemStyle',
-        parent=styles['Normal'],
-        fontName="Helvetica",
-        fontSize=subitem_font_size,
-        leading=subitem_font_size * 1.15,
-        leftIndent=8,
-        spaceAfter=1
-    )
-    
-    story.append(Paragraph(apply_emoji_font("Deities Guide: Mythology and Moral Lessons", selected_font_name), title_style))
-    story.append(Spacer(1, spacer_height))
-    
-    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(cells) for cells in column_cells)
-    for cells in column_cells:
-        cells.extend([Paragraph("", item_style)] * (max_cells - len(cells)))
-    
-    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), (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),
-        ('BOTTOMPADDING', (0, 0), (-1, -1), 1),
-    ]))
-    
-    story.append(table)
-    doc.build(story)
-    buffer.seek(0)
-    return buffer.getvalue()
-
-def pdf_to_image(pdf_bytes):
-    try:
-        doc = fitz.open(stream=pdf_bytes, filetype="pdf")
-        images = []
-        for page in doc:
-            pix = page.get_pixmap(matrix=fitz.Matrix(2.0, 2.0))
-            img = Image.frombytes("RGB", [pix.width, pix.height], pix.samples)
-            images.append(img)
-        doc.close()
-        return images
-    except Exception as e:
-        st.error(f"Failed to render PDF preview: {e}")
-        return None
-
-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, num_columns)
-
-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"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_multi_column.pdf",
-    mime="application/pdf"
-)
-
-edited_markdown = st.text_area(
-    "Modify the markdown content below:",
-    value=st.session_state.markdown_content,
-    height=300
-)
-
-if st.button("Update PDF"):
-    st.session_state.markdown_content = edited_markdown
-    st.rerun()
-
-st.download_button(
-    label="Save Markdown",
-    data=st.session_state.markdown_content,
-    file_name="deities_guide.md",
-    mime="text/markdown"
-)
+
+create_pdf_tab(default_markdown)