Update app.py

app.py

@@ -70,8 +70,6 @@ if uploaded_file is not None:
 
             if result.tables:
                 for table in result.tables:
-                    temp_kv_df = kv_df.copy()
-
                     data = []
                     for cell in table.cells:
                         data.append([cell.row_index, cell.column_index, cell.content])
@@ -86,8 +84,6 @@ if uploaded_file is not None:
 # Display the results if processing is done
 if st.session_state.processed:
     if not st.session_state.kv_df.empty:
-        st.write("Extracted Key-Value Pairs:")
-        st.dataframe(st.session_state.kv_df)
         st.download_button(
             label="Download Key-Value Pairs as CSV",
             data=st.session_state.kv_df.to_csv(index=False, header=False).encode('utf-8'),
@@ -99,56 +95,74 @@ if st.session_state.processed:
 
     
     if st.session_state.tables:
-        if not st.session_state.kv_df.empty:
-            # Allow the user to select multiple columns
-            first_row_values = st.session_state.kv_df.iloc[0].values
-            selected_columns = st.multiselect(
-                "Select columns from Key-Value Pairs to add to tables:", first_row_values
-            )
-            for i, table_df in enumerate(st.session_state.tables):
-                st.write(f"Table {i + 1} with Selected Columns Added:")
+        # Display available tables for selection, with "None" as the first option
+        table_options = ["Select"] + [f"Table {i + 1}" for i in range(len(st.session_state.tables))]
+        selected_table_option = st.selectbox(
+            "Select table to join key-value pairs:", table_options, index=0
+        )
 
-                # Find the column names corresponding to the selected first row values
-                column_indices = [list(first_row_values).index(val) for val in selected_columns]
-                column_names = st.session_state.kv_df.columns[column_indices]
+        # Proceed only if a table is selected (i.e., not "None")
+        if selected_table_option != "Select":
+            selected_table_index = table_options.index(selected_table_option) - 1  # Adjust for "None"
+            selected_table_df = st.session_state.tables[selected_table_index]
+
+            if not st.session_state.kv_df.empty:
+                # Allow the user to select multiple columns
+                first_row_values = st.session_state.kv_df.iloc[0].values
+                second_row_values = st.session_state.kv_df.iloc[1].values
+                # Format as "Key (row 0 value): Value (row 1 value)"
+                key_value_options = [
+                    f"{key}: {value}" for key, value in zip(first_row_values, second_row_values)
+                ]
                 
+                selected_columns = st.multiselect(
+                    "Select key-value pairs to add to the selected table:", key_value_options
+                )
+
+                # Extract the column indices of the selected items
+                selected_indices = [
+                    key_value_options.index(selected) for selected in selected_columns
+                ]
+                
+                st.write(f"Selected Table {selected_table_index + 1} with Key-Value Pairs Added:")
+
                 # Add the selected columns to the table
                 if selected_columns:
-                    # Create a new DataFrame for the selected columns
-                    selected_columns_df = st.session_state.kv_df[column_names].copy()
-                    
+                    # Create a DataFrame for the selected columns using the original kv_df
+                    selected_columns_df = st.session_state.kv_df.iloc[:, selected_indices].copy()
+
                     # Align the length of selected_columns_df with the table_df rows
-                    selected_columns_df = selected_columns_df.iloc[:len(table_df)].reset_index(drop=True)
-                    
-                    rows_to_add = len(table_df) - len(selected_columns_df)
+                    selected_columns_df = selected_columns_df.iloc[:len(selected_table_df)].reset_index(drop=True)
+
+                    # Add more rows if needed to match the table_df size
+                    rows_to_add = len(selected_table_df) - len(selected_columns_df)
                     if rows_to_add > 0:
                         last_row = selected_columns_df.iloc[-1]
                         additional_rows = pd.DataFrame([last_row] * rows_to_add, columns=selected_columns_df.columns)
                         selected_columns_df = pd.concat([selected_columns_df, additional_rows], ignore_index=True)
-                        
-                    # Concatenate the selected columns DataFrame with the table DataFrame
-                    table_df = pd.concat([selected_columns_df, table_df.reset_index(drop=True)], axis=1, ignore_index=True)
 
-                st.dataframe(table_df)
+                    # Concatenate the selected columns with the selected table DataFrame
+                    selected_table_df = pd.concat([selected_columns_df, selected_table_df.reset_index(drop=True)], axis=1, ignore_index=True)
+
+                st.dataframe(selected_table_df)
                 st.download_button(
-                label=f"Download Table {i + 1} as CSV",
-                data=table_df.to_csv(index=False, header=False).encode('utf-8'),
-                file_name=f"table_with_kv_{i + 1}.csv",
-                mime='text/csv',
+                    label=f"Download Table {selected_table_index + 1} as CSV",
+                    data=selected_table_df.to_csv(index=False, header=False).encode('utf-8'),
+                    file_name=f"table_with_kv_{selected_table_index + 1}.csv",
+                    mime='text/csv',
                 )
-        else:
-            for i, table_df in enumerate(st.session_state.tables):
-                st.write(f"Table {i + 1}:")
-                st.dataframe(table_df)
+            else:
+               
+                st.write(f"Selected Table {selected_table_index + 1}:")
+                st.dataframe(selected_table_df)
                 st.download_button(
-                label=f"Download Table {i + 1} as CSV",
+                label=f"Download Table {selected_table_index + 1} as CSV",
                 data=table_df.to_csv(index=False, header=False).encode('utf-8'),
-                file_name=f"table{i + 1}.csv",
+                file_name=f"table{selected_table_index + 1}.csv",
                 mime='text/csv',
-                )      
+                ) 
     else:
         st.write("No tables found in the document.")
-        
 
 # Option to remove the file and clear the session
 if st.session_state.processed and st.button("Remove File"):