Update app.py

app.py

@@ -9,8 +9,6 @@ import os
 load_dotenv()
 
 # Azure Form Recognizer credentials
-# endpoint = st.secrets["endpoint"]
-# key = st.secrets["key"]
 endpoint = os.environ["endpoint"]
 key = os.environ["key"]
 
@@ -49,8 +47,8 @@ if uploaded_file is not None:
             kv_dict = {}
             for kv_pair in result.key_value_pairs:
                 if kv_pair.key and kv_pair.value:  # Ensure both key and value exist
-                    kv_dict[kv_pair.key.content] = kv_pair.value.content
-
+                    clean_key = kv_pair.key.content.rstrip(':')
+                    kv_dict[clean_key] = kv_pair.value.content
             kv_df = pd.DataFrame(list(kv_dict.items()), columns=["Key", "Value"]).T
 
             # Set the first row as the header
@@ -81,55 +79,54 @@ if uploaded_file is not None:
 
             st.session_state.processed = True
 
-# Display the results if processing is done
+# Sidebar for table and key-value pair selection
 if st.session_state.processed:
     if not st.session_state.kv_df.empty:
-        st.download_button(
+        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',
         )
     else:
-        st.write("No Key-Value Pairs found in document.")
+        st.sidebar.write("No Key-Value Pairs found in document.")
 
     
     if st.session_state.tables:
-        # Display available tables for selection, with "None" as the first option
+        # Display available tables for selection in the sidebar
         table_options = ["Select"] + [f"Table {i + 1}" for i in range(len(st.session_state.tables))]
-        selected_table_option = st.selectbox(
+        selected_table_option = st.sidebar.selectbox(
             "Select table to join key-value pairs:", table_options, index=0
         )
 
-        # Proceed only if a table is selected (i.e., not "None")
+        # Proceed only if a table is selected
         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
+                # Allow the user to select multiple columns using checkboxes
                 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
-                ]
+                selected_columns = []
+                
+                st.sidebar.write("Select key-value pairs to add to the selected table:")
                 
+                # Create checkboxes for each key-value pair
+                for i, option in enumerate(key_value_options):
+                    if st.sidebar.checkbox(option):
+                        selected_columns.append(i)
+
                 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 DataFrame for the selected columns using the original kv_df
-                    selected_columns_df = st.session_state.kv_df.iloc[:, selected_indices].copy()
+                    selected_columns_df = st.session_state.kv_df.iloc[:, selected_columns].copy()
 
                     # Align the length of selected_columns_df with the table_df rows
                     selected_columns_df = selected_columns_df.iloc[:len(selected_table_df)].reset_index(drop=True)
@@ -152,20 +149,19 @@ if st.session_state.processed:
                     mime='text/csv',
                 )
             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=table_df.to_csv(index=False, header=False).encode('utf-8'),
+                data=selected_table_df.to_csv(index=False, header=False).encode('utf-8'),
                 file_name=f"table{selected_table_index + 1}.csv",
                 mime='text/csv',
                 ) 
     else:
-        st.write("No tables found in the document.")
+        st.sidebar.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"):
+if st.session_state.processed and st.sidebar.button("Remove File"):
     st.session_state.processed = False
     st.session_state.tables = []
     st.session_state.kv_df = pd.DataFrame()