Update app.py

app.py

@@ -3,11 +3,12 @@ import pandas as pd
 import io
 import tempfile
 import re
+import os
 from openpyxl import load_workbook
 from openpyxl.styles import Font, PatternFill
 
 # Core processing logic
-def process_excel(data_bytes):
+def process_excel(data_bytes, original_filename):
     buffer = io.BytesIO(data_bytes)
     xls = pd.ExcelFile(buffer)
     df = xls.parse(xls.sheet_names[0], header=None, dtype=str)
@@ -133,16 +134,18 @@ def process_excel(data_bytes):
     # Extract shareholders
     shareholders = add_company_name_row(df.iloc[b["shareholders"]:b["shareholders_history"]])
 
-    # Write to in-memory Excel
-    out = io.BytesIO()
-    with pd.ExcelWriter(out, engine='openpyxl') as writer:
+    # Save Excel file with same base name + '_formatted.xlsx'
+    base = os.path.splitext(os.path.basename(original_filename))[0]
+    filename = base + "_formatted.xlsx"
+    tmp_path = os.path.join(tempfile.gettempdir(), filename)
+
+    with pd.ExcelWriter(tmp_path, engine='openpyxl') as writer:
         balance_sheet.to_excel(writer, sheet_name="balance_sheet", index=False, header=False)
         cash_flow.to_excel(writer, sheet_name="cash_flow", index=False, header=False)
         shareholders.to_excel(writer, sheet_name="shareholders", index=False, header=False)
-    out.seek(0)
 
-    # Apply formatting
-    wb = load_workbook(out)
+    # Format with openpyxl
+    wb = load_workbook(tmp_path)
     fill = PatternFill(start_color='C5D9F1', end_color='C5D9F1', fill_type='solid')
     bold = Font(bold=True)
 
@@ -158,14 +161,10 @@ def process_excel(data_bytes):
     if 'cash_flow' in wb.sheetnames:
         ws = wb['cash_flow']
         highlight_labels = {
-            "cash flow from operations:",
-            "'= net cash from operations",
-            "cash flow from investing:",
-            "'= net cash from investing",
-            "cash flow from financing:",
-            "'= net cash from financing",
-            "net increase/(decrease) in cash",
-            "'= cash at end of period"
+            "cash flow from operations:", "'= net cash from operations",
+            "cash flow from investing:", "'= net cash from investing",
+            "cash flow from financing:", "'= net cash from financing",
+            "net increase/(decrease) in cash", "'= cash at end of period"
         }
         for row in ws.iter_rows():
             first_cell = row[0]
@@ -196,23 +195,18 @@ def process_excel(data_bytes):
                         cell.fill = fill
                         cell.font = bold
 
-    final = io.BytesIO()
-    wb.save(final)
-    final.seek(0)
-    return final.getvalue()
+    wb.save(tmp_path)
+    return tmp_path
 
 # Gradio wrapper
 def extract_and_download(uploaded_file):
+    filename = uploaded_file.name if hasattr(uploaded_file, "name") else "output"
     if hasattr(uploaded_file, "read"):
         data = uploaded_file.read()
     else:
         with open(uploaded_file, "rb") as f:
             data = f.read()
-    out_bytes = process_excel(data)
-    tmp_path = tempfile.mktemp(suffix=".xlsx")
-    with open(tmp_path, "wb") as tmp:
-        tmp.write(out_bytes)
-    return tmp_path
+    return process_excel(data, filename)
 
 iface = gr.Interface(
     fn=extract_and_download,