Update app.py

app.py

@@ -1,10 +1,9 @@
 import os
-import urllib.request
 import io
 import re
 import streamlit as st
 
-# Set the page configuration as the very first Streamlit command.
+# Must be the first Streamlit command.
 st.set_page_config(layout="wide", initial_sidebar_state="collapsed")
 
 from PIL import Image
@@ -17,41 +16,30 @@ from reportlab.lib import colors
 from reportlab.pdfbase import pdfmetrics
 from reportlab.pdfbase.ttfonts import TTFont
 
-# --- Step 1: Define and Download Available Emoji Fonts ---
-font_files = [
-    "Noto-COLRv1-emojicompat.ttf",
-    "Noto-COLRv1-noflags.ttf",
-    "Noto-COLRv1.ttf",
-    "NotoColorEmoji-emojicompat.ttf",
-    "NotoColorEmoji-flagsonly.ttf",
-    "NotoColorEmoji-noflags.ttf",
-    "NotoColorEmoji.ttf",
-    "NotoColorEmoji_WindowsCompatible.ttf"
-]
-
-base_font_url = "https://github.com/googlefonts/noto-emoji/raw/main/fonts/"
-
-for font_file in font_files:
-    if not os.path.exists(font_file):
-        st.info(f"Downloading {font_file}...")
-        try:
-            urllib.request.urlretrieve(base_font_url + font_file, font_file)
-            st.success(f"Downloaded {font_file}")
-        except Exception as e:
-            st.error(f"Failed to download {font_file}: {e}")
-
-# --- Step 2: Allow User to Select the Emoji Font ---
-font_display_names = {f: f.replace(".ttf", "") for f in font_files}
-selected_font_file = st.sidebar.selectbox(
-    "Select Emoji Font",
-    options=font_files,
-    format_func=lambda f: font_display_names[f]
+# ---------------------------------------------------------------
+# Define available NotoEmoji fonts (local files)
+# One font is at the root and others are in the 'static' subdirectory.
+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 desired NotoEmoji font.
+selected_font_name = st.sidebar.selectbox(
+    "Select NotoEmoji Font",
+    options=list(available_fonts.keys())
 )
+selected_font_path = available_fonts[selected_font_name]
 
-registered_font_name = font_display_names[selected_font_file]
-pdfmetrics.registerFont(TTFont(registered_font_name, selected_font_file))
+# Register the chosen font with ReportLab.
+pdfmetrics.registerFont(TTFont(selected_font_name, selected_font_path))
 
-# --- Default Markdown Content with Emojis ---
+# ---------------------------------------------------------------
+# Default markdown content with emojis.
 default_markdown = """# Cutting-Edge ML Outline
 
 ## Core ML Techniques
@@ -119,7 +107,8 @@ default_markdown = """# Cutting-Edge ML Outline
     - Documentation synthesis
 """
 
-# --- Markdown to PDF Content Processing ---
+# ---------------------------------------------------------------
+# Process markdown into PDF content.
 def markdown_to_pdf_content(markdown_text):
     lines = markdown_text.strip().split('\n')
     pdf_content = []
@@ -133,7 +122,7 @@ def markdown_to_pdf_content(markdown_text):
             continue
             
         if line.startswith('# '):
-            # Optionally skip main title
+            # Optionally skip the main title.
             pass
         elif line.startswith('## '):
             if current_item and sub_items:
@@ -164,7 +153,8 @@ def markdown_to_pdf_content(markdown_text):
     
     return left_column, right_column
 
-# --- Main PDF Creation ---
+# ---------------------------------------------------------------
+# Create PDF using ReportLab.
 def create_main_pdf(markdown_text, base_font_size=10, auto_size=False):
     buffer = io.BytesIO()
     doc = SimpleDocTemplate(
@@ -178,181 +168,12 @@ def create_main_pdf(markdown_text, base_font_size=10, auto_size=False):
     
     styles = getSampleStyleSheet()
     story = []
-    
     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:
             if isinstance(item, list):
-                main_item, sub_items = item
-                total_items += 1 + len(sub_items)
-            else:
-                total_items += 1
-    
-    if auto_size:
-        base_font_size = max(6, min(12, 200 / total_items))
-    
-    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 = ParagraphStyle(
-        'Heading1',
-        parent=styles['Heading1'],
-        fontName=registered_font_name,
-        textColor=colors.darkblue,
-        alignment=1,
-        fontSize=title_font_size
-    )
-    
-    section_style = ParagraphStyle(
-        'SectionStyle',
-        parent=styles['Heading2'],
-        fontName=registered_font_name,
-        textColor=colors.darkblue,
-        fontSize=section_font_size,
-        leading=section_font_size * 1.2,
-        spaceAfter=2
-    )
-    
-    item_style = ParagraphStyle(
-        'ItemStyle',
-        parent=styles['Normal'],
-        fontName=registered_font_name,
-        fontSize=item_font_size,
-        leading=item_font_size * 1.2,
-        spaceAfter=1
-    )
-    
-    subitem_style = ParagraphStyle(
-        'SubItemStyle',
-        parent=styles['Normal'],
-        fontName=registered_font_name,
-        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(Spacer(1, spacer_height))
-    
-    left_cells = []
-    for item in left_column:
-        if isinstance(item, str) and item.startswith('<b>'):
-            text = item.replace('<b>', '').replace('</b>', '')
-            left_cells.append(Paragraph(text, section_style))
-        elif isinstance(item, list):
-            main_item, sub_items = item
-            left_cells.append(Paragraph(main_item, item_style))
-            for sub_item in sub_items:
-                left_cells.append(Paragraph(sub_item, subitem_style))
-        else:
-            left_cells.append(Paragraph(item, 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))
-        elif isinstance(item, list):
-            main_item, sub_items = item
-            right_cells.append(Paragraph(main_item, item_style))
-            for sub_item in sub_items:
-                right_cells.append(Paragraph(sub_item, subitem_style))
-        else:
-            right_cells.append(Paragraph(item, item_style))
-    
-    max_cells = max(len(left_cells), len(right_cells))
-    left_cells.extend([""] * (max_cells - len(left_cells)))
-    right_cells.extend([""] * (max_cells - len(right_cells)))
-    
-    table_data = list(zip(left_cells, right_cells))
-    col_width = (A4[1] - 72) / 2.0
-    
-    table = Table(table_data, colWidths=[col_width, col_width], 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), (0, -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()
-
-# --- 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))
-        img = Image.frombytes("RGB", [pix.width, pix.height], pix.samples)
-        doc.close()
-        return img
-    except Exception as e:
-        st.error(f"Failed to render PDF preview: {e}")
-        return None
-
-# --- Sidebar UI for Additional Settings ---
-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 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 UI ---
-with st.container():
-    pdf_image = pdf_to_image(pdf_bytes)
-    if pdf_image:
-        st.image(pdf_image, 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="ml_outline.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.experimental_rerun()
-
-# --- Markdown Download Button ---
-st.download_button(
-    label="Save Markdown",
-    data=st.session_state.markdown_content,
-    file_name="ml_outline.md",
-    mime="text/markdown"
-)
+                main_item, sub_items