Update src/app.py

src/app.py

@@ -204,10 +204,10 @@ with st.sidebar:
         # 2. SAFETY CHECK VISUAL
         if selected_index:
             # Get current model dimension dynamically
-            # (We import get_embedding_func to check the active model)
+            current_model = st.session_state.get("active_embed_model", "sentence-transformers/all-MiniLM-L6-v2")
             try:
-                emb_fn = rag_engine.get_embedding_func()
-                test_vec = emb_fn.embed_query("test")
+                emb_fn = rag_engine.get_embedding_func(current_model)
+                test_vec = emb_fn.embed_query("this is a test")
                 active_model_dim = len(test_vec)
                 
                 is_compatible = pm.check_dimension_compatibility(selected_index, active_model_dim)