Update app.py

app.py

@@ -1,5 +1,11 @@
 # ------------------- MUST BE FIRST -------------------
 import streamlit as st
+from pathlib import Path
+
+# Create folder if it doesn't exist
+KNOWLEDGE_DIR = Path("knowledge_base")
+KNOWLEDGE_DIR.mkdir(parents=True, exist_ok=True)
+
 st.set_page_config(
     page_title="Sirraya xBrain - Intelligent Assistant", 
     layout="centered",
@@ -40,12 +46,15 @@ def render_sidebar():
         
         if uploaded_file:
             if st.session_state.lisa:
-                success = st.session_state.lisa.save_uploaded_file(uploaded_file, uploaded_file.name)
-                if success:
-                    st.success(f"✅ Saved {uploaded_file.name}")
+                save_path = KNOWLEDGE_DIR / uploaded_file.name
+                try:
+                    # Save the uploaded file into knowledge_base folder
+                    with open(save_path, "wb") as f:
+                        f.write(uploaded_file.getbuffer())
+                    st.success(f"✅ Saved {uploaded_file.name} to knowledge_base folder")
                     st.info("💡 Click 'Rebuild Knowledge Base' to update the index")
-                else:
-                    st.error(f"❌ Error saving {uploaded_file.name}")
+                except Exception as e:
+                    st.error(f"❌ Error saving {uploaded_file.name}: {e}")
             else:
                 st.error("❌ Knowledge engine not initialized")
         
@@ -146,4 +155,4 @@ def main():
 
 
 if __name__ == "__main__":
-    main()
+    main()