Update src/app.py

src/app.py

@@ -411,7 +411,7 @@ with tab2:
                     summ, usage = query_model_universal(msgs, 1000, model_choice, st.session_state.get("user_openai_key"))
                     st.subheader("Summary"); st.markdown(summ)
         
-        # COLUMN C: Flatten (THE FIX)
+        # COLUMN C: Flatten
         with col_c:
             st.write(""); st.write("")
             if "flattened_result" not in st.session_state: st.session_state.flattened_result = None
@@ -450,12 +450,23 @@ with tab2:
                 st.success("Complete!")
                 st.text_area("Result", res["text"], height=200)
                 if st.button("📥 Index Flat"):
-                    if not st.session_state.active_index: st.error("Select Index.")
+                    if not st.session_state.active_index: 
+                        st.error("Please select an Active Index.")
                     else:
                         with st.spinner("Indexing..."):
-                            ok, msg = rag_engine.process_and_add_text(res["text"], res["source"], st.session_state.username, st.session_state.active_index)
-                            if ok: tracker.upload_user_db(st.session_state.username); st.success(msg)
-                            else: st.error(msg)
+                            # FIX: Pass the active_embed_model here!
+                            ok, msg = rag_engine.process_and_add_text(
+                                text=res["text"], 
+                                source_name=res["source"], 
+                                username=st.session_state.username,
+                                index_name=st.session_state.active_index,
+                                embed_model_name=st.session_state.active_embed_model
+                            )
+                            if ok: 
+                                tracker.upload_user_db(st.session_state.username)
+                                st.success(msg)
+                            else: 
+                                st.error(msg)
                             
     st.divider()
     st.subheader("Database Management")