Update app.py

app.py

@@ -1,7 +1,8 @@
 import streamlit as st
 import os
+import re
 
-# Improved function to read markdown files with error handling
+# Function to read markdown files with error handling
 def load_markdown_file(path):
     try:
         with open(path, 'r', encoding='utf-8') as file:
@@ -10,27 +11,33 @@ def load_markdown_file(path):
     except Exception as e:
         return f"Error reading file: {e}"
 
+# Function to create a simple TOC from markdown content
+def create_toc(markdown_content):
+    toc = []
+    for line in markdown_content.split('\n'):
+        if line.startswith('#'):
+            level = line.count('#')
+            title = line.strip('# ').strip()
+            anchor = title.lower().replace(' ', '-').replace('/', '-').replace('.', '')
+            toc.append((level, title, anchor))
+    return toc
+
 # Streamlit UI Enhancements
 st.title('📚 Documentation')
 
-# Assuming your documentation is in the 'docs' folder
 docs_base_path = './docs'
 
-# Enhanced error handling for directory listing
 if not os.path.exists(docs_base_path):
     st.error('Documentation directory does not exist.')
 else:
-    # List categories based on folder names with improved error handling
     try:
         categories = [d for d in os.listdir(docs_base_path) if os.path.isdir(os.path.join(docs_base_path, d))]
         if not categories:
             st.error('No categories found in the documentation directory.')
         else:
-            st.sidebar.markdown('## 🧷 Navigation')
-            
+            st.sidebar.markdown('# 🧷 Navigation')
             category = st.sidebar.selectbox('📁 Select the Space', categories)
             
-            # List pages based on markdown files in the selected category folder
             pages_path = os.path.join(docs_base_path, category)
             pages = [f for f in os.listdir(pages_path) if os.path.isfile(os.path.join(pages_path, f)) and f.endswith('.md')]
             if not pages:
@@ -38,9 +45,18 @@ else:
             else:
                 page = st.sidebar.selectbox('📜 Select the Page', pages)
                 
-                # Load and display the selected markdown file
                 markdown_path = os.path.join(pages_path, page)
                 markdown_content = load_markdown_file(markdown_path)
-                st.markdown(markdown_content, unsafe_allow_html=True)
+                
+                # Create TOC
+                toc = create_toc(markdown_content)
+                if toc:
+                    st.sidebar.markdown('# 📑 Table of Contents')
+                    for level, title, anchor in toc:
+                        st.sidebar.markdown(f"{' ' * (level-1) * 2}- [{title}](#{anchor})")
+                
+                # Display markdown with potential anchors
+                modified_markdown_content = re.sub(r'(#{1,6}) (.*)', r'\1 <a name="\2"></a>\2', markdown_content)
+                st.markdown(modified_markdown_content, unsafe_allow_html=True)
     except Exception as e:
-        st.error(f"Error listing categories or pages: {e}")
+        st.error(f"Error listing categories or pages: {e}")