Update app.py

app.py

@@ -9,8 +9,10 @@ import os
 load_dotenv()
 
 # Azure Form Recognizer credentials
-endpoint = st.secrets["endpoint"]
-key = st.secrets["key"]
+# endpoint = st.secrets["endpoint"]
+# key = st.secrets["key"]
+endpoint = os.environ["endpoint"]
+key = os.environ["key"]
 
 # Initialize DocumentAnalysisClient
 document_analysis_client = DocumentAnalysisClient(
@@ -76,7 +78,7 @@ if uploaded_file is not None:
 
                     table_df = pd.DataFrame(data, columns=["row_index", "column_index", "content"])
                     table_df = table_df.pivot(index="row_index", columns="column_index", values="content")
-
+                    
                     st.session_state.tables.append(table_df)
 
             st.session_state.processed = True
@@ -103,38 +105,47 @@ if st.session_state.processed:
             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:")
-
-            # 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]
-            
-            # 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()
-                
-                # 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)
+            for i, table_df in enumerate(st.session_state.tables):
+                st.write(f"Table {i + 1} with Selected Columns Added:")
+
+                # 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]
                 
-                rows_to_add = len(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)
+                # 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()
                     
-                # 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)
-
-            st.download_button(
+                    # 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)
+                    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)
+                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',
-            )
+                )
+        else:
+            for i, table_df in enumerate(st.session_state.tables):
+                st.write(f"Table {i + 1}:")
+                st.dataframe(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{i + 1}.csv",
+                mime='text/csv',
+                )      
     else:
         st.write("No tables found in the document.")