Update app.py

app.py

@@ -1,184 +1,132 @@
 import streamlit as st
 import os
+from pathlib import Path
 
-DATA_DIR = "/home/user/app"  # Replace with your actual data directory
+# Use relative path for better portability
+DATA_DIR = Path("data")  # Create a 'data' directory in the same folder as the script
 
 def get_subjects():
     """Returns a list of subject directories."""
-    return [
-        name
-        for name in os.listdir(DATA_DIR)
-        if os.path.isdir(os.path.join(DATA_DIR, name))
-    ]
+    DATA_DIR.mkdir(exist_ok=True)  # Create data directory if it doesn't exist
+    return sorted([
+        name.name
+        for name in DATA_DIR.iterdir()
+        if name.is_dir()
+    ])
 
 def get_units(subject):
     """Returns a list of unit directories within a subject."""
-    subject_path = os.path.join(DATA_DIR, subject)
-    return [
-        name
-        for name in os.listdir(subject_path)
-        if os.path.isdir(os.path.join(subject_path, name))
-    ]
+    subject_path = DATA_DIR / subject
+    return sorted([
+        name.name
+        for name in subject_path.iterdir()
+        if name.is_dir()
+    ]) if subject_path.exists() else []
 
 def get_notes(subject, unit):
     """Returns a list of note files within a unit."""
-    unit_path = os.path.join(DATA_DIR, subject, unit)
-    return [
-        name
-        for name in os.listdir(unit_path)
-        if name.endswith(".md")
-    ]
+    unit_path = DATA_DIR / subject / unit
+    return sorted([
+        name.name
+        for name in unit_path.iterdir()
+        if name.suffix == '.md'
+    ]) if unit_path.exists() else []
 
 def load_note(subject, unit, note):
     """Loads the content of a note file."""
-    note_path = os.path.join(DATA_DIR, subject, unit, note)
+    note_path = DATA_DIR / subject / unit / note
     try:
-        with open(note_path, "r", encoding="utf-8") as f:
-            return f.read()
+        return note_path.read_text(encoding="utf-8")
     except Exception as e:
         st.error(f"Error loading note: {e}")
         return None
 
 st.set_page_config(
-    page_title="Beautiful Notes App",
+    page_title="Notes App",
     page_icon="📝",
-    layout="wide",  # Keep wide for potential wider content, adjust with CSS
+    layout="wide",
     initial_sidebar_state="expanded",
 )
 
-st.title("📝 Beautiful Notes App")
+# Custom CSS with simplified styling
+st.markdown("""
+    <style>
+    .main > div {
+        padding: 2rem;
+        max-width: 100%;
+    }
+    .stMarkdown {
+        font-size: 1rem;
+    }
+    </style>
+    """, unsafe_allow_html=True)
 
-# --- State Management ---
-if 'selected_subject' not in st.session_state:
-    st.session_state['selected_subject'] = None
-if 'selected_unit' not in st.session_state:
-    st.session_state['selected_unit'] = None
-if 'selected_note' not in st.session_state:
-    st.session_state['selected_note'] = None
+st.title("📝 Notes App")
 
-# Sidebar for navigation
+# Initialize session state
+for key in ['selected_subject', 'selected_unit', 'selected_note']:
+    if key not in st.session_state:
+        st.session_state[key] = None
+
+# Sidebar navigation
 with st.sidebar:
     st.header("Navigation")
+    
+    # Subject selection
     subjects = get_subjects()
-    selected_subject = st.selectbox("Select Subject", subjects, key="subject_selectbox")
-    st.session_state['selected_subject'] = selected_subject
-
-    if st.session_state['selected_subject']:
-        units = get_units(st.session_state['selected_subject'])
+    if subjects:
+        selected_subject = st.selectbox(
+            "Select Subject",
+            subjects,
+            key="subject_selectbox",
+            index=subjects.index(st.session_state['selected_subject']) if st.session_state['selected_subject'] in subjects else 0
+        )
+        st.session_state['selected_subject'] = selected_subject
+        
+        # Unit selection
+        units = get_units(selected_subject)
         if units:
-            selected_unit = st.selectbox("Select Unit", units, key="unit_selectbox")
+            selected_unit = st.selectbox(
+                "Select Unit",
+                units,
+                key="unit_selectbox",
+                index=units.index(st.session_state['selected_unit']) if st.session_state['selected_unit'] in units else 0
+            )
             st.session_state['selected_unit'] = selected_unit
-            if st.session_state['selected_unit']:
-                notes = get_notes(st.session_state['selected_subject'], st.session_state['selected_unit'])
-                if notes:
-                    selected_note = st.selectbox("Select Note", notes, key="note_selectbox")
-                    st.session_state['selected_note'] = selected_note
-                else:
-                    st.info("No notes found in this unit.")
-                    st.session_state['selected_note'] = None
+            
+            # Note selection
+            notes = get_notes(selected_subject, selected_unit)
+            if notes:
+                selected_note = st.selectbox(
+                    "Select Note",
+                    notes,
+                    key="note_selectbox",
+                    index=notes.index(st.session_state['selected_note']) if st.session_state['selected_note'] in notes else 0
+                )
+                st.session_state['selected_note'] = selected_note
             else:
-                st.info("No units found in this subject.")
+                st.info("No notes found in this unit.")
                 st.session_state['selected_note'] = None
         else:
             st.info("No units found in this subject.")
-            st.session_state['selected_note'] = None
-    else:
-        st.info("No subjects found in the data directory.")
-        st.session_state['selected_note'] = None
-
-# Main area to display notes
-col1, col2 = st.columns([1, 3])  # Adjust column widths as needed for mobile
-
-with col2:
-    if st.session_state['selected_subject'] and st.session_state['selected_unit'] and st.session_state['selected_note']:
-        st.header(f"{st.session_state['selected_subject']} / {st.session_state['selected_unit']} / {st.session_state['selected_note'][:-3]}")  # Remove .md extension
-        note_content = load_note(st.session_state['selected_subject'], st.session_state['selected_unit'], st.session_state['selected_note'])
-        if note_content:
-            st.markdown(note_content, unsafe_allow_html=True)
-    elif st.session_state['selected_subject'] and st.session_state['selected_unit']:
-        st.info("Select a note from the sidebar to view it.")
-    elif st.session_state['selected_subject']:
-        st.info("Select a unit from the sidebar to view notes.")
+            st.session_state['selected_unit'] = None
     else:
-        st.info("Select a subject from the sidebar to start exploring your notes.")
-
-# --- Styling for Mobile ---
-st.markdown(
-    """
-    <style>
-    /* General Styles */
-    body {
-        font-family: sans-serif;
-        color: #333;
-        background-color: #f4f4f4;
-        margin: 0; /* Remove default body margin */
-    }
-    .stApp {
-        max-width: 100%; /* Occupy full width */
-        padding: 10px; /* Reduced padding for smaller screens */
-    }
-    h1 {
-        color: #007bff;
-        text-align: center; /* Center the title on mobile */
-    }
-    h2, h3, h4, h5, h6 {
-        color: #0056b3;
-    }
-    .st-bf { /* Styling for selectbox labels */
-        font-weight: bold;
-    }
-    .streamlit-expander {
-        border: 1px solid #ccc;
-        border-radius: 5px;
-        margin-bottom: 10px;
-    }
-    .streamlit-expander-header {
-        font-weight: bold;
-        padding: 8px;
-    }
-    .streamlit-expander-content {
-        padding: 8px;
-    }
-
-    /* Sidebar Styles */
-    .stSidebar {
-        padding-top: 1rem;
-    }
-    .stSidebar > div[data-testid="stVerticalBlock"] {
-        padding: 20px; /* Add padding to sidebar content */
-        overflow: auto; /* Enable scrolling if content overflows */
-        max-height: calc(100vh - 50px); /* Limit height and enable scrolling */
-    }
-
-    /* Main Content Styles */
-    .block-container {
-        padding-top: 1rem !important;
-        padding-bottom: 1rem !important;
-        padding-left: 1rem !important;
-        padding-right: 1rem !important;
-    }
-
-    /* Mobile Specific Styles */
-    @media (max-width: 768px) {
-        .stApp {
-            padding: 5px; /* Further reduce padding on very small screens */
-        }
-        h1 {
-            font-size: 24px; /* Smaller title on mobile */
-        }
-        .st-bf {
-            font-size: 14px; /* Smaller label font */
-        }
-        /* Adjust selectbox size and padding */
-        div[data-baseweb="select"] > div {
-            padding-top: 6px;
-            padding-bottom: 6px;
-        }
-        .stMarkdown {
-            font-size: 16px; /* More readable font size for content */
-        }
-    }
-    </style>
-    """,
-    unsafe_allow_html=True,
-)
+        st.info("No subjects found. Add some notes to get started!")
+
+# Main content area
+if all([st.session_state['selected_subject'], 
+        st.session_state['selected_unit'], 
+        st.session_state['selected_note']]):
+    
+    st.header(f"{st.session_state['selected_subject']} / {st.session_state['selected_unit']} / {Path(st.session_state['selected_note']).stem}")
+    
+    note_content = load_note(
+        st.session_state['selected_subject'],
+        st.session_state['selected_unit'],
+        st.session_state['selected_note']
+    )
+    
+    if note_content:
+        st.markdown(note_content)
+else:
+    st.info("Select a subject, unit, and note from the sidebar to view content.")