Update app.py

app.py

@@ -7,8 +7,8 @@ from reportlab.lib import colors
 import io
 import re
 
-# Define the ML outline as a markdown string
-ml_markdown = """# Cutting-Edge ML Outline
+# Initial markdown content
+default_markdown = """# Cutting-Edge ML Outline
 
 ## Core ML Techniques
 1. 🌟 **Mixture of Experts (MoE)**
@@ -77,7 +77,6 @@ ml_markdown = """# Cutting-Edge ML Outline
 
 # Process multilevel markdown for PDF output
 def markdown_to_pdf_content(markdown_text):
-    """Convert markdown text to a format suitable for PDF generation"""
     lines = markdown_text.strip().split('\n')
     pdf_content = []
     in_list_item = False
@@ -122,9 +121,8 @@ def markdown_to_pdf_content(markdown_text):
     
     return left_column, right_column
 
-# Main PDF creation using ReportLab
-def create_main_pdf(markdown_text):
-    """Create a single-page landscape PDF with the outline in two columns"""
+# Main PDF creation with parameterized text sizes
+def create_main_pdf(markdown_text, base_font_size=10, auto_size=False):
     buffer = io.BytesIO()
     doc = SimpleDocTemplate(
         buffer, 
@@ -141,12 +139,11 @@ def create_main_pdf(markdown_text):
     page_height = A4[0] - 72
     title_height = 20
     spacer_height = 10
-    available_content_height = page_height - title_height - spacer_height
     
-    # Process columns first
+    # Process columns
     left_column, right_column = markdown_to_pdf_content(markdown_text)
     
-    # Calculate total items by explicitly handling the unpacking
+    # Calculate total items for autosizing
     total_items = 0
     for col in (left_column, right_column):
         for item in col:
@@ -156,17 +153,21 @@ def create_main_pdf(markdown_text):
             else:
                 total_items += 1
     
-    # Dynamic font sizes
-    base_font_size = max(6, min(11, 200 / total_items))
+    # 🔧 Adjust this multiplier to control autosizing sensitivity
+    if auto_size:
+        base_font_size = max(6, min(12, 200 / total_items))  # Range: 6-12 points
+    
+    # 🔧 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)
     
     # Create custom styles
     title_style = styles['Heading1']
     title_style.textColor = colors.darkblue
-    title_style.alignment = 1 
-    title_style.fontSize = min(16, base_font_size * 1.5)
+    title_style.alignment = 1
+    title_style.fontSize = title_font_size
     
     section_style = ParagraphStyle(
         'SectionStyle',
@@ -259,16 +260,63 @@ def create_main_pdf(markdown_text):
 # Streamlit UI
 st.title("🚀 Cutting-Edge ML Outline Generator")
 
-if st.button("Generate Main PDF"):
-    with st.spinner("Generating PDF..."):
-        pdf_bytes = create_main_pdf(ml_markdown)
-        st.download_button(
-            label="Download Main PDF",
-            data=pdf_bytes,
-            file_name="ml_outline.pdf",
-            mime="application/pdf"
-        )
-        base64_pdf = base64.b64encode(pdf_bytes).decode('utf-8')
-        pdf_display = f'<embed src="data:application/pdf;base64,{base64_pdf}" width="100%" height="400px" type="application/pdf">'
-        st.markdown(pdf_display, unsafe_allow_html=True)
-        st.success("PDF generated successfully!")
+# Sidebar for settings
+with st.sidebar:
+    st.header("PDF Settings")
+    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  # Default, overridden by auto-sizing
+        # 🔧 Adjust autosizing formula in create_main_pdf if needed
+        st.info("Font size will auto-adjust between 6-12 points based on content length.")
+
+# Use session state to persist markdown content
+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)
+    base64_pdf = base64.b64encode(pdf_bytes).decode('utf-8')
+
+# Display PDF full-screen
+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)
+
+# Download button
+st.download_button(
+    label="Download PDF",
+    data=pdf_bytes,
+    file_name="ml_outline.pdf",
+    mime="application/pdf"
+)
+
+# Markdown editor
+st.subheader("Edit Markdown Outline")
+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
+if st.button("Update PDF"):
+    st.session_state.markdown_content = edited_markdown
+    st.rerun()  # Rerun to update PDF with new markdown
+
+# Save markdown option
+st.download_button(
+    label="Save Markdown",
+    data=st.session_state.markdown_content,
+    file_name="ml_outline.md",
+    mime="text/markdown"
+)