Update app.py

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 import pandas as pd
 from fpdf import FPDF
 from datetime import datetime
-import os  # To check file existence
+import os
 
 # Initialize an empty DataFrame to store all invoice records
 invoice_data = pd.DataFrame(columns=["Company Name", "Name", "Invoice No", "Date", "Amount"])
@@ -10,31 +10,35 @@ invoice_data = pd.DataFrame(columns=["Company Name", "Name", "Invoice No", "Date
 # Function to add new invoice entry and display records by company name
 def add_invoice(company_name, name, invoice_no, date, amount):
     global invoice_data
-    # Add the new record to the DataFrame
-    new_row = {
-        "Company Name": company_name,
-        "Name": name,
-        "Invoice No": invoice_no,
-        "Date": date,
-        "Amount": float(amount)
-    }
-    invoice_data = pd.concat([invoice_data, pd.DataFrame([new_row])], ignore_index=True)
-
-    # Filter records by company name
-    filtered_data = invoice_data[invoice_data["Company Name"] == company_name]
-    
-    # Calculate the total amount for that company
-    total_amount = filtered_data["Amount"].sum()
-    return filtered_data, f"Total Amount for {company_name}: {total_amount}"
+    try:
+        # Add the new record to the DataFrame
+        new_row = {
+            "Company Name": company_name,
+            "Name": name,
+            "Invoice No": invoice_no,
+            "Date": date,
+            "Amount": float(amount)
+        }
+        invoice_data = pd.concat([invoice_data, pd.DataFrame([new_row])], ignore_index=True)
+
+        # Filter records by company name
+        filtered_data = invoice_data[invoice_data["Company Name"] == company_name]
+
+        # Calculate the total amount for that company
+        total_amount = filtered_data["Amount"].sum()
+        return filtered_data, f"Total Amount for {company_name}: {total_amount}"
+
+    except Exception as e:
+        print(f"Error during adding invoice: {e}")
+        return pd.DataFrame(), "Error during adding invoice"
 
 # Function to generate and save the filtered invoice data to a PDF
-def generate_pdf(company_name):
+def generate_invoice_pdf(company_name):
     global invoice_data
-    pdf_file_path = "/content/invoice_records.pdf"  # Define the file path
+    pdf_file_path = "/content/invoice_report.pdf"  # Define the file path
 
     try:
-        print(f"Generating PDF for company: {company_name}")
-        # Create PDF
+        # Create a PDF
         pdf = FPDF()
         pdf.add_page()
         pdf.set_font("Arial", 'B', 16)
@@ -44,8 +48,6 @@ 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")
@@ -53,11 +55,11 @@ def generate_pdf(company_name):
         pdf.cell(30, 10, txt="Date", border=1, align="C")
         pdf.cell(30, 10, txt="Amount", border=1, ln=True, align="C")
 
-        # Add filtered data for the company
+        # Filter records by company name
         filtered_data = invoice_data[invoice_data["Company Name"] == company_name]
-        print(f"Number of rows for {company_name}: {len(filtered_data)}")
+        print(f"Processing {len(filtered_data)} records for {company_name}...")
 
-        for index, row in filtered_data.iterrows():
+        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")
@@ -69,8 +71,6 @@ 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)
 
@@ -79,7 +79,6 @@ 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:
@@ -111,7 +110,7 @@ with gr.Blocks() as app:
     add_button.click(add_invoice, [company_name, name, invoice_no, date, amount], [invoice_records, total_amount])
     
     def show_pdf_link(company_name):
-        pdf_path = generate_pdf(company_name)
+        pdf_path = generate_invoice_pdf(company_name)
         if pdf_path != "Error: PDF generation failed.":
             return f'<a href="{pdf_path}" target="_blank">Download PDF</a>'
         else: