Update app.py

app.py

@@ -213,7 +213,7 @@ def create_pdf(markdown_text, base_font_size, num_columns, add_space_before_numb
     usable_width = page_width - 72
     avg_line_chars = total_chars / total_lines if total_lines > 0 else 50
     ideal_lines_per_col = 20
-    suggested_columns = max(1, min(6, int(total_lines / ideal_lines_per_col) + 1))
+    suggested_columns = max(2, min(4, int(total_lines / ideal_lines_per_col) + 1))
     num_columns = num_columns if num_columns != 0 else suggested_columns
     col_width = usable_width / num_columns
     min_font_size = 6
@@ -492,33 +492,37 @@ with st.sidebar:
     )
     auto_columns = st.checkbox("AutoColumns", value=True)
     
-    if auto_columns and 'markdown_content' in st.session_state:
+    # Calculate document stats
+    longest_line_words = 0
+    total_lines = 0
+    if 'markdown_content' in st.session_state:
         current_markdown = st.session_state.markdown_content
         lines = current_markdown.strip().split('\n')
-        longest_line_words = 0
+        total_lines = len([line for line in lines if line.strip()])
         for line in lines:
             if line.strip():
                 word_count = len(line.split())
                 longest_line_words = max(longest_line_words, word_count)
-        if longest_line_words > 38:
-            recommended_columns = 2
-        elif longest_line_words >= 25:
-            recommended_columns = 3
-        elif longest_line_words >= 18:
-            recommended_columns = 4
+        if auto_columns:
+            if longest_line_words > 38:
+                recommended_columns = 2
+            elif longest_line_words < 18 and total_lines < 20:
+                recommended_columns = 4
+            else:
+                recommended_columns = 3
         else:
-            recommended_columns = 5
-        st.info(f"Longest line has {longest_line_words} words. Recommending {recommended_columns} columns.")
-    else:
-        recommended_columns = 3
+            recommended_columns = 3
+        st.markdown(f"**Document Stats**")
+        st.write(f"- Longest Line: {longest_line_words} words")
+        st.write(f"- Total Lines: {total_lines}")
+        st.write(f"- Recommended Columns: {recommended_columns}")
     
-    column_options = ["Auto"] + list(range(1, 7))
+    column_options = [2, 3, 4]
     num_columns = st.selectbox(
         "Number of Columns",
         options=column_options,
-        index=0 if recommended_columns == "Auto" else column_options.index(recommended_columns)
+        index=column_options.index(recommended_columns)
     )
-    num_columns = 0 if num_columns == "Auto" else int(num_columns)
     st.info("Font size and columns adjust to fit one page.")
     
     st.markdown("### ✍️ Edit Markdown")