Update app.py

app.py

@@ -33,6 +33,7 @@ def generate_pdf(company_name):
     pdf_file_path = "/content/invoice_records.pdf"  # Define the file path
 
     try:
+        print(f"Generating PDF for company: {company_name}")
         # Create PDF
         pdf = FPDF()
         pdf.add_page()
@@ -43,6 +44,8 @@ def generate_pdf(company_name):
         pdf.cell(200, 10, txt=f"Report Date: {current_date}", ln=True, align="C")
         pdf.ln(10)
 
+        # Log the generation process
+        print("Adding table headers to the PDF")
         # Table headers
         pdf.cell(50, 10, txt="Company Name", border=1, align="C")
         pdf.cell(40, 10, txt="Name", border=1, align="C")
@@ -52,6 +55,8 @@ def generate_pdf(company_name):
 
         # Add filtered data for the company
         filtered_data = invoice_data[invoice_data["Company Name"] == company_name]
+        print(f"Number of rows for {company_name}: {len(filtered_data)}")
+
         for index, 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")
@@ -64,6 +69,8 @@ def generate_pdf(company_name):
         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")
 
+        # Log that we're saving the PDF
+        print(f"Saving PDF to {pdf_file_path}")
         # Save PDF file
         pdf.output(pdf_file_path)
 
@@ -72,6 +79,7 @@ def generate_pdf(company_name):
             print(f"PDF successfully created at {pdf_file_path}")
             return pdf_file_path
         else:
+            print("Error: PDF file not found.")
             return "Error: PDF not generated."
     
     except Exception as e:
@@ -114,4 +122,3 @@ with gr.Blocks() as app:
 # Launch the app
 app.launch(share=True)
 
-