Update app.py

app.py

@@ -111,13 +111,18 @@ if st.session_state.processed:
             selected_table_index = table_options.index(selected_table_option) - 1  # Adjust for "None"
             selected_table_df = st.session_state.tables[selected_table_index]
 
-            # Add default columns to the left side of the table with NaN values initially
+           
+             # 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
+                if col not in selected_table_df.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)]]]
-
+            # We include only those columns that exist in the table
+            existing_columns = [col for col in default_columns if col in selected_table_df.columns]
+            remaining_columns = [col for col in selected_table_df.columns if col not in existing_columns]
+            selected_table_df = selected_table_df[[*existing_columns, *remaining_columns]]
+            
             if not st.session_state.kv_df.empty:
                 # Allow the user to select multiple columns using checkboxes
                 first_row_values = st.session_state.kv_df.iloc[0].values