import io
import re
import streamlit as st
import glob
import os
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):
    # Dynamically load all .ttf fonts from the current directory
    # Fonts are sourced from: https://fonts.google.com/download/next-steps
    # We use glob to find all .ttf files, making the font list dynamic instead of hardcoded.
    font_files = glob.glob("*.ttf")
    if not font_files:
        st.error("No .ttf font files found in the current directory. Please add some, e.g., NotoColorEmoji-Regular.ttf.")
        return
    available_fonts = {os.path.splitext(os.path.basename(f))[0]: f for f in font_files}

    # Sidebar configuration
    with st.sidebar:
        selected_font_name = st.selectbox("Select Font", options=list(available_fonts.keys()), index=0 if "NotoColorEmoji-Regular" in available_fonts else 0)
        selected_font_path = available_fonts[selected_font_name]
        base_font_size = st.slider("Font Size (points)", min_value=6, max_value=16, value=9, step=1)  # Default to 9
        plain_text_mode = st.checkbox("Render as Plain Text (Preserve Bold Only)", value=False)
        auto_bold_numbers = st.checkbox("Auto-Bold Numbered Lines", value=False)
        num_columns = st.selectbox("Number of Columns", options=[1, 2, 3, 4, 5, 6], index=3)  # Default to 4

        # Markdown editor and buttons
        if 'markdown_content' not in st.session_state:
            st.session_state.markdown_content = default_markdown
        
        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")

    # Register the selected font with ReportLab
    # Note: Fonts must be TrueType (.ttf) for ReportLab compatibility.
    pdfmetrics.registerFont(TTFont(selected_font_name, selected_font_path))

    # Emoji font application function
    # This function handles Unicode emojis in the text, applying the selected font to them.
    # We use a regex to match emoji ranges (Unicode blocks like Miscellaneous Symbols, Emoticons, etc.).
    # To avoid multi-character emoji issues (e.g., base char + variation selector U+FE0F), we limit to the first character.
    # Noto Color Emoji (NotoColorEmoji-Regular.ttf) is recommended as it supports a full range of modern emojis,
    # including color rendering, though ReportLab renders in black-and-white by default.
    def apply_emoji_font(text, emoji_font):
        emoji_pattern = re.compile(
            r"([\U0001F300-\U0001F5FF"  # Miscellaneous Symbols and Pictographs
            r"\U0001F600-\U0001F64F"  # Emoticons
            r"\U0001F680-\U0001F6FF"  # Transport and Map Symbols
            r"\U0001F700-\U0001F77F"  # Alchemical Symbols
            r"\U0001F780-\U0001F7FF"  # Geometric Shapes Extended
            r"\U0001F800-\U0001F8FF"  # Supplemental Arrows-C
            r"\U0001F900-\U0001F9FF"  # Supplemental Symbols and Pictographs
            r"\U0001FA00-\U0001FA6F"  # Chess Symbols
            r"\U0001FA70-\U0001FAFF"  # Symbols and Pictographs Extended-A
            r"\u2600-\u26FF"          # Miscellaneous Symbols (e.g., ⚡ U+26A1)
            r"\u2700-\u27BF]+"        # Dingbats (e.g., ✝ U+271D)
            r")"
        )
        def replace_emoji(match):
            emoji = match.group(1)
            # Limit to first character to avoid rendering issues with multi-codepoint emojis
            # Example: ✝️ (U+271D U+FE0F) becomes just ✝ (U+271D), dropping the variation selector
            if len(emoji) > 1:
                emoji = emoji[0]
            return f'<font face="{emoji_font}">{emoji}</font>'
        return emoji_pattern.sub(replace_emoji, text)

    # Convert markdown to PDF content
    # This function processes markdown lines, filtering out headers and applying bolding rules.
    # If auto_bold_numbers is True, lines starting with "number. " (e.g., "1. ") are bolded.
    def markdown_to_pdf_content(markdown_text, plain_text_mode, auto_bold_numbers):
        lines = markdown_text.strip().split('\n')
        pdf_content = []
        number_pattern = re.compile(r'^\d+\.\s')

        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>")
                elif auto_bold_numbers and number_pattern.match(line):
                    pdf_content.append(f"<b>{line}</b>")
                else:
                    pdf_content.append(line.strip())

        total_lines = len(pdf_content)
        return pdf_content, total_lines

    # Create PDF
    # This function builds a PDF using ReportLab, arranging content in columns.
    # We use a double A4 landscape layout (A4 width * 2) for wide content.
    # Spacers are added before numbered sections for visual separation.
    # Paragraph styles define font sizes and bolding, with emojis rendered via the selected font.
    def create_pdf(markdown_text, base_font_size, plain_text_mode, num_columns, auto_bold_numbers):
        buffer = io.BytesIO()
        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
        section_spacer_height = 15
        pdf_content, total_lines = markdown_to_pdf_content(markdown_text, plain_text_mode, auto_bold_numbers)

        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

        number_pattern = re.compile(r'^\d+\.\s')
        for i, item in enumerate(pdf_content):
            if i > 0 and number_pattern.match(item.replace('<b>', '').replace('</b>', '')):
                columns[current_column].append(Spacer(1, section_spacer_height))
            
            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, Spacer):
                    column_cells[col_idx].append(item)
                elif 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()

    # Convert PDF to image for preview
    # Uses PyMuPDF (fitz) to render PDF pages as images for Streamlit display.
    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
    with st.spinner("Generating PDF..."):
        pdf_bytes = create_pdf(st.session_state.markdown_content, base_font_size, plain_text_mode, num_columns, auto_bold_numbers)

    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.")

    with st.sidebar:
        st.download_button(label="Download PDF", data=pdf_bytes, file_name="deities_guide.pdf", mime="application/pdf")

default_markdown = """# Deities Guide: Mythology and Moral Lessons 🌟

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

2. 🛠️ Core Concepts of Divinity
   - Powers: Creation, omniscience, shapeshifting across entities.
   - Life Cycle: Mortality, immortality, transitions such as saints and avatars.
   - Communication: Omens, visions, miracles from gods and spirits.

3. ⚡ Standard Abilities
   - Creation: Gods and spirits shape worlds, such as Allah and Vishnu.
   - Influence: Saints and prophets intercede, for example, Muhammad and Paul.
   - Transformation: Angels and avatars shift forms, like Gabriel and Krishna.
   - Knowledge: Foresight or revelation, as seen with the Holy Spirit and Brahma.
   - Judgment: Divine authority, exemplified by Yahweh and Yama.

4. ⏳ Mortality and Immortality
   - Gods: Eternal, such as Allah and Shiva.
   - Spirits: Realm-bound, like jinn and devas.
   - Saints/Prophets: Mortal to divine, for instance, Moses and Rama.
   - Beings: Limbo states, such as cherubim and rakshasas.
   - Lessons: Faith and duty define transitions.

5. 🌠 Ascension and Signs
   - Paths: Birth, deeds, revelation, as with Jesus and Arjuna.
   - Signs: Miracles and prophecies, like those in the Quran and Gita Kobe.
   - Morals: Obedience and devotion shape destiny.

6. 🎲 Storytelling and Games
   - Portrayal: Gods, spirits, and saints in narratives or RPGs.
   - Dynamics: Clerics, imams, and sadhus serve higher powers.
   - Balance: Power versus personality for depth.

7. 🎮 Dungeon Mastering Beings
   - Gods: Epic scope, such as Allah and Vishnu.
   - Spirits: Local influence, like jinn and apsaras.
   - Saints: Moral anchors, for example, St. Francis and Ali.

8. 🙏 Devotee Relationships
   - Clerics: Serve gods, such as Krishna’s priests.
   - Mediums: Channel spirits, like jinn whisperers.
   - Faithful: Venerate saints and prophets, for instance, Fatima’s followers.

9. 🦅 American Indian Traditions
   - Coyote, Raven, White Buffalo Woman: Trickster kin and wise mother.
   - Relation: Siblings and guide teach balance.
   - Lesson: Chaos breeds wisdom.

10. ⚔️ Arthurian Legends
    - Merlin, Morgan le Fay, Arthur: Mentor, rival, son.
    - Relation: Family tests loyalty.
    - Lesson: Honor versus betrayal.

11. 🏛️ Babylonian Mythology
    - Marduk, Tiamat, Ishtar: Son, mother, lover.
    - Relation: Kinship drives order.
    - Lesson: Power reshapes chaos.

12. ✝️ Christian Trinity
    - God (Yahweh), Jesus, Holy Spirit: Father, Son, Spirit.
    - Relation: Divine family redeems.
    - Lesson: Faith restores grace.

13. 😇 Christian Saints & Angels
    - St. Michael, Gabriel, Mary: Warrior, messenger, mother.
    - Relation: Heavenly kin serve God.
    - Lesson: Duty upholds divine will.

14. 🍀 Celtic Mythology
    - Lugh, Morrigan, Cernunnos: Son, mother, father.
    - Relation: Family governs cycles.
    - Lesson: Courage in fate.

15. 🌄 Central American Traditions
    - Quetzalcoatl, Tezcatlipoca, Huitzilopochtli: Brothers and war son.
    - Relation: Sibling rivalry creates.
    - Lesson: Sacrifice builds worlds.

16. 🐉 Chinese Mythology
    - Jade Emperor, Nuwa, Sun Wukong: Father, mother, rebel son.
    - Relation: Family enforces harmony.
    - Lesson: Duty curbs chaos.

17. 🐙 Cthulhu Mythos
    - Cthulhu, Nyarlathotep, Yog-Sothoth: Elder kin.
    - Relation: Cosmic trio overwhelms.
    - Lesson: Insignificance humbles.

18. ☥ Egyptian Mythology
    - Ra, Osiris, Isis: Father, son, mother.
    - Relation: Family ensures renewal.
    - Lesson: Justice prevails.

19. ❄️ Finnish Mythology
    - Väinämöinen, Louhi, Ukko: Son, mother, father.
    - Relation: Kinship tests wisdom.
    - Lesson: Perseverance wins.

20. 🏛️ Greek Mythology
    - Zeus, Hera, Athena: Father, mother, daughter.
    - Relation: Family rules with tension.
    - Lesson: Hubris meets wisdom.

21. 🕉️ Hindu Trimurti
    - Brahma, Vishnu, Shiva: Creator, preserver, destroyer.
    - Relation: Divine trio cycles existence.
    - Lesson: Balance sustains life.

22. 🌺 Hindu Avatars & Devis
    - Krishna, Rama, Durga: Sons and fierce mother.
    - Relation: Avatars and goddess protect dharma.
    - Lesson: Duty defeats evil.

23. 🌸 Japanese Mythology
    - Amaterasu, Susanoo, Tsukuyomi: Sister, brothers.
    - Relation: Siblings balance cosmos.
    - Lesson: Harmony versus chaos.

24. 🗡️ Melnibonean Legends
    - Arioch, Xiombarg, Elric: Lords and mortal son.
    - Relation: Pact binds chaos.
    - Lesson: Power corrupts.

25. ☪️ Muslim Divine & Messengers
    - Allah, Muhammad, Gabriel: God, prophet, angel.
    - Relation: Messenger reveals divine will.
    - Lesson: Submission brings peace.

26. 👻 Muslim Spirits & Kin
    - Jinn, Iblis, Khidr: Spirits and guide defy or aid.
    - Relation: Supernatural kin test faith.
    - Lesson: Obedience versus rebellion.

27. 🏰 Nehwon Legends
    - Death, Ningauble, Sheelba: Fateful trio.
    - Relation: Guides shape destiny.
    - Lesson: Cunning defies fate.

28. 🧝 Nonhuman Traditions
    - Corellon, Moradin, Gruumsh: Elf, dwarf, orc fathers.
    - Relation: Rivals define purpose.
    - Lesson: Community endures.

29. ᚱ Norse Mythology
    - Odin, Frigg, Loki: Father, mother, trickster son.
    - Relation: Family faces doom.
    - Lesson: Sacrifice costs.

30. 🗿 Sumerian Mythology
    - Enki, Inanna, Anu: Son, daughter, father.
    - Relation: Kin wield knowledge.
    - Lesson: Ambition shapes.

31. 📚 Appendices
    - Planes: Realms of gods, spirits, saints, such as Paradise and Svarga.
    - Symbols: Rituals and artifacts of faith.
    - Charts: Domains and duties for devotees.

32. 🌌 Planes of Existence
    - Heaven/Paradise: Christian/Muslim abode.
    - Svarga: Hindu divine realm.
    - Underworld: Spirits linger, for example, Sheol and Naraka.

33. 🕍 Temple Trappings
    - Cross/Crescent: Christian/Muslim faith.
    - Mandalas: Hindu devotion.
    - Relics: Saints’ and prophets’ legacy.

34. 📊 Clerical Chart
    - Gods: Domains, such as creation and mercy.
    - Spirits: Influence, like guidance and mischief.
    - Saints/Prophets: Virtues, for instance, justice and prophecy.
"""

create_pdf_tab(default_markdown)