Update app.py

app.py

@@ -1,49 +1,61 @@
 import streamlit as st
 import os
 
-# Custom CSS to style the markdown content
-def local_css(file_name):
-    with open(file_name, "r") as f:
-        st.markdown(f"<style>{f.read()}</style>", unsafe_allow_html=True)
-
-# Function to read markdown files
+# Improved function to read markdown files with error handling
 def load_markdown_file(path):
-    with open(path, "r") as file:
-        content = file.read()
-    return content
-
-# Initialize the app with a nicer layout
-st.set_page_config(layout="wide")
-
-# Assuming you will create a style.css for custom styles
-# Make sure to create a 'style.css' file in your project directory
-local_css("style.css")
+    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}"
 
-# Streamlit UI
-st.title('📚 Documentation')
+# Streamlit UI Enhancements
+st.title('📚 Documentation Viewer')
 
 # Assuming your documentation is in the 'docs' folder
 docs_base_path = './docs'
 
-# Create columns for sidebar and main content
-col1, col2 = st.columns([1, 3])
-
-with col1:
-    st.write("## Navigation")
-    
-    # List categories based on folder names
-    categories = [d for d in os.listdir(docs_base_path) if os.path.isdir(os.path.join(docs_base_path, d))]
-    category = st.selectbox('Select a Category', 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))]
-    page = st.selectbox('Select a Page', pages)
-
-with col2:
-    st.write(f"## {page[:-3]}")  # Remove .md extension and display as title
-    
-    # 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)
+# 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:
+            category = st.sidebar.selectbox('Select a Category', 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:
+                st.error('No markdown pages found in the selected category.')
+            else:
+                page = st.sidebar.selectbox('Select a 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)
+    except Exception as e:
+        st.error(f"Error listing categories or pages: {e}")
+
+# Optional: Add a theme or custom CSS for styling
+st.markdown(
+    """
+    <style>
+    .css-18e3th9 {
+        background-color: #f0f2f6;
+    }
+    .css-1d391kg {
+        padding-top: 1rem;
+        padding-left: 1rem;
+        padding-right: 1rem;
+    }
+    </style>
+    """,
+    unsafe_allow_html=True,
+)