Update app.py

app.py

@@ -118,45 +118,52 @@ def main():
     if "vectorstore" not in st.session_state:
         st.session_state.vectorstore = None
 
-    # 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")
-    }
+    # Define folders for Current Affairs and Essays
+    data_folder = "data"
+    essay_folder = "essay"
 
     # Content type selection
-    content_type = st.sidebar.radio("Select Content Type:", list(content_types.keys()))
-    selected_folder = content_types[content_type]
+    content_type = st.sidebar.radio("Select Content Type:", ["Current Affairs", "Essays"])
 
-    # Subject selection
-    if os.path.exists(selected_folder):
-        subjects = [f.replace("_", " ").replace(".txt", "") for f in os.listdir(selected_folder) if f.endswith('.txt')]
-    else:
-        subjects = []
+    # Handle Current Affairs (each subject has its own folder)
+    if content_type == "Current Affairs":
+        if os.path.exists(data_folder):
+            subjects = [f for f in os.listdir(data_folder) if os.path.isdir(os.path.join(data_folder, f))]
+        else:
+            subjects = []
+    # Handle Essays (all essays are in a single folder)
+    elif content_type == "Essays":
+        if os.path.exists(essay_folder):
+            subjects = [f.replace(".txt", "") for f in os.listdir(essay_folder) if f.endswith('.txt')]
+        else:
+            subjects = []
 
+    # Subject selection
     selected_subject = st.sidebar.selectbox("Select a Subject:", subjects)
 
-    # Process data folder for vectorstore
+    # Process selected subject
     raw_text = ""
-    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.error(f"File not found for {selected_subject}.")
+    if content_type == "Current Affairs" and selected_subject:
+        subject_folder = os.path.join(data_folder, selected_subject)
+        raw_text = get_text_files_content(subject_folder)
+    elif content_type == "Essays" and selected_subject:
+        subject_file = os.path.join(essay_folder, selected_subject + ".txt")
+        if os.path.exists(subject_file):
+            with open(subject_file, "r", encoding="utf-8") as file:
+                raw_text = file.read()
 
     # Display preview of notes
     if raw_text:
         st.subheader("Preview of Notes")
         st.text_area("Preview Content:", value=raw_text[:2000], height=300, disabled=True)  # Show a snippet of the notes
 
+        # Create vectorstore for Current Affairs or Essays
+        text_chunks = get_chunks(raw_text)
+        vectorstore = get_vectorstore(text_chunks)
+        st.session_state.vectorstore = vectorstore
+    else:
+        st.warning("No content available for the selected subject.")
+
     # Chat interface
     st.subheader("Ask Your Question")
     question = st.text_input("Ask a question about your selected subject:")