import os
import io
import re
import streamlit as st

# Must be the very first Streamlit command.
st.set_page_config(layout="wide", initial_sidebar_state="collapsed")

from PIL import Image
import fitz  # PyMuPDF

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

# ---------------------------------------------------------------
# Define available NotoEmoji fonts (assumed in the base directory)
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: Let the user choose the NotoEmoji font.
selected_font_name = st.sidebar.selectbox(
    "Select NotoEmoji Font",
    options=list(available_fonts.keys())
)
selected_font_path = available_fonts[selected_font_name]

# Register the chosen emoji font with ReportLab.
pdfmetrics.registerFont(TTFont(selected_font_name, selected_font_path))

# ---------------------------------------------------------------
# Helper function to wrap emoji characters with a font tag.
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]+)"
    )
    return emoji_pattern.sub(r'<font face="{}">\1</font>'.format(emoji_font), text)

# ---------------------------------------------------------------
# Default markdown content with your Deities Guide.
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.  
    Gabri    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).
"""

# ---------------------------------------------------------------
# Process markdown into a two-page layout for the PDF.
def markdown_to_pdf_content(markdown_text):
    lines = markdown_text.strip().split('\n')
    pdf_content = []
    in_list_item = False
    current_item = None
    sub_items = []
    
    for line in lines:
        line = line.strip()
        if not line:
            continue
            
        if line.startswith('# '):
            # Skip the main title for now, added separately.
            pass
        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_item = 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_item = True
        elif line.startswith('- ') and in_list_item:
            sub_items.append(line.strip())
        elif re.match(r'^\d+\.\d+\.', line):
            if current_item and sub_items:
                pdf_content.append([current_item, sub_items])
                sub_items = []
            current_item = line.strip()
            in_list_item = True
        else:
            if not in_list_item:
                pdf_content.append(line.strip())
    
    if current_item and sub_items:
        pdf_content.append([current_item, sub_items])
    
    # Split into two pages (roughly balanced).
    total_items = sum(1 + len(subs) if isinstance(item, list) else 1 for item in pdf_content)
    target_per_page = total_items // 2
    page1_content = []
    page2_content = []
    current_count = 0
    
    for item in pdf_content:
        item_count = 1 + (len(item[1]) if isinstance(item, list) else 0)
        if current_count < target_per_page or len(page1_content) == 0:
            page1_content.append(item)
            current_count += item_count
        else:
            page2_content.append(item)
    
    return page1_content, page2_content

# ---------------------------------------------------------------
# Create the PDF with two pages.
def create_main_pdf(markdown_text, base_font_size=10, auto_size=False):
    buffer = io.BytesIO()
    doc = SimpleDocTemplate(
        buffer, 
        pagesize=A4,
        leftMargin=36,
        rightMargin=36,
        topMargin=36,
        bottomMargin=36
    )
    
    styles = getSampleStyleSheet()
    story = []
    spacer_height = 10
    page1_content, page2_content = markdown_to_pdf_content(markdown_text)
    
    total_items = sum(1 + len(subs) if isinstance(item, list) else 1 for item in page1_content + page2_content)
    if auto_size:
        base_font_size = max(6, min(12, 400 / total_items))  # Adjusted for two pages.
    
    item_font_size = base_font_size
    subitem_font_size = base_font_size * 0.9
    section_font_size = base_font_size * 1.2
    title_font_size = min(16, base_font_size * 1.5)
    
    # Define ParagraphStyles using Helvetica for normal text.
    title_style = ParagraphStyle(
        'Heading1',
        parent=styles['Heading1'],
        fontName="Helvetica-Bold",
        textColor=colors.darkblue,
        alignment=1,
        fontSize=title_font_size
    )
    
    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.2,
        spaceAfter=1
    )
    
    subitem_style = ParagraphStyle(
        'SubItemStyle',
        parent=styles['Normal'],
        fontName="Helvetica",
        fontSize=subitem_font_size,
        leading=subitem_font_size * 1.2,
        leftIndent=10,
        spaceAfter=1
    )
    
    # Page 1
    story.append(Paragraph(apply_emoji_font("Deities Guide: Mythology and Moral Lessons", selected_font_name), title_style))
    story.append(Spacer(1, spacer_height))
    
    for item in page1_content:
        if isinstance(item, str) and item.startswith('<b>'):
            text = item.replace('<b>', '').replace('</b>', '')
            story.append(Paragraph(apply_emoji_font(text, selected_font_name), section_style))
        elif isinstance(item, list):
            main_item, sub_items = item
            story.append(Paragraph(apply_emoji_font(main_item, selected_font_name), item_style))
            for sub_item in sub_items:
                story.append(Paragraph(apply_emoji_font(sub_item, selected_font_name), subitem_style))
        else:
            story.append(Paragraph(apply_emoji_font(item, selected_font_name), item_style))
    
    story.append(Spacer(1, A4[1] - sum(p.minHeight() for p in story) - 72))  # Fill page.
    
    # Page 2
    story.append(Paragraph(apply_emoji_font("Deities Guide: Continued", selected_font_name), title_style))
    story.append(Spacer(1, spacer_height))
    
    for item in page2_content:
        if isinstance(item, str) and item.startswith('<b>'):
            text = item.replace('<b>', '').replace('</b>', '')
            story.append(Paragraph(apply_emoji_font(text, selected_font_name), section_style))
        elif isinstance(item, list):
            main_item, sub_items = item
            story.append(Paragraph(apply_emoji_font(main_item, selected_font_name), item_style))
            for sub_item in sub_items:
                story.append(Paragraph(apply_emoji_font(sub_item, selected_font_name), subitem_style))
        else:
            story.append(Paragraph(apply_emoji_font(item, selected_font_name), item_style))
    
    doc.build(story)
    buffer.seek(0)
    return buffer.getvalue()

# ---------------------------------------------------------------
# Convert PDF bytes to an image for preview using PyMuPDF.
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

# ---------------------------------------------------------------
# Sidebar options for text size.
with st.sidebar:
    auto_size = st.checkbox("Auto-size text", value=True)
    if not auto_size:
        base_font_size = st.slider("Base Font Size (points)", min_value=6, max_value=16, value=10, step=1)
    else:
        base_font_size = 10
        st.info("Font size will auto-adjust between 6-12 points based on content length.")

# Persist markdown content in session state.
if 'markdown_content' not in st.session_state:
    st.session_state.markdown_content = default_markdown

# ---------------------------------------------------------------
# Generate the PDF.
with st.spinner("Generating PDF..."):
    pdf_bytes = create_main_pdf(st.session_state.markdown_content, base_font_size, auto_size)

# Display PDF preview.
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"Page {i+1}", use_container_width=True)
    else:
        st.info("Download the PDF to view it locally.")

# PDF Download button.
st.download_button(
    label="Download PDF",
    data=pdf_bytes,
    file_name="deities_guide.pdf",
    mime="application/pdf"
)

# Markdown editor.
edited_markdown = st.text_area(
    "Modify the markdown content below:",
    value=st.session_state.markdown_content,
    height=300
)

# Update PDF on button click.
if st.button("Update PDF"):
    st.session_state.markdown_content = edited_markdown
    st.rerun()

# Markdown Download button.
st.download_button(
    label="Save Markdown",
    data=st.session_state.markdown_content,
    file_name="deities_guide.md",
    mime="text/markdown"
)