Update app.py

app.py

@@ -6,6 +6,8 @@ from reportlab.lib.styles import getSampleStyleSheet, ParagraphStyle
 from reportlab.lib import colors
 import io
 import re
+from pdf2image import convert_from_bytes
+from PIL import Image
 
 # Initial markdown content
 default_markdown = """# Cutting-Edge ML Outline
@@ -126,7 +128,7 @@ def create_main_pdf(markdown_text, base_font_size=10, auto_size=False):
     buffer = io.BytesIO()
     doc = SimpleDocTemplate(
         buffer, 
-        pagesize=(A4[1], A4[0]),  # Landscape A4: 841.89 x 595.27 points
+        pagesize=(A4[1], A4[0]),
         leftMargin=36,
         rightMargin=36,
         topMargin=36,
@@ -140,10 +142,8 @@ def create_main_pdf(markdown_text, base_font_size=10, auto_size=False):
     title_height = 20
     spacer_height = 10
     
-    # Process columns
     left_column, right_column = markdown_to_pdf_content(markdown_text)
     
-    # Calculate total items for autosizing
     total_items = 0
     for col in (left_column, right_column):
         for item in col:
@@ -155,7 +155,7 @@ def create_main_pdf(markdown_text, base_font_size=10, auto_size=False):
     
     # 🔧 Adjust this multiplier to control autosizing sensitivity
     if auto_size:
-        base_font_size = max(6, min(12, 200 / total_items))  # Range: 6-12 points
+        base_font_size = max(6, min(12, 200 / total_items))
     
     # 🔧 Font size parameters - tweak these ratios as needed
     item_font_size = base_font_size
@@ -163,7 +163,6 @@ 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)
     
-    # Create custom styles
     title_style = styles['Heading1']
     title_style.textColor = colors.darkblue
     title_style.alignment = 1
@@ -196,11 +195,9 @@ def create_main_pdf(markdown_text, base_font_size=10, auto_size=False):
         spaceAfter=1
     )
     
-    # Add title
     story.append(Paragraph("Cutting-Edge ML Outline (ReportLab)", title_style))
     story.append(Spacer(1, spacer_height))
     
-    # Prepare data for table
     left_cells = []
     for item in left_column:
         if isinstance(item, str) and item.startswith('<b>'):
@@ -227,18 +224,13 @@ def create_main_pdf(markdown_text, base_font_size=10, auto_size=False):
         else:
             right_cells.append(Paragraph(item, item_style))
     
-    # Make columns equal length
     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)))
     
-    # Create table data
     table_data = list(zip(left_cells, right_cells))
-    
-    # Calculate column widths
     col_width = (A4[1] - 72) / 2.0
     
-    # Create and style table
     table = Table(table_data, colWidths=[col_width, col_width], hAlign='CENTER')
     table.setStyle(TableStyle([
         ('VALIGN', (0, 0), (-1, -1), 'TOP'),
@@ -267,8 +259,7 @@ with st.sidebar:
     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  # Default, overridden by auto-sizing
-        # 🔧 Adjust autosizing formula in create_main_pdf if needed
+        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
@@ -278,19 +269,15 @@ if 'markdown_content' not in st.session_state:
 # Generate PDF
 with st.spinner("Generating PDF..."):
     pdf_bytes = create_main_pdf(st.session_state.markdown_content, base_font_size, auto_size)
-    base64_pdf = base64.b64encode(pdf_bytes).decode('utf-8')
 
-# Display PDF full-screen
+# Display PDF preview as image
 st.subheader("PDF Preview")
-pdf_display = f"""
-    <embed 
-        src="data:application/pdf;base64,{base64_pdf}" 
-        width="100%" 
-        height="600px" 
-        type="application/pdf"
-        style="border: none;">
-"""
-st.markdown(pdf_display, unsafe_allow_html=True)
+try:
+    images = convert_from_bytes(pdf_bytes, dpi=200)  # Convert PDF to image
+    st.image(images[0], caption="PDF Page 1", use_column_width=True)
+except Exception as e:
+    st.error(f"Failed to render PDF preview: {e}")
+    st.info("Download the PDF to view it locally.")
 
 # Download button
 st.download_button(
@@ -311,7 +298,7 @@ edited_markdown = st.text_area(
 # Update markdown and regenerate PDF on change
 if st.button("Update PDF"):
     st.session_state.markdown_content = edited_markdown
-    st.rerun()  # Rerun to update PDF with new markdown
+    st.rerun()
 
 # Save markdown option
 st.download_button(