Update app.py

app.py

@@ -5,10 +5,9 @@ from numbers_parser import Document
 from openpyxl import Workbook
 import gradio as gr
 import pandas as pd
-from typing import Union
 from pathlib import Path
 
-def numbers_to_xlsx(numbers_file) -> Union[str, tuple[str, bytes]]:
+def numbers_to_xlsx(numbers_file):
     """
     Efficiently converts a Numbers file to XLSX format with optimized memory usage.
 
@@ -16,58 +15,59 @@ def numbers_to_xlsx(numbers_file) -> Union[str, tuple[str, bytes]]:
         numbers_file: The uploaded Numbers file object.
 
     Returns:
-        Union[str, tuple[str, bytes]]: Either error message or tuple of (filename, file_content)
+        str: Path to the converted xlsx file, or error message if conversion fails.
     """
     if not numbers_file:
-        return "Please upload a Numbers file"
+        return None
     
-    # Create a temporary directory with context management
-    with tempfile.TemporaryDirectory() as temp_dir:
-        output_path = os.path.join(temp_dir, "converted.xlsx")
+    try:
+        # Create output directory if it doesn't exist
+        output_dir = "outputs"
+        os.makedirs(output_dir, exist_ok=True)
+        output_path = os.path.join(output_dir, "converted.xlsx")
         
-        try:
-            # Read the Numbers file efficiently
-            doc = Document(numbers_file.name)
-            
-            # Validate document structure
-            if not doc.sheets or not doc.sheets[0].tables:
-                return "Invalid Numbers file: No data tables found"
-            
-            # Get the first table's data efficiently
-            table = doc.sheets[0].tables[0]
-            
-            # Extract headers and data in one pass
-            rows = list(table.rows(values_only=True))
-            if not rows:
-                return "No data found in the table"
-            
-            headers = rows[0]
-            data = rows[1:]
-            
-            # Use pandas optimized DataFrame construction
-            df = pd.DataFrame(data, columns=headers)
-            
-            # Optimize Excel writing with correct options
-            writer = pd.ExcelWriter(
-                output_path,
-                engine='openpyxl'
-            )
-            
-            df.to_excel(
-                writer,
-                index=False,
-                sheet_name='Sheet1'
-            )
-            
-            # Freeze the header row using openpyxl directly
-            writer.sheets['Sheet1'].freeze_panes = 'A2'
-            writer.close()
+        # Read the Numbers file efficiently
+        doc = Document(numbers_file.name)
+        
+        # Validate document structure
+        if not doc.sheets or not doc.sheets[0].tables:
+            return None
+        
+        # Get the first table's data efficiently
+        table = doc.sheets[0].tables[0]
+        
+        # Extract headers and data in one pass
+        rows = list(table.rows(values_only=True))
+        if not rows:
+            return None
+        
+        headers = rows[0]
+        data = rows[1:]
+        
+        # Use pandas optimized DataFrame construction
+        df = pd.DataFrame(data, columns=headers)
+        
+        # Optimize Excel writing with correct options
+        writer = pd.ExcelWriter(
+            output_path,
+            engine='openpyxl'
+        )
+        
+        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 output_path
             
-            # 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)}"
+    except Exception as e:
+        print(f"Error converting file: {str(e)}")
+        return None
 
 # Define the Gradio interface with correct file handling
 interface = gr.Interface(