Update app.py

app.py

@@ -118,34 +118,39 @@ def main():
     if "vectorstore" not in st.session_state:
         st.session_state.vectorstore = None
 
-    # Folder for subject data
-    data_folder = "data"
+    # Root folder for content
+    root_folder = "data"
+    content_types = {
+        "Current Affairs": os.path.join(root_folder, "current_affairs"),
+        "Essays": os.path.join(root_folder, "essays")
+    }
+
+    # Content type selection
+    content_type = st.sidebar.radio("Select Content Type:", list(content_types.keys()))
+    selected_folder = content_types[content_type]
 
     # Subject selection
-    subjects = [
-        "A Trumped World", "Agri Tax in Punjab", "Assad's Fall in Syria", "Elusive National Unity", "Europe and Trump 2.0",
-        "Going Down with Democracy", "Indonesia's Pancasila Philosophy", "Pakistan in Choppy Waters",
-        "Pakistan's Semiconductor Ambitions", "Preserving Pakistan's Cultural Heritage", "Tackling Informal Economy",
-        "Technical Education in Pakistan", "The Case for Solidarity Levies", "The Decline of the Sole Superpower",
-        "The Power of Big Oil", "Trump 2.0 and Pakistan's Emerging Foreign Policy", "Trump and the World 2.0",
-        "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}
+    if os.path.exists(selected_folder):
+        subjects = [f.replace("_", " ").replace(".txt", "") for f in os.listdir(selected_folder) if f.endswith('.txt')]
+    else:
+        subjects = []
+
     selected_subject = st.sidebar.selectbox("Select a Subject:", subjects)
 
     # Process data folder for vectorstore
-    subject_folder_path = subject_folders[selected_subject]
     raw_text = ""
-    if os.path.exists(subject_folder_path):
-        raw_text = get_text_files_content(subject_folder_path)
-        if raw_text:
-            text_chunks = get_chunks(raw_text)
-            vectorstore = get_vectorstore(text_chunks)
-            st.session_state.vectorstore = vectorstore
+    if selected_subject:
+        subject_path = os.path.join(selected_folder, selected_subject.replace(" ", "_") + ".txt")
+        if os.path.exists(subject_path):
+            raw_text = get_text_files_content(selected_folder)
+            if raw_text:
+                text_chunks = get_chunks(raw_text)
+                vectorstore = get_vectorstore(text_chunks)
+                st.session_state.vectorstore = vectorstore
+            else:
+                st.warning("No content found for the selected subject.")
         else:
-            st.warning("No content found for the selected subject.")
-    else:
-        st.error(f"Folder not found for {selected_subject}.")
+            st.error(f"File not found for {selected_subject}.")
 
     # Display preview of notes
     if raw_text: