Update app.py

Add the progress feature

app.py

@@ -8,9 +8,10 @@ load_dotenv()
 data_folder = "data"
 preview_folder = "Preview"
 
-# Extracting text from .txt files
 def get_text_files_content(folder):
     text = ""
+    if not os.path.exists(folder):
+        return text
     for filename in os.listdir(folder):
         if filename.endswith('.txt'):
             with open(os.path.join(folder, filename), 'r', encoding='utf-8') as file:
@@ -29,18 +30,36 @@ def notes_page():
         "Trump vs BRICS", "US-China Trade War", "War on Humanity", "Women's Suppression in Afghanistan"
     ]
 
-    subject_folders = {subject: os.path.join(data_folder, subject.replace(' ', '_')) for subject in subjects}
-    preview_folders = {subject: os.path.join(preview_folder, subject.replace(' ', '_')) for subject in subjects}
-
+    # Map subject name -> folder path
+    subject_folders = {
+        subject: os.path.join(data_folder, subject.replace(' ', '_')) 
+        for subject in subjects
+    }
+    preview_folders = {
+        subject: os.path.join(preview_folder, subject.replace(' ', '_')) 
+        for subject in subjects
+    }
+
+    # Initialize session state variables
     if "completed_subjects" not in st.session_state:
         st.session_state.completed_subjects = []
-
     if "current_subject_index" not in st.session_state:
         st.session_state.current_subject_index = 0
-
     if "grading_enabled" not in st.session_state:
         st.session_state.grading_enabled = False
 
+    # Calculate progress
+    total_subjects = len(subjects)
+    completed_count = len(st.session_state.completed_subjects)
+    progress_fraction = completed_count / total_subjects  # e.g. 0.5 means 50%
+    progress_percentage = progress_fraction * 100
+
+    # --- DISPLAY PROGRESS IN THE SIDEBAR ---
+    st.sidebar.title("Your Progress")
+    st.sidebar.progress(progress_fraction)
+    st.sidebar.write(f"You have completed {completed_count} out of {total_subjects} notes.")
+    st.sidebar.write(f"Overall progress: {progress_percentage:.2f}%")
+
     # Separate completed and unread subjects
     unread_subjects = [sub for sub in subjects if sub not in st.session_state.completed_subjects]
     completed_subjects = st.session_state.completed_subjects
@@ -55,7 +74,7 @@ def notes_page():
     else:
         displayed_subjects = subjects
 
-    # Dropdown to directly jump to specific unread note
+    # If unread are filtered, allow direct jump
     if filter_option == "Unread" and unread_subjects:
         selected_unread_subject = st.sidebar.selectbox("Go to Unread Note:", unread_subjects)
         st.session_state.current_subject_index = subjects.index(selected_unread_subject)
@@ -64,9 +83,9 @@ def notes_page():
     st.sidebar.subheader("All Subjects")
     for i, subject in enumerate(displayed_subjects):
         status = "✓" if subject in st.session_state.completed_subjects else ""
-        st.sidebar.text(f"{i + 1}. {subject} {status}")
+        st.sidebar.text(f"{i+1}. {subject} {status}")
 
-    # Get the current subject
+    # Show current subject
     selected_subject = subjects[st.session_state.current_subject_index]
     st.sidebar.subheader("Current Subject:")
     st.sidebar.text(selected_subject)
@@ -93,6 +112,8 @@ def notes_page():
                 if st.button(f"Mark '{selected_subject}' as Read"):
                     st.session_state.completed_subjects.append(selected_subject)
                     st.success(f"Marked '{selected_subject}' as read!")
+                    # Recalculate progress so it updates immediately
+                    st.experimental_rerun()
 
             # Check if all notes are completed
             if len(st.session_state.completed_subjects) == len(subjects):
@@ -104,6 +125,9 @@ def notes_page():
             if st.session_state.current_subject_index < len(subjects) - 1:
                 if st.button("Go to Next Note"):
                     st.session_state.current_subject_index += 1
+                    st.experimental_rerun()
+        else:
+            st.error(f"No text data found for {selected_subject}.")
     else:
         st.error(f"No data available for {selected_subject}.")
 
@@ -115,4 +139,4 @@ def main():
         notes_page()
 
 if __name__ == "__main__":
-    main()
+    main()