Update app.py

app.py

@@ -1,41 +1,31 @@
-import os
 import streamlit as st
 from knowledge_engine import KnowledgeManager
 
-st.set_page_config(
-    page_title="Sirraya xBrain - LangChain QA Assistant",
-    layout="centered",
-    page_icon="🧠",
-    initial_sidebar_state="expanded",
-)
+st.set_page_config(page_title="LangChain QA Assistant", layout="centered")
 
-st.title("🧠 Sirraya xBrain — Intelligent QA Assistant")
-
-if "km" not in st.session_state:
-    with st.spinner("Initializing knowledge base and LLM..."):
+def initialize_knowledge_manager():
+    if "km" not in st.session_state:
         try:
             st.session_state.km = KnowledgeManager()
-            st.success("✅ Knowledge engine initialized successfully!")
+            st.success("Knowledge base loaded successfully!")
             st.info(st.session_state.km.get_knowledge_summary())
         except Exception as e:
-            st.error(f"❌ Failed to initialize system: {e}")
+            st.error(f"Error loading knowledge base: {e}")
             st.session_state.km = None
 
-if st.session_state.km is None:
-    st.warning("Knowledge base not loaded or failed to initialize.")
-    st.stop()
-
-question = st.text_input("Ask a question about the knowledge base:", "")
+def main():
+    st.title("LangChain QA Assistant")
 
-if question:
-    with st.spinner("Generating answer..."):
-        answer, sources = st.session_state.km.query(question)
+    initialize_knowledge_manager()
 
-        st.markdown("### Answer:")
-        st.write(answer)
+    if st.session_state.km:
+        query = st.text_input("Ask something about your knowledge base:")
+        if query:
+            with st.spinner("Getting answer..."):
+                answer = st.session_state.km.ask(query)
+            st.markdown(f"**Answer:** {answer}")
+    else:
+        st.info("Please upload your `.txt` files in the root of the repo.")
 
-        if sources:
-            with st.expander("📚 Source documents"):
-                for i, src in enumerate(sources, 1):
-                    st.write(f"Source {i}:")
-                    st.write(src)
+if __name__ == "__main__":
+    main()