Update app.py

app.py

@@ -4,6 +4,8 @@ import gradio as gr
 import fitz  # PyMuPDF
 import pandas as pd
 from io import BytesIO
+import tempfile
+import os
 
 def parse_federal_transformers(file_text: str) -> Dict[str, Union[str, List[Dict[str, Union[str, int, float]]]]]:
     parsed_data = {
@@ -50,35 +52,35 @@ def read_pdf(file_path):
     return text
 
 def process_pdf(file):
-    """Handles PDF file input, extracts text, parses it, and returns an Excel file."""
+    """Handles PDF file input, extracts text, parses it, and returns an Excel file path."""
     file_text = read_pdf(file.name)  # Read text from PDF
     parsed_data = parse_federal_transformers(file_text)  # Parse extracted text
 
     # Convert parsed data to DataFrame and create an Excel file
-    output = BytesIO()
-    with pd.ExcelWriter(output, engine='openpyxl') as writer:
-        # Write main PO details to the first sheet
-        main_info = {
-            "Field": ["Purchase Order No", "Date", "Supplier", "Invoice Address", "Delivery Address", "Currency", "Payment Terms"],
-            "Value": [
-                parsed_data["Purchase Order No"],
-                parsed_data["Date"],
-                parsed_data["Supplier"],
-                parsed_data["Invoice Address"],
-                parsed_data["Delivery Address"],
-                parsed_data["Currency"],
-                parsed_data["Payment Terms"]
-            ]
-        }
-        main_df = pd.DataFrame(main_info)
-        main_df.to_excel(writer, index=False, sheet_name="Purchase Order Details")
+    with tempfile.NamedTemporaryFile(delete=False, suffix=".xlsx") as temp_file:
+        with pd.ExcelWriter(temp_file.name, engine='openpyxl') as writer:
+            # Write main PO details to the first sheet
+            main_info = {
+                "Field": ["Purchase Order No", "Date", "Supplier", "Invoice Address", "Delivery Address", "Currency", "Payment Terms"],
+                "Value": [
+                    parsed_data["Purchase Order No"],
+                    parsed_data["Date"],
+                    parsed_data["Supplier"],
+                    parsed_data["Invoice Address"],
+                    parsed_data["Delivery Address"],
+                    parsed_data["Currency"],
+                    parsed_data["Payment Terms"]
+                ]
+            }
+            main_df = pd.DataFrame(main_info)
+            main_df.to_excel(writer, index=False, sheet_name="Purchase Order Details")
 
-        # Write item details to a second sheet
-        items_df = pd.DataFrame(parsed_data["Items"])
-        items_df.to_excel(writer, index=False, sheet_name="Items")
+            # Write item details to a second sheet
+            items_df = pd.DataFrame(parsed_data["Items"])
+            items_df.to_excel(writer, index=False, sheet_name="Items")
 
-    output.seek(0)
-    return output
+    # Return the temporary file path for download
+    return temp_file.name
 
 # Create Gradio interface
 iface = gr.Interface(