Update app.py

app.py

@@ -1,6 +1,5 @@
 import pdfplumber
 import pandas as pd
-from io import BytesIO
 import re
 import gradio as gr
 
@@ -8,11 +7,8 @@ def extract_data_from_pdf(pdf_file):
     data = []
     po_number = None
 
-    # Save BytesIO to temporary file
-    with open("temp.pdf", "wb") as f:
-        f.write(pdf_file.getbuffer())
-
-    with pdfplumber.open("temp.pdf") as pdf:
+    # Open PDF file directly
+    with pdfplumber.open(pdf_file.name) as pdf:
         for page in pdf.pages:
             text = page.extract_text()
 
@@ -57,16 +53,8 @@ def extract_data_from_pdf(pdf_file):
 
     # Convert data to DataFrame and save to Excel
     df = pd.DataFrame(data)
-    output = BytesIO()
-    with pd.ExcelWriter(output, engine="xlsxwriter") as writer:
-        df.to_excel(writer, index=False, sheet_name="Extracted Data")
-    output.seek(0)
-
-    # Remove temporary PDF file
-    import os
-    os.remove("temp.pdf")
-
-    return output
+    output = df.to_excel("output.xlsx", index=False)
+    return "output.xlsx"
 
 # Gradio Interface
 iface = gr.Interface(