Update app.py

app.py

@@ -1,7 +1,11 @@
+# Install necessary libraries in Hugging Face Space
+!pip install gradio fpdf pandas
+
 import gradio as gr
 import pandas as pd
 from fpdf import FPDF
 from datetime import datetime
+import tempfile  # Use this for temporary file handling
 import os
 
 # Initialize an empty DataFrame to store all invoice records
@@ -35,52 +39,50 @@ def add_invoice(company_name, name, invoice_no, date, amount):
 # Function to generate and save the filtered invoice data to a PDF
 def generate_invoice_pdf(company_name):
     global invoice_data
-    pdf_file_path = "/content/invoice_report.pdf"  # Define the file path
-
     try:
-        # Create a PDF
-        pdf = FPDF()
-        pdf.add_page()
-        pdf.set_font("Arial", 'B', 16)
-        pdf.cell(200, 10, txt=f"Invoice Report for {company_name}", ln=True, align="C")
-        pdf.set_font("Arial", size=12)
-        current_date = datetime.now().strftime("%d/%m/%Y")
-        pdf.cell(200, 10, txt=f"Report Date: {current_date}", ln=True, align="C")
-        pdf.ln(10)
-
-        # Table headers
-        pdf.cell(50, 10, txt="Company Name", border=1, align="C")
-        pdf.cell(40, 10, txt="Name", border=1, align="C")
-        pdf.cell(40, 10, txt="Invoice No", border=1, align="C")
-        pdf.cell(30, 10, txt="Date", border=1, align="C")
-        pdf.cell(30, 10, txt="Amount", border=1, ln=True, align="C")
-
-        # Filter records by company name
-        filtered_data = invoice_data[invoice_data["Company Name"] == company_name]
-        print(f"Processing {len(filtered_data)} records for {company_name}...")
-
-        for _, row in filtered_data.iterrows():
-            pdf.cell(50, 10, txt=str(row['Company Name']), border=1, align="C")
-            pdf.cell(40, 10, txt=str(row['Name']), border=1, align="C")
-            pdf.cell(40, 10, txt=str(row['Invoice No']), border=1, align="C")
-            pdf.cell(30, 10, txt=str(row['Date']), border=1, align="C")
-            pdf.cell(30, 10, txt=str(row['Amount']), border=1, ln=True, align="C")
-
-        # Total amount
-        total_amount = filtered_data["Amount"].sum()
-        pdf.cell(160, 10, txt="Total Amount", border=1, align="R")
-        pdf.cell(30, 10, txt=str(total_amount), border=1, ln=True, align="C")
-
-        # Save PDF file
-        pdf.output(pdf_file_path)
-
-        # Check if file exists
-        if os.path.exists(pdf_file_path):
-            print(f"PDF successfully created at {pdf_file_path}")
+        # Use a temporary directory to save the file
+        with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as tmp_file:
+            pdf_file_path = tmp_file.name  # Get the temporary file path
+
+            # Create a PDF
+            pdf = FPDF()
+            pdf.add_page()
+            pdf.set_font("Arial", 'B', 16)
+            pdf.cell(200, 10, txt=f"Invoice Report for {company_name}", ln=True, align="C")
+            pdf.set_font("Arial", size=12)
+            current_date = datetime.now().strftime("%d/%m/%Y")
+            pdf.cell(200, 10, txt=f"Report Date: {current_date}", ln=True, align="C")
+            pdf.ln(10)
+
+            # Table headers
+            pdf.cell(50, 10, txt="Company Name", border=1, align="C")
+            pdf.cell(40, 10, txt="Name", border=1, align="C")
+            pdf.cell(40, 10, txt="Invoice No", border=1, align="C")
+            pdf.cell(30, 10, txt="Date", border=1, align="C")
+            pdf.cell(30, 10, txt="Amount", border=1, ln=True, align="C")
+
+            # Filter records by company name
+            filtered_data = invoice_data[invoice_data["Company Name"] == company_name]
+            print(f"Processing {len(filtered_data)} records for {company_name}...")
+
+            for _, row in filtered_data.iterrows():
+                pdf.cell(50, 10, txt=str(row['Company Name']), border=1, align="C")
+                pdf.cell(40, 10, txt=str(row['Name']), border=1, align="C")
+                pdf.cell(40, 10, txt=str(row['Invoice No']), border=1, align="C")
+                pdf.cell(30, 10, txt=str(row['Date']), border=1, align="C")
+                pdf.cell(30, 10, txt=str(row['Amount']), border=1, ln=True, align="C")
+
+            # Total amount
+            total_amount = filtered_data["Amount"].sum()
+            pdf.cell(160, 10, txt="Total Amount", border=1, align="R")
+            pdf.cell(30, 10, txt=str(total_amount), border=1, ln=True, align="C")
+
+            # Save PDF file
+            pdf.output(pdf_file_path)
+
+            # Return the file path for download
             return pdf_file_path
-        else:
-            return "Error: PDF not generated."
-    
+
     except Exception as e:
         print(f"Error during PDF generation: {e}")
         return "Error: PDF generation failed."