Update src/streamlit_app.py

src/streamlit_app.py

@@ -293,19 +293,18 @@ with col2:
     else:
         num_items = st.session_state.num_line_items[selected_name]
         
-        line_col1, line_col2 = st.columns([2, 1])
+        # FLATTENED COLUMNS: no nested columns beyond one level inside col2
+        line_col1, add_col, rem_col = st.columns([2, 1, 1])
         with line_col1:
             line_item_options = [f"Line {i+1}" for i in range(num_items)]
             selected_line_item = st.selectbox("Line", line_item_options, label_visibility="collapsed")
             line_item_num = int(selected_line_item.split()[1])
         
-        with line_col2:
-            add_col, rem_col = st.columns(2)
-            with add_col:
-                st.button("➕", key=f"addli_{selected_name}", on_click=add_line_item, help="Add line item")
-            with rem_col:
-                if st.session_state.num_line_items[selected_name] > 1:
-                    st.button("➖", key=f"remli_{selected_name}", on_click=remove_line_item, help="Remove line item")
+        with add_col:
+            st.button("➕", key=f"addli_{selected_name}", on_click=add_line_item, help="Add line item")
+        with rem_col:
+            if st.session_state.num_line_items[selected_name] > 1:
+                st.button("➖", key=f"remli_{selected_name}", on_click=remove_line_item, help="Remove line item")
         
         base_field = st.selectbox("Field", LINE_ITEM_FIELDS, label_visibility="collapsed")
         display_field_name = f"{selected_line_item}: {base_field}"
@@ -558,4 +557,4 @@ with col1:
                     else:
                         st.toast(f"✅ Rectangle saved (no text detected)")
                 except Exception:
-                    st.toast(f"✅ Rectangle saved")
+                    st.toast(f"✅ Rectangle saved")