Update src/app.py

src/app.py

@@ -290,15 +290,22 @@ with tab2:
         # ACTION BAR
         col_a, col_b, col_c = st.columns(3)
         
-        # 1. ADD TO DB
+        # 1. ADD TO DB (With Strategy Selection)
         with col_a:
+            chunk_strategy = st.selectbox(
+                "Chunking Strategy", 
+                ["paragraph", "token", "page"],
+                help="Paragraph: Standard. Token: Dense text. Page: Forms.",
+                key="chunk_selector"
+            )
+            
             if st.button("📥 Add to Knowledge Base", type="primary"):
                 with st.spinner("Ingesting..."):
                     # Use Admin or User Key for Vision
                     key = st.session_state.get("user_openai_key") or OPENAI_KEY
                     
                     ok, msg = rag_engine.process_and_add_document(
-                        temp_path, st.session_state.username, "paragraph", 
+                        temp_path, st.session_state.username, chunk_strategy, # Pass selected strategy
                         use_vision=use_vision, api_key=key
                     )
                     
@@ -309,6 +316,9 @@ with tab2:
 
         # 2. SUMMARIZE
         with col_b:
+            # Spacer to align buttons visually since col_a has a selectbox
+            st.write("") 
+            st.write("") 
             if st.button("📝 Summarize Document"):
                 with st.spinner("Reading & Summarizing..."):
                     key = st.session_state.get("user_openai_key") or OPENAI_KEY
@@ -338,6 +348,10 @@ with tab2:
 
         # 3. FLATTEN
         with col_c:
+            # Spacer to align buttons
+            st.write("") 
+            st.write("") 
+            
             # We use a session state variable to store the result so it persists for the "Index" step
             if "flattened_result" not in st.session_state:
                 st.session_state.flattened_result = None
@@ -394,9 +408,6 @@ with tab2:
                         if ok:
                             tracker.upload_user_db(st.session_state.username) # Sync!
                             st.success(msg)
-                            # Optional: Clear result after adding
-                            # st.session_state.flattened_result = None 
-                            # st.rerun()
                         else:
                             st.error(msg)