Update app.py

app.py

@@ -21,11 +21,11 @@ from reportlab.pdfbase.ttfonts import TTFont
 # 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"
+    "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"
 }
 
 # Sidebar: Let the user choose the desired NotoEmoji font.
@@ -176,4 +176,176 @@ def create_main_pdf(markdown_text, base_font_size=10, auto_size=False):
     for col in (left_column, right_column):
         for item in col:
             if isinstance(item, list):
-                main_item, sub_items
+                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)
+    
+    # Define styles using the selected NotoEmoji font.
+    title_style = ParagraphStyle(
+        'Heading1',
+        parent=styles['Heading1'],
+        fontName=selected_font_name,
+        textColor=colors.darkblue,
+        alignment=1,
+        fontSize=title_font_size
+    )
+    
+    section_style = ParagraphStyle(
+        'SectionStyle',
+        parent=styles['Heading2'],
+        fontName=selected_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=selected_font_name,
+        fontSize=item_font_size,
+        leading=item_font_size * 1.2,
+        spaceAfter=1
+    )
+    
+    subitem_style = ParagraphStyle(
+        'SubItemStyle',
+        parent=styles['Normal'],
+        fontName=selected_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()
+
+# ---------------------------------------------------------------
+# 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")
+        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 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 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_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"
+)