import streamlit as st
import os
import re

# Function to read markdown files with error handling
def load_markdown_file(path):
    try:
        with open(path, 'r', encoding='utf-8') as file:
            content = file.read()
        return content
    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')

docs_base_path = './docs'

if not os.path.exists(docs_base_path):
    st.error('Documentation directory does not exist.')
else:
    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')
            category = st.sidebar.selectbox('📁 Select the Categorie', categories)
            
            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:
                st.error('No markdown pages found in the selected category.')
            else:
                page = st.sidebar.selectbox('📜 Select the Page', pages)
                
                markdown_path = os.path.join(pages_path, page)
                markdown_content = load_markdown_file(markdown_path)
                
                # 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}")