Update app.py

app.py

@@ -112,7 +112,7 @@ def handle_question(question, vectorstore=None):
     return llm_chain.invoke({"instruction": question})
 
 def main():
-    st.title("Ask AI :books:")
+    st.title("Chat with Notes :books:")
 
     # Initialize session state
     if "vectorstore" not in st.session_state:
@@ -135,19 +135,31 @@ def main():
 
     # 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
+        else:
+            st.warning("No content found for the selected subject.")
+    else:
+        st.error(f"Folder not found for {selected_subject}.")
+
+    # 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
 
     # Chat interface
+    st.subheader("Ask Your Question")
     question = st.text_input("Ask a question about your selected subject:")
     if question:
         if st.session_state.vectorstore:
             response = handle_question(question, st.session_state.vectorstore)
-            st.write(response)
+            st.subheader("Answer:")
+            st.write(response.get("text", "No response found."))
         else:
             st.warning("Please load the content for the selected subject before asking a question.")