Update app.py

app.py

@@ -1,8 +1,8 @@
-
 import gradio as gr
 import pandas as pd
 from fpdf import FPDF
 from datetime import datetime
+import os  # To check file existence
 
 # Initialize an empty DataFrame to store all invoice records
 invoice_data = pd.DataFrame(columns=["Company Name", "Name", "Invoice No", "Date", "Amount"])
@@ -30,7 +30,6 @@ def add_invoice(company_name, name, invoice_no, date, amount):
 # Function to generate and save the filtered invoice data to a PDF
 def generate_pdf(company_name):
     global invoice_data
-    
     try:
         pdf = FPDF()
         pdf.add_page()
@@ -71,7 +70,13 @@ def generate_pdf(company_name):
         # Save PDF to file
         pdf_file_path = "/content/invoice_records.pdf"
         pdf.output(pdf_file_path)
-
+        
+        # Check if file exists
+        if os.path.exists(pdf_file_path):
+            print("PDF generated successfully.")
+        else:
+            print("Error: PDF not generated.")
+        
         return pdf_file_path
     
     except Exception as e:
@@ -105,4 +110,3 @@ with gr.Blocks() as app:
 
 # Launch the app
 app.launch(share=True)
-