Update streamlit_app.py

streamlit_app.py

@@ -25,11 +25,16 @@ except KeyError as e:
 
 @st.cache_resource(show_spinner="Loading Local Knowledge Base...")
 def get_local_kb():
-    return load_knowledge_base()
+
+    local_kn_b = load_knowledge_base()
+
+    if local_kn_b:
+        local_kn_b.load(recreate=False)
+    return local_kn_b
 
 
 local_kb = get_local_kb()
-local_kb.load(recreate=False)  # Load the knowledge base into memory
+#local_kb.load(recreate=False)  # Load the knowledge base into memory
 # --- Core Logic Functions ---
 
 def enrich_and_generate_images(video_summary: str) -> str: