Update app.py

app.py

@@ -84,7 +84,7 @@ if uploaded_file is not None:
 
                     # Optionally reset the index if required
                     table_df.reset_index(drop=True, inplace=True)
-                                        
+
                     st.session_state.tables.append(table_df)
 
             st.session_state.processed = True
@@ -95,17 +95,10 @@ default_columns = ['Invoice No', 'Invoice Date', 'Customer Name']
 # Sidebar for table and key-value pair selection
 if st.session_state.processed:
     if not st.session_state.kv_df.empty:
-        # st.sidebar.download_button(
-        #     label="Download Key-Value Pairs as CSV",
-        #     data=st.session_state.kv_df.to_csv(index=False, header=False).encode('utf-8'),
-        #     file_name="kv_df.csv",
-        #     mime='text/csv',
-        # )
-        pass
+        st.sidebar.write("Key-Value Pairs found.")
     else:
         st.sidebar.write("No Key-Value Pairs found in document.")
 
-    
     if st.session_state.tables:
         # Display available tables for selection in the sidebar
         table_options = ["Select"] + [f"Table {i + 1}" for i in range(len(st.session_state.tables))]
@@ -121,7 +114,7 @@ if st.session_state.processed:
             # Add default columns to the left side of the table with NaN values initially
             for col in default_columns:
                 selected_table_df[col] = np.nan
-            
+
             # Move default columns to the left
             selected_table_df = selected_table_df[[*default_columns, *selected_table_df.columns[:-len(default_columns)]]]
 
@@ -157,22 +150,37 @@ if st.session_state.processed:
 
                 st.write(f"Selected Table {selected_table_index + 1} with Key-Value Pairs Added:")
 
-                st.dataframe(selected_table_df)
-                st.download_button(
-                    label=f"Download Table {selected_table_index + 1} as CSV",
-                    data=selected_table_df.to_csv(index=False).encode('utf-8'),
-                    file_name=f"table_with_kv_{selected_table_index + 1}.csv",
-                    mime='text/csv',
+            # st.dataframe(selected_table_df)
+
+            # Allow row and column selection for deletion
+            st.sidebar.subheader("Manage Table")
+            # Delete columns
+            if st.sidebar.checkbox("Delete Columns"):
+                cols_to_delete = st.sidebar.multiselect(
+                    "Select Columns to Delete", list(selected_table_df.columns)
                 )
-            else:
-                st.write(f"Selected Table {selected_table_index + 1}:")
-                st.dataframe(selected_table_df)
-                st.download_button(
-                    label=f"Download Table {selected_table_index + 1} as CSV",
-                    data=selected_table_df.to_csv(index=False).encode('utf-8'),
-                    file_name=f"table{selected_table_index + 1}.csv",
-                    mime='text/csv',
-                ) 
+                if st.sidebar.button("Remove Columns"):
+                    st.session_state.tables[selected_table_index].drop(columns=cols_to_delete, inplace=True)
+
+            # Delete rows
+            if st.sidebar.checkbox("Delete Rows"):
+                row_indices = st.sidebar.multiselect(
+                    "Select Rows to Delete", selected_table_df.index.tolist()
+                )
+                if st.sidebar.button("Remove Rows"):
+                    st.session_state.tables[selected_table_index].drop(index=row_indices, inplace=True)
+
+            # Show the updated table
+            # st.write(f"Updated Table {selected_table_index + 1}:")
+            st.dataframe(selected_table_df)
+
+            # Download updated table
+            st.download_button(
+                label=f"Download Updated Table {selected_table_index + 1} as CSV",
+                data=selected_table_df.to_csv(index=False).encode('utf-8'),
+                file_name=f"updated_table_{selected_table_index + 1}.csv",
+                mime='text/csv',
+            )
     else:
         st.sidebar.write("No tables found in the document.")