Update app.py

app.py

@@ -1,12 +1,13 @@
+
 import gradio as gr
 import pandas as pd
 from fpdf import FPDF
 from datetime import datetime
 
-# Initialize an empty DataFrame to store invoice records
+# Initialize an empty DataFrame to store all invoice records
 invoice_data = pd.DataFrame(columns=["Company Name", "Name", "Invoice No", "Date", "Amount"])
 
-# Function to add new invoice entry and display total amount
+# 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
@@ -18,52 +19,64 @@ def add_invoice(company_name, name, invoice_no, date, amount):
         "Amount": float(amount)
     }
     invoice_data = pd.concat([invoice_data, pd.DataFrame([new_row])], ignore_index=True)
-    # Calculate the total amount
-    total_amount = invoice_data["Amount"].sum()
-    return invoice_data, f"Total Amount: {total_amount}"
 
-# Function to generate and save the invoice data to a PDF
-def generate_pdf():
-    global invoice_data
-    pdf = FPDF()
-    pdf.add_page()
+    # Filter records by company name
+    filtered_data = invoice_data[invoice_data["Company Name"] == company_name]
     
-    # Add header
-    pdf.set_font("Arial", 'B', 16)  # Bold and larger font for header
-    pdf.cell(200, 10, txt="Invoice Report", ln=True, align="C")  # Report title
-    pdf.set_font("Arial", size=12)  # Reset to normal font size
+    # Calculate the total amount for that company
+    total_amount = filtered_data["Amount"].sum()
+    return filtered_data, f"Total Amount for {company_name}: {total_amount}"
+
+# Function to generate and save the filtered invoice data to a PDF
+def generate_pdf(company_name):
+    global invoice_data
     
-    # Dynamic date in header
-    current_date = datetime.now().strftime("%d/%m/%Y")
-    pdf.cell(200, 10, txt="Generated Report", ln=True, align="C")
-    pdf.cell(200, 10, txt=f"Report Date: {current_date}", ln=True, align="C")
-    pdf.ln(10)  # Add some space before the table starts
+    try:
+        pdf = FPDF()
+        pdf.add_page()
+        
+        # Add header
+        pdf.set_font("Arial", 'B', 16)  # Bold and larger font for header
+        pdf.cell(200, 10, txt=f"Invoice Report for {company_name}", ln=True, align="C")  # Report title
+        pdf.set_font("Arial", size=12)  # Reset to normal font size
+        
+        # Dynamic date in header
+        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)  # Add some space before the table starts
 
-    # Table header with Company Name and Name
-    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")
+        # Table header with Company Name and Name
+        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")
 
-    # Add invoice data to PDF
-    for index, row in invoice_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")
-    
-    # Add total amount at the end
-    total_amount = invoice_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")
+        # Filter records by company name
+        filtered_data = invoice_data[invoice_data["Company Name"] == company_name]
+
+        # Add filtered invoice data to PDF
+        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")
+            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")
+        
+        # Add total amount at the end for the company
+        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 to file
-    pdf_file_path = "/content/invoice_records.pdf"
-    pdf.output(pdf_file_path)
+        # Save PDF to file
+        pdf_file_path = "/content/invoice_records.pdf"
+        pdf.output(pdf_file_path)
 
-    return pdf_file_path
+        return pdf_file_path
+    
+    except Exception as e:
+        print(f"Error generating PDF: {e}")
+        return None
 
 # Gradio interface
 with gr.Blocks() as app:
@@ -88,7 +101,7 @@ with gr.Blocks() as app:
 
     # Define button functionalities
     add_button.click(add_invoice, [company_name, name, invoice_no, date, amount], [invoice_records, total_amount])
-    generate_pdf_button.click(generate_pdf, [], [pdf_output])
+    generate_pdf_button.click(generate_pdf, [company_name], [pdf_output])
 
 # Launch the app
 app.launch(share=True)