Update app.py

app.py

@@ -63,33 +63,40 @@ elif page == "Invoices to Table":
 
             if st.button("🧠 Parse Table"):
                 try:
-                    # Step 1: Run parsing only once, store in session_state
-                    if "parsed_table_df" not in st.session_state:
-                        try:
-                            df = extract_markdown_table(table_md)
-                            st.session_state.parsed_table_df = df
-                        except Exception as e:
-                            st.error(f"❌ Table parsing failed: {e}")
-                    # Step 2: Use session state for editing
-                    if "parsed_table_df" in st.session_state:
-                        st.subheader("✏️ Editable Parsed Table")
-                        edited_df = st.data_editor(
-                            st.session_state.parsed_table_df,
-                            num_rows="dynamic",
-                            use_container_width=True,
-                            key="invoice_edit"
-                        )
-                                        
-                    # Download as Excel
-                    excel_buffer = io.BytesIO()
-                    edited_df.to_excel(excel_buffer, index=False)
-                    excel_buffer.seek(0)
-
-                    st.download_button(
-                        label="📥 Download Table as Excel",
-                        data=excel_buffer,
-                        file_name="invoice_table.xlsx",
-                        mime="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"
-                    )
+                    df = extract_markdown_table(table_md)
+            
+                    # ✅ Store it freshly (overwrite old one if it exists)
+                    st.session_state["parsed_table_df"] = df
+                    st.success("✅ Table parsed successfully. You can now edit it.")
                 except Exception as e:
-                    st.error(f"❌ {e}")
+                    st.error(f"❌ Parsing failed: {e}")
+                    
+            if "parsed_table_df" in st.session_state:
+                st.subheader("✏️ Editable Parsed Table")
+            
+                edited_df = st.data_editor(
+                    st.session_state["parsed_table_df"],
+                    num_rows="dynamic",
+                    use_container_width=True,
+                    key="invoice_editor"
+                )
+            
+                # Update session state only after editing
+                st.session_state["parsed_table_df"] = edited_df
+            
+                # Export options
+                import io
+                buffer = io.BytesIO()
+                edited_df.to_excel(buffer, index=False)
+                buffer.seek(0)
+            
+                st.download_button(
+                    label="📥 Download Edited Table as Excel",
+                    data=buffer,
+                    file_name="edited_invoice_table.xlsx",
+                    mime="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"
+                )
+
+            if st.button("🔄 Reset Table"):
+                st.session_state.pop("parsed_table_df", None)
+                st.success("Table has been cleared. You can parse again.")