Update app.py

app.py

@@ -1,15 +1,22 @@
 import gradio as gr
 import pandas as pd
 from fpdf import FPDF
+from datetime import datetime
 
 # Initialize an empty DataFrame to store invoice records
-invoice_data = pd.DataFrame(columns=["Invoice No", "Date", "Amount"])
+invoice_data = pd.DataFrame(columns=["Company Name", "Name", "Invoice No", "Date", "Amount"])
 
 # Function to add new invoice entry and display total amount
-def add_invoice(invoice_no, date, amount):
+def add_invoice(company_name, name, invoice_no, date, amount):
     global invoice_data
     # Add the new record to the DataFrame
-    new_row = {"Invoice No": invoice_no, "Date": date, "Amount": float(amount)}
+    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)
     # Calculate the total amount
     total_amount = invoice_data["Amount"].sum()
@@ -20,26 +27,37 @@ def generate_pdf():
     global invoice_data
     pdf = FPDF()
     pdf.add_page()
-    pdf.set_font("Arial", size=12)
-
-    # Title of the PDF
-    pdf.cell(200, 10, txt="Invoice Records", ln=True, align="C")
+    
+    # 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
+    
+    # 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
 
-    # Table header
-    pdf.cell(60, 10, txt="Invoice No", border=1, align="C")
-    pdf.cell(60, 10, txt="Date", border=1, align="C")
-    pdf.cell(60, 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(60, 10, txt=str(row['Invoice No']), border=1, align="C")
-        pdf.cell(60, 10, txt=str(row['Date']), border=1, align="C")
-        pdf.cell(60, 10, txt=str(row['Amount']), border=1, ln=True, align="C")
+        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(120, 10, txt="Total Amount", border=1, align="R")
-    pdf.cell(60, 10, txt=str(total_amount), border=1, ln=True, align="C")
+    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"
@@ -49,8 +67,10 @@ def generate_pdf():
 
 # Gradio interface
 with gr.Blocks() as app:
-    # Section for adding new invoices
+    # Section for adding new invoices with Company Name and Name fields
     with gr.Row():
+        company_name = gr.Textbox(label="Company Name")
+        name = gr.Textbox(label="Name")
         invoice_no = gr.Textbox(label="Invoice No")
         date = gr.Textbox(label="Date")
         amount = gr.Number(label="Amount")
@@ -67,8 +87,9 @@ with gr.Blocks() as app:
         pdf_output = gr.File(label="Download PDF")
 
     # Define button functionalities
-    add_button.click(add_invoice, [invoice_no, date, amount], [invoice_records, total_amount])
+    add_button.click(add_invoice, [company_name, name, invoice_no, date, amount], [invoice_records, total_amount])
     generate_pdf_button.click(generate_pdf, [], [pdf_output])
 
 # Launch the app
 app.launch(share=True)
+