Update app.py

app.py

@@ -316,8 +316,9 @@ with st.sidebar:
     auto_columns = st.checkbox("AutoColumns", value=False, key="auto_columns")
     
     # Auto-determine column count based on longest line if AutoColumns is checked
-    if auto_columns:
-        lines = edited_markdown.strip().split('\n')
+    if auto_columns and 'markdown_content' in st.session_state:
+        current_markdown = st.session_state.markdown_content
+        lines = current_markdown.strip().split('\n')
         longest_line_words = 0
         for line in lines:
             if line.strip():  # Skip empty lines