Update app.py

app.py

@@ -5,9 +5,10 @@ from numbers_parser import Document
 from openpyxl import Workbook
 import gradio as gr
 import pandas as pd
-from typing import Optional, Union
+from typing import Union
+from pathlib import Path
 
-def numbers_to_xlsx(numbers_file) -> Union[str, bytes]:
+def numbers_to_xlsx(numbers_file) -> Union[str, tuple[str, bytes]]:
     """
     Efficiently converts a Numbers file to XLSX format with optimized memory usage.
 
@@ -15,7 +16,7 @@ def numbers_to_xlsx(numbers_file) -> Union[str, bytes]:
         numbers_file: The uploaded Numbers file object.
 
     Returns:
-        Union[str, bytes]: Path to the converted xlsx file, or error message if conversion fails.
+        Union[str, tuple[str, bytes]]: Either error message or tuple of (filename, file_content)
     """
     if not numbers_file:
         return "Please upload a Numbers file"
@@ -46,28 +47,29 @@ def numbers_to_xlsx(numbers_file) -> Union[str, bytes]:
             # Use pandas optimized DataFrame construction
             df = pd.DataFrame(data, columns=headers)
             
-            # Optimize Excel writing
-            with pd.ExcelWriter(
+            # Optimize Excel writing with correct options
+            writer = pd.ExcelWriter(
                 output_path,
-                engine='openpyxl',
-                mode='w',
-                engine_kwargs={'options': {'strings_to_urls': False}}
-            ) as writer:
-                df.to_excel(
-                    writer,
-                    index=False,
-                    sheet_name='Sheet1',
-                    freeze_panes=(1,0)  # Freeze header row
-                )
+                engine='openpyxl'
+            )
             
-            # Read the file into memory before the temp directory is cleaned up
-            with open(output_path, 'rb') as f:
-                return f.read()
+            df.to_excel(
+                writer,
+                index=False,
+                sheet_name='Sheet1'
+            )
+            
+            # Freeze the header row using openpyxl directly
+            writer.sheets['Sheet1'].freeze_panes = 'A2'
+            writer.close()
+            
+            # Return filename and content as tuple
+            return ("converted.xlsx", Path(output_path).read_bytes())
                 
         except Exception as e:
             return f"Error converting file: {str(e)}"
 
-# Define the Gradio interface
+# Define the Gradio interface with correct file handling
 interface = gr.Interface(
     fn=numbers_to_xlsx,
     inputs=gr.File(label="Numbers File", file_types=[".numbers"]),