app.py

import io
import re
import streamlit as st
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
from reportlab.lib import colors
from reportlab.pdfbase import pdfmetrics
from reportlab.pdfbase.ttfonts import TTFont

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=9, step=1)  # Default to 9
        if auto_size:
            st.info("Font size will adjust between 6-16 points to fit content on one page, 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=3)  # Default to 4 columns

    # 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):
        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)
        return pdf_content, total_lines

    # Create PDF
    def create_pdf(markdown_text, base_font_size, auto_size, plain_text_mode, num_columns):
        buffer = io.BytesIO()
        # Double A4 page: A4 width * 2 (landscape)
        page_width = A4[0] * 2
        page_height = A4[1]
        doc = SimpleDocTemplate(buffer, pagesize=(page_width, page_height), leftMargin=36, rightMargin=36, topMargin=36, bottomMargin=36)
        styles = getSampleStyleSheet()
        story = []
        spacer_height = 10
        pdf_content, total_lines = markdown_to_pdf_content(markdown_text, plain_text_mode)

        # Auto-size logic: reduce font size as content grows to fit one page
        if auto_size and total_lines > 0:
            # Target fitting content within page height (accounting for margins and spacing)
            target_lines_per_column = (page_height - 72 - spacer_height) / (base_font_size * 1.15)  # Approx lines per column
            total_target_lines = target_lines_per_column * num_columns
            if total_lines > total_target_lines:
                scale_factor = total_target_lines / total_lines
                base_font_size = max(6, min(16, base_font_size * scale_factor))
            else:
                base_font_size = min(16, base_font_size * (total_lines / total_target_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))
        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

        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 = (page_width - 72) / num_columns if num_columns > 0 else page_width - 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 🌟

## 1. 📜 Introduction
1. Purpose: Explore deities, spirits, saints, and beings with their stories and morals.  
2. Usage: Guide for learning and storytelling across traditions.  
3. Themes: Justice, faith, hubris, redemption, cosmic order.

## 2. 🛠️ Core Concepts of Divinity
1. Powers: Creation, omniscience, shapeshifting across entities.  
2. Life Cycle: Mortality, immortality, transitions (e.g., saints, avatars).  
3. Communication: Omens, visions, miracles from gods and spirits.

### 2.1. ⚡ Standard Abilities
1. Creation: Gods and spirits shape worlds (e.g., Allah, Vishnu).  
2. Influence: Saints, prophets intercede (e.g., Muhammad, Paul).  
3. Transformation: Angels, avatars shift forms (e.g., Gabriel, Krishna).  
4. Knowledge: Foresight or revelation (e.g., Holy Spirit, Brahma).  
5. Judgment: Divine authority (e.g., Yahweh, Yama).

### 2.2. ⏳ Mortality and Immortality
1. Gods: Eternal (e.g., Allah, Shiva).  
2. Spirits: Realm-bound (e.g., jinn, devas).  
3. Saints/Prophets: Mortal to divine (e.g., Moses, Rama).  
4. Beings: Limbo states (e.g., cherubim, rakshasas).  
5. Lessons: Faith, duty define transitions.

### 2.3. 🌠 Ascension and Signs
1. Paths: Birth, deeds, revelation (e.g., Jesus, Arjuna).  
2. Signs: Miracles, prophecies (e.g., Quran, Gita).  
3. Morals: Obedience, devotion shape destiny.

## 3. 🎲 Storytelling and Games
1. Portrayal: Gods, spirits, saints in narratives or RPGs.  
2. Dynamics: Clerics, imams, sadhus serve higher powers.  
3. Balance: Power vs. personality for depth.

### 3.1. 🎮 Dungeon Mastering Beings
1. Gods: Epic scope (e.g., Allah, Vishnu).  
2. Spirits: Local influence (e.g., jinn, apsaras).  
3. Saints: Moral anchors (e.g., St. Francis, Ali).

### 3.2. 🙏 Devotee Relationships
1. Clerics: Serve gods (e.g., Krishna’s priests).  
2. Mediums: Channel spirits (e.g., jinn whisperers).  
3. Faithful: Venerate saints/prophets (e.g., Fatima’s followers).

## 4. 🌐 Traditions and Legends
*(Top three figures—gods, spirits, saints, beings—with relationships and morals)*  
1. **American Indian** 🦅  
   1. Coyote, Raven, White Buffalo Woman: Trickster kin and wise mother.  
   2. Relation: Siblings and guide teach balance.  
   3. Lesson: Chaos breeds wisdom.  
2. **Arthurian** ⚔️  
   1. Merlin, Morgan le Fay, Arthur: Mentor, rival, son.  
   2. Relation: Family tests loyalty.  
   3. Lesson: Honor vs. betrayal.  
3. **Babylonian** 🏛️  
   1. Marduk, Tiamat, Ishtar: Son, mother, lover.  
   2. Relation: Kinship drives order.  
   3. Lesson: Power reshapes chaos.  
4. **Christian: Trinity** ✝️  
   1. God (Yahweh), Jesus, Holy Spirit: Father, Son, Spirit.  
   2. Relation: Divine family redeems.  
   3. Lesson: Faith restores grace.  
5. **Christian: Saints & Angels** 😇  
   1. St. Michael, Gabriel, Mary: Warrior, messenger, mother.  
   2. Relation: Heavenly kin serve God.  
   3. Lesson: Duty upholds divine will.  
6. **Celtic** 🍀  
   1. Lugh, Morrigan, Cernunnos: Son, mother, father.  
   2. Relation: Family governs cycles.  
   3. Lesson: Courage in fate.  
7. **Central American** 🌄  
   1. Quetzalcoatl, Tezcatlipoca, Huitzilopochtli: Brothers and war son.  
   2. Relation: Sibling rivalry creates.  
   3. Lesson: Sacrifice builds worlds.  
8. **Chinese** 🐉  
   1. Jade Emperor, Nuwa, Sun Wukong: Father, mother, rebel son.  
   2. Relation: Family enforces harmony.  
   3. Lesson: Duty curbs chaos.  
9. **Cthulhu** 🐙  
   1. Cthulhu, Nyarlathotep, Yog-Sothoth: Elder kin.  
   2. Relation: Cosmic trio overwhelms.  
   3. Lesson: Insignificance humbles.  
10. **Egyptian** ☥  
    1. Ra, Osiris, Isis: Father, son, mother.  
    2. Relation: Family ensures renewal.  
    3. Lesson: Justice prevails.  
11. **Finnish** ❄️  
    1. Väinämöinen, Louhi, Ukko: Son, mother, father.  
    2. Relation: Kinship tests wisdom.  
    3. Lesson: Perseverance wins.  
12. **Greek** 🏛️  
    1. Zeus, Hera, Athena: Father, mother, daughter.  
    2. Relation: Family rules with tension.  
    3. Lesson: Hubris meets wisdom.  
13. **Hindu: Trimurti** 🕉️  
    1. Brahma, Vishnu, Shiva: Creator, preserver, destroyer.  
    2. Relation: Divine trio cycles existence.  
    3. Lesson: Balance sustains life.  
14. **Hindu: Avatars & Devis** 🌺  
    1. Krishna, Rama, Durga: Sons and fierce mother.  
    2. Relation: Avatars and goddess protect dharma.  
    3. Lesson: Duty defeats evil.  
15. **Japanese** 🌸  
    1. Amaterasu, Susanoo, Tsukuyomi: Sister, brothers.  
    2. Relation: Siblings balance cosmos.  
    3. Lesson: Harmony vs. chaos.  
16. **Melnibonean** 🗡️  
    1. Arioch, Xiombarg, Elric: Lords and mortal son.  
    2. Relation: Pact binds chaos.  
    3. Lesson: Power corrupts.  
17. **Muslim: Divine & Messengers** ☪️  
    1. Allah, Muhammad, Gabriel: God, prophet, angel.  
    2. Relation: Messenger reveals divine will.  
    3. Lesson: Submission brings peace.  
18. **Muslim: Spirits & Kin** 👻  
    1. Jinn, Iblis, Khidr: Spirits and guide defy or aid.  
    2. Relation: Supernatural kin test faith.  
    3. Lesson: Obedience vs. rebellion.  
19. **Nehwon** 🏰  
    1. Death, Ningauble, Sheelba: Fateful trio.  
    2. Relation: Guides shape destiny.  
    3. Lesson: Cunning defies fate.  
20. **Nonhumans’** 🧝  
    1. Corellon, Moradin, Gruumsh: Elf, dwarf, orc fathers.  
    2. Relation: Rivals define purpose.  
    3. Lesson: Community endures.  
21. **Norse** ᚱ  
    1. Odin, Frigg, Loki: Father, mother, trickster son.  
    2. Relation: Family faces doom.  
    3. Lesson: Sacrifice costs.  
22. **Sumerian** 🗿  
    1. Enki, Inanna, Anu: Son, daughter, father.  
    2. Relation: Kin wield knowledge.  
    3. Lesson: Ambition shapes.

## 5. 📚 Appendices
1. Planes: Realms of gods, spirits, saints (e.g., Paradise, Svarga).  
2. Symbols: Rituals and artifacts of faith.  
3. Charts: Domains and duties for devotees.

### 5.1. 🌌 Planes of Existence
1. Heaven/Paradise: Christian/Muslim abode.  
2. Svarga: Hindu divine realm.  
3. Underworld: Spirits linger (e.g., Sheol, Naraka).

### 5.2. 🕍 Temple Trappings
1. Cross/Crescent: Christian/Muslim faith.  
2. Mandalas: Hindu devotion.  
3. Relics: Saints’ and prophets’ legacy.

### 5.3. 📊 Clerical Chart
1. Gods: Domains (e.g., creation, mercy).  
2. Spirits: Influence (e.g., guidance, mischief).  
3. Saints/Prophets: Virtues (e.g., justice, prophecy).
"""

create_pdf_tab(default_markdown)