donut_UI

Sleeping

App Files Files Community

Bhuvi13 commited on Sep 22, 2025

Commit

12f16ab

verified ·

1 Parent(s): 5b3d6b6

Update src/streamlit_app.py

Browse files

Files changed (1) hide show

src/streamlit_app.py +64 -59

src/streamlit_app.py CHANGED Viewed

@@ -41,7 +41,7 @@ def trigger_rerun():
 # UI: main
 # ---------------------------
 st.set_page_config(page_title="Invoice Extractor (Donut) - Batch Mode", layout="wide")
-st.title("Batch Invoice Extraction")
 # Reduce top margin and tighten layout
 st.markdown(
@@ -603,6 +603,69 @@ if not st.session_state.is_processing_batch and len(st.session_state.batch_resul
 # RESULTS VIEW — Show selector + editable form
 # ---------------------------
 elif len(st.session_state.batch_results) > 0:
     # File selector
     file_options = {
         f"{v['file_name']} ({k[:6]})": k
@@ -838,65 +901,7 @@ elif len(st.session_state.batch_results) > 0:
                 key=f"dl_csv_{selected_hash}"
             )
 # Global Download All — produce a single Excel file (concatenated rows) and trigger direct download
-if st.button("📦 Download All Results (Excel)", key="download_all"):
-        # Collect rows from all invoices and concatenate into one DataFrame
-        all_rows = []
-        for file_hash, result in st.session_state.batch_results.items():
-            rows = flatten_invoice_to_rows(result["edited_data"])
-            # Annotate rows with source file name so user can identify which invoice each row came from
-            for r in rows:
-                r["Source File"] = result.get("file_name", file_hash)
-            all_rows.extend(rows)
-        if len(all_rows) == 0:
-            st.warning("No invoice data available to download.")
-        else:
-            full_df = pd.DataFrame(all_rows)
-            # Reorder columns to put Source File first
-            cols = list(full_df.columns)
-            if "Source File" in cols:
-                cols = ["Source File"] + [c for c in cols if c != "Source File"]
-            full_df = full_df[cols]
-            # Try to write XLSX (preferred). If engine not available, fall back to CSV.
-            buffer = BytesIO()
-            dl_filename = "all_extracted_invoices.xlsx"
-            tried_xlsx = False
-            try:
-                with pd.ExcelWriter(buffer, engine="openpyxl") as writer:
-                    full_df.to_excel(writer, index=False, sheet_name="Invoices")
-                tried_xlsx = True
-                buffer.seek(0)
-                file_bytes = buffer.read()
-                mime = "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"
-            except Exception:
-                # Fallback to CSV
-                buffer = BytesIO()
-                csv_data = full_df.to_csv(index=False).encode("utf-8")
-                buffer.write(csv_data)
-                buffer.seek(0)
-                file_bytes = buffer.read()
-                dl_filename = "all_extracted_invoices.csv"
-                mime = "text/csv"
-            # Trigger immediate download via a data URI and small HTML snippet
-            import base64
-            import streamlit.components.v1 as components
-            b64 = base64.b64encode(file_bytes).decode()
-            data_uri = f"data:{mime};base64,{b64}"
-            auto_dl_html = f'''<html>
-                <body>
-                    <a id="dlLink" href="{data_uri}" download="{dl_filename}"></a>
-                    <script>
-                        const a = document.getElementById('dlLink');
-                        a.click();
-                    </script>
-                </body>
-            </html>'''
-            components.html(auto_dl_html, height=0)
 # ---------------------------
 # PROCESSING STATE

 # UI: main
 # ---------------------------
 st.set_page_config(page_title="Invoice Extractor (Donut) - Batch Mode", layout="wide")
+st.title("Invoice Extraction")
 # Reduce top margin and tighten layout
 st.markdown(
 # RESULTS VIEW — Show selector + editable form
 # ---------------------------
 elif len(st.session_state.batch_results) > 0:
+            # ---------------------------
+        # Global Download All — produce a single Excel file (concatenated rows) and trigger direct download
+        # ---------------------------
+    if st.button("📦 Download All Results (Excel)", key="download_all"):
+            # Collect rows from all invoices and concatenate into one DataFrame
+        all_rows = []
+        for file_hash, result in st.session_state.batch_results.items():
+            rows = flatten_invoice_to_rows(result["edited_data"])
+                # Annotate rows with source file name so user can identify which invoice each row came from
+            for r in rows:
+                r["Source File"] = result.get("file_name", file_hash)
+            all_rows.extend(rows)
+        if len(all_rows) == 0:
+            st.warning("No invoice data available to download.")
+        else:
+            full_df = pd.DataFrame(all_rows)
+                # Reorder columns to put Source File first
+            cols = list(full_df.columns)
+            if "Source File" in cols:
+                cols = ["Source File"] + [c for c in cols if c != "Source File"]
+            full_df = full_df[cols]
+                # Try to write XLSX (preferred). If engine not available, fall back to CSV.
+            buffer = BytesIO()
+            dl_filename = "all_extracted_invoices.xlsx"
+            tried_xlsx = False
+            try:
+                with pd.ExcelWriter(buffer, engine="openpyxl") as writer:
+                    full_df.to_excel(writer, index=False, sheet_name="Invoices")
+                tried_xlsx = True
+                buffer.seek(0)
+                file_bytes = buffer.read()
+                mime = "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"
+            except Exception:
+                # Fallback to CSV
+                buffer = BytesIO()
+                csv_data = full_df.to_csv(index=False).encode("utf-8")
+                buffer.write(csv_data)
+                buffer.seek(0)
+                file_bytes = buffer.read()
+                dl_filename = "all_extracted_invoices.csv"
+                mime = "text/csv"
+                # Trigger immediate download via a data URI and small HTML snippet
+            import base64
+            import streamlit.components.v1 as components
+            b64 = base64.b64encode(file_bytes).decode()
+            data_uri = f"data:{mime};base64,{b64}"
+            auto_dl_html = f'''<html>
+                <body>
+                    <a id="dlLink" href="{data_uri}" download="{dl_filename}"></a>
+                    <script>
+                        const a = document.getElementById('dlLink');
+                        a.click();
+                    </script>
+                </body>
+            </html>'''
+            components.html(auto_dl_html, height=0)
     # File selector
     file_options = {
         f"{v['file_name']} ({k[:6]})": k
                 key=f"dl_csv_{selected_hash}"
             )
 # Global Download All — produce a single Excel file (concatenated rows) and trigger direct download
 # ---------------------------
 # PROCESSING STATE