Update app.py

app.py

@@ -3,7 +3,7 @@ import io
 import re
 import streamlit as st
 
-# Must be the first Streamlit command.
+# Must be the very first Streamlit command.
 st.set_page_config(layout="wide", initial_sidebar_state="collapsed")
 
 from PIL import Image
@@ -17,15 +17,14 @@ from reportlab.pdfbase import pdfmetrics
 from reportlab.pdfbase.ttfonts import TTFont
 
 # ---------------------------------------------------------------
-# Define available NotoEmoji fonts (local files)
-# One font is at the root and others are in the 'static' subdirectory.
+# Define available NotoEmoji fonts (all in the base directory now)
 available_fonts = {
     "NotoEmoji Variable": "NotoEmoji-VariableFont_wght.ttf",
-    "NotoEmoji Bold": "static/NotoEmoji-Bold.ttf",
-    "NotoEmoji Light": "static/NotoEmoji-Light.ttf",
-    "NotoEmoji Medium": "static/NotoEmoji-Medium.ttf",
-    "NotoEmoji Regular": "static/NotoEmoji-Regular.ttf",
-    "NotoEmoji SemiBold": "static/NotoEmoji-SemiBold.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 desired NotoEmoji font.
@@ -35,9 +34,29 @@ selected_font_name = st.sidebar.selectbox(
 )
 selected_font_path = available_fonts[selected_font_name]
 
-# Register the chosen font with ReportLab.
+# 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):
+    # This regex attempts to capture many common emoji ranges.
+    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]+)"
+    )
+    # Wrap found emoji with a font tag using the selected emoji font.
+    return emoji_pattern.sub(r'<font face="{}">\1</font>'.format(emoji_font), text)
+
 # ---------------------------------------------------------------
 # Default markdown content with emojis.
 default_markdown = """# Cutting-Edge ML Outline
@@ -108,7 +127,7 @@ default_markdown = """# Cutting-Edge ML Outline
 """
 
 # ---------------------------------------------------------------
-# Process markdown into PDF content.
+# Process markdown into a two-column layout for the PDF.
 def markdown_to_pdf_content(markdown_text):
     lines = markdown_text.strip().split('\n')
     pdf_content = []
@@ -154,7 +173,7 @@ def markdown_to_pdf_content(markdown_text):
     return left_column, right_column
 
 # ---------------------------------------------------------------
-# Create PDF using ReportLab.
+# Create the PDF using ReportLab.
 def create_main_pdf(markdown_text, base_font_size=10, auto_size=False):
     buffer = io.BytesIO()
     doc = SimpleDocTemplate(
@@ -171,7 +190,6 @@ def create_main_pdf(markdown_text, base_font_size=10, auto_size=False):
     spacer_height = 10
     left_column, right_column = markdown_to_pdf_content(markdown_text)
     
-    # Count total items to possibly adjust font size.
     total_items = 0
     for col in (left_column, right_column):
         for item in col:
@@ -189,11 +207,11 @@ def create_main_pdf(markdown_text, base_font_size=10, auto_size=False):
     section_font_size = base_font_size * 1.2
     title_font_size = min(16, base_font_size * 1.5)
     
-    # Define styles using the selected NotoEmoji font.
+    # Define ParagraphStyles using Helvetica for normal text.
     title_style = ParagraphStyle(
         'Heading1',
         parent=styles['Heading1'],
-        fontName=selected_font_name,
+        fontName="Helvetica-Bold",
         textColor=colors.darkblue,
         alignment=1,
         fontSize=title_font_size
@@ -202,7 +220,7 @@ def create_main_pdf(markdown_text, base_font_size=10, auto_size=False):
     section_style = ParagraphStyle(
         'SectionStyle',
         parent=styles['Heading2'],
-        fontName=selected_font_name,
+        fontName="Helvetica-Bold",
         textColor=colors.darkblue,
         fontSize=section_font_size,
         leading=section_font_size * 1.2,
@@ -212,7 +230,7 @@ def create_main_pdf(markdown_text, base_font_size=10, auto_size=False):
     item_style = ParagraphStyle(
         'ItemStyle',
         parent=styles['Normal'],
-        fontName=selected_font_name,
+        fontName="Helvetica",
         fontSize=item_font_size,
         leading=item_font_size * 1.2,
         spaceAfter=1
@@ -221,41 +239,42 @@ def create_main_pdf(markdown_text, base_font_size=10, auto_size=False):
     subitem_style = ParagraphStyle(
         'SubItemStyle',
         parent=styles['Normal'],
-        fontName=selected_font_name,
+        fontName="Helvetica",
         fontSize=subitem_font_size,
         leading=subitem_font_size * 1.2,
         leftIndent=10,
         spaceAfter=1
     )
     
-    story.append(Paragraph("Cutting-Edge ML Outline (ReportLab)", title_style))
+    story.append(Paragraph(apply_emoji_font("Cutting-Edge ML Outline (ReportLab)", selected_font_name), title_style))
     story.append(Spacer(1, spacer_height))
     
     left_cells = []
     for item in left_column:
         if isinstance(item, str) and item.startswith('<b>'):
+            # Process section headings.
             text = item.replace('<b>', '').replace('</b>', '')
-            left_cells.append(Paragraph(text, section_style))
+            left_cells.append(Paragraph(apply_emoji_font(text, selected_font_name), section_style))
         elif isinstance(item, list):
             main_item, sub_items = item
-            left_cells.append(Paragraph(main_item, item_style))
+            left_cells.append(Paragraph(apply_emoji_font(main_item, selected_font_name), item_style))
             for sub_item in sub_items:
-                left_cells.append(Paragraph(sub_item, subitem_style))
+                left_cells.append(Paragraph(apply_emoji_font(sub_item, selected_font_name), subitem_style))
         else:
-            left_cells.append(Paragraph(item, item_style))
+            left_cells.append(Paragraph(apply_emoji_font(item, selected_font_name), item_style))
     
     right_cells = []
     for item in right_column:
         if isinstance(item, str) and item.startswith('<b>'):
             text = item.replace('<b>', '').replace('</b>', '')
-            right_cells.append(Paragraph(text, section_style))
+            right_cells.append(Paragraph(apply_emoji_font(text, selected_font_name), section_style))
         elif isinstance(item, list):
             main_item, sub_items = item
-            right_cells.append(Paragraph(main_item, item_style))
+            right_cells.append(Paragraph(apply_emoji_font(main_item, selected_font_name), item_style))
             for sub_item in sub_items:
-                right_cells.append(Paragraph(sub_item, subitem_style))
+                right_cells.append(Paragraph(apply_emoji_font(sub_item, selected_font_name), subitem_style))
         else:
-            right_cells.append(Paragraph(item, item_style))
+            right_cells.append(Paragraph(apply_emoji_font(item, selected_font_name), item_style))
     
     max_cells = max(len(left_cells), len(right_cells))
     left_cells.extend([""] * (max_cells - len(left_cells)))
@@ -310,7 +329,7 @@ if 'markdown_content' not in st.session_state:
     st.session_state.markdown_content = default_markdown
 
 # ---------------------------------------------------------------
-# Generate PDF.
+# Generate the PDF.
 with st.spinner("Generating PDF..."):
     pdf_bytes = create_main_pdf(st.session_state.markdown_content, base_font_size, auto_size)