Update app.py

app.py

@@ -1,18 +1,40 @@
+import os
+import urllib.request
+import io
+import re
+
 import streamlit as st
-import base64
+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
-import io
-import re
-import fitz  # PyMuPDF
-from PIL import Image
+from reportlab.pdfbase import pdfmetrics
+from reportlab.pdfbase.ttfonts import TTFont
+
+# --- Step 1: Auto-download and register the emoji-supporting font ---
 
-# Set page config for wide layout and collapsed sidebar
+# URL for the Noto Emoji TTF file
+font_url = "https://github.com/googlefonts/noto-emoji/raw/main/fonts/NotoEmoji-Regular.ttf"
+font_path = "NotoEmoji-Regular.ttf"
+
+if not os.path.exists(font_path):
+    st.info("Downloading Noto Emoji font...")
+    try:
+        urllib.request.urlretrieve(font_url, font_path)
+        st.success("Noto Emoji font downloaded.")
+    except Exception as e:
+        st.error(f"Failed to download font: {e}")
+
+# Register the downloaded font with ReportLab under the name "NotoEmoji"
+pdfmetrics.registerFont(TTFont('NotoEmoji', font_path))
+
+# --- Streamlit App Setup ---
 st.set_page_config(layout="wide", initial_sidebar_state="collapsed")
 
-# Initial markdown content
+# Default markdown content with emojis
 default_markdown = """# Cutting-Edge ML Outline
 
 ## Core ML Techniques
@@ -80,7 +102,7 @@ default_markdown = """# Cutting-Edge ML Outline
     - Documentation synthesis
 """
 
-# Process multilevel markdown for PDF output
+# --- Markdown to PDF content processing ---
 def markdown_to_pdf_content(markdown_text):
     lines = markdown_text.strip().split('\n')
     pdf_content = []
@@ -94,13 +116,13 @@ def markdown_to_pdf_content(markdown_text):
             continue
             
         if line.startswith('# '):
+            # Skip main title if desired
             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
@@ -108,7 +130,6 @@ def markdown_to_pdf_content(markdown_text):
             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:
@@ -126,7 +147,7 @@ def markdown_to_pdf_content(markdown_text):
     
     return left_column, right_column
 
-# Main PDF creation with parameterized text sizes
+# --- Main PDF Creation ---
 def create_main_pdf(markdown_text, base_font_size=10, auto_size=False):
     buffer = io.BytesIO()
     doc = SimpleDocTemplate(
@@ -141,10 +162,7 @@ def create_main_pdf(markdown_text, base_font_size=10, auto_size=False):
     styles = getSampleStyleSheet()
     story = []
     
-    page_height = A4[0] - 72
-    title_height = 20
     spacer_height = 10
-    
     left_column, right_column = markdown_to_pdf_content(markdown_text)
     
     total_items = 0
@@ -156,24 +174,28 @@ def create_main_pdf(markdown_text, base_font_size=10, auto_size=False):
             else:
                 total_items += 1
     
-    # 🔧 Adjust this multiplier to control autosizing sensitivity
     if auto_size:
         base_font_size = max(6, min(12, 200 / total_items))
     
-    # 🔧 Font size parameters - tweak these ratios as needed
     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)
     
-    title_style = styles['Heading1']
-    title_style.textColor = colors.darkblue
-    title_style.alignment = 1
-    title_style.fontSize = title_font_size
+    # Define styles using the emoji-supporting font "NotoEmoji"
+    title_style = ParagraphStyle(
+        'Heading1',
+        parent=styles['Heading1'],
+        fontName='NotoEmoji',
+        textColor=colors.darkblue,
+        alignment=1,
+        fontSize=title_font_size
+    )
     
     section_style = ParagraphStyle(
         'SectionStyle',
         parent=styles['Heading2'],
+        fontName='NotoEmoji',
         textColor=colors.darkblue,
         fontSize=section_font_size,
         leading=section_font_size * 1.2,
@@ -183,15 +205,16 @@ def create_main_pdf(markdown_text, base_font_size=10, auto_size=False):
     item_style = ParagraphStyle(
         'ItemStyle',
         parent=styles['Normal'],
+        fontName='NotoEmoji',
         fontSize=item_font_size,
         leading=item_font_size * 1.2,
-        fontName='Helvetica-Bold',
         spaceAfter=1
     )
     
     subitem_style = ParagraphStyle(
         'SubItemStyle',
         parent=styles['Normal'],
+        fontName='NotoEmoji',
         fontSize=subitem_font_size,
         leading=subitem_font_size * 1.2,
         leftIndent=10,
@@ -252,12 +275,12 @@ def create_main_pdf(markdown_text, base_font_size=10, auto_size=False):
     buffer.seek(0)
     return buffer.getvalue()
 
-# Function to convert PDF bytes to image using fitz
+# --- Function to Convert PDF Bytes to Image (for Preview) ---
 def pdf_to_image(pdf_bytes):
     try:
         doc = fitz.open(stream=pdf_bytes, filetype="pdf")
         page = doc[0]
-        pix = page.get_pixmap(matrix=fitz.Matrix(2.0, 2.0))  # 2x zoom
+        pix = page.get_pixmap(matrix=fitz.Matrix(2.0, 2.0))  # 2x zoom for clarity
         img = Image.frombytes("RGB", [pix.width, pix.height], pix.samples)
         doc.close()
         return img
@@ -265,7 +288,7 @@ def pdf_to_image(pdf_bytes):
         st.error(f"Failed to render PDF preview: {e}")
         return None
 
-# Sidebar for settings (collapsed by default)
+# --- Sidebar UI for Settings ---
 with st.sidebar:
     auto_size = st.checkbox("Auto-size text", value=True)
     if not auto_size:
@@ -274,15 +297,15 @@ with st.sidebar:
         base_font_size = 10
         st.info("Font size will auto-adjust between 6-12 points based on content length.")
 
-# Use session state to persist markdown content
+# --- Persist Markdown Content in Session State ---
 if 'markdown_content' not in st.session_state:
     st.session_state.markdown_content = default_markdown
 
-# Generate PDF
+# --- Generate PDF ---
 with st.spinner("Generating PDF..."):
     pdf_bytes = create_main_pdf(st.session_state.markdown_content, base_font_size, auto_size)
 
-# Display PDF preview in a full-width container
+# --- Display PDF Preview in UI ---
 with st.container():
     pdf_image = pdf_to_image(pdf_bytes)
     if pdf_image:
@@ -290,7 +313,7 @@ with st.container():
     else:
         st.info("Download the PDF to view it locally.")
 
-# Download button
+# --- PDF Download Button ---
 st.download_button(
     label="Download PDF",
     data=pdf_bytes,
@@ -298,22 +321,22 @@ st.download_button(
     mime="application/pdf"
 )
 
-# Markdown editor
+# --- Markdown Editor ---
 edited_markdown = st.text_area(
     "Modify the markdown content below:",
     value=st.session_state.markdown_content,
     height=300
 )
 
-# Update markdown and regenerate PDF on change
+# --- Update PDF on Button Click ---
 if st.button("Update PDF"):
     st.session_state.markdown_content = edited_markdown
-    st.rerun()
+    st.experimental_rerun()
 
-# Save markdown option
+# --- Markdown Download Button ---
 st.download_button(
     label="Save Markdown",
     data=st.session_state.markdown_content,
     file_name="ml_outline.md",
     mime="text/markdown"
-)
+)