Update app.py

app.py

@@ -5,37 +5,54 @@ from datetime import datetime
 import tempfile  # Use for temporary file handling
 import os
 
-# Initialize an empty DataFrame to store all invoice records
-invoice_data = pd.DataFrame(columns=["Company Name", "Name", "Invoice No", "Date", "Amount"])
+# Initialize an empty DataFrame to store all records of invoices (acting as our ledger)
+ledger = pd.DataFrame(columns=["Head", "Company Name", "SP/Station", "Invoice No", "Date", "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
+# Function to add a new invoice under a specific head
+def add_invoice(head, company_name, sp_station, invoice_no, date, amount):
+    global ledger
     try:
-        # Add the new record to the DataFrame
+        # Add the new record to the ledger
         new_row = {
+            "Head": head,
             "Company Name": company_name,
-            "Name": name,
+            "SP/Station": sp_station,
             "Invoice No": invoice_no,
             "Date": date,
             "Amount": float(amount)
         }
-        invoice_data = pd.concat([invoice_data, pd.DataFrame([new_row])], ignore_index=True)
+        ledger = pd.concat([ledger, pd.DataFrame([new_row])], ignore_index=True)
 
-        # Filter records by company name
-        filtered_data = invoice_data[invoice_data["Company Name"] == company_name]
+        # Filter the ledger for the selected head
+        filtered_ledger = ledger[ledger["Head"] == head]
 
-        # Calculate the total amount for that company
-        total_amount = filtered_data["Amount"].sum()
-        return filtered_data, f"Total Amount for {company_name}: {total_amount}"
+        # Calculate the total for the selected head
+        total_amount = filtered_ledger["Amount"].sum()
+        return filtered_ledger, f"Total for {head}: {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_invoice_pdf(company_name):
-    global invoice_data
+# Function to search and display all previous records by head
+def search_by_head(head):
+    global ledger
+    try:
+        # Filter the ledger for the selected head
+        filtered_ledger = ledger[ledger["Head"] == head]
+
+        # Calculate the total for the selected head
+        total_amount = filtered_ledger["Amount"].sum()
+
+        return filtered_ledger, f"Total for {head}: {total_amount}"
+
+    except Exception as e:
+        print(f"Error during search by head: {e}")
+        return pd.DataFrame(), "Error during search by head"
+
+# Function to generate a PDF report for a specific head
+def generate_ledger_pdf(head):
+    global ledger
     try:
         # Use a temporary directory to save the file
         with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as tmp_file:
@@ -45,33 +62,33 @@ def generate_invoice_pdf(company_name):
             pdf = FPDF()
             pdf.add_page()
             pdf.set_font("Arial", 'B', 16)
-            pdf.cell(200, 10, txt=f"Invoice Report for {company_name}", ln=True, align="C")
+            pdf.cell(200, 10, txt=f"Ledger Report for {head}", ln=True, align="C")
             pdf.set_font("Arial", size=12)
             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)
 
             # Table headers
-            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(40, 10, txt="Company Name", border=1, align="C")
+            pdf.cell(40, 10, txt="SP/Station", border=1, align="C")
+            pdf.cell(30, 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")
+            pdf.cell(50, 10, txt="Amount", border=1, ln=True, align="C")
 
-            # Filter records by company name
-            filtered_data = invoice_data[invoice_data["Company Name"] == company_name]
+            # Filter records by head
+            filtered_ledger = ledger[ledger["Head"] == head]
 
-            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")
+            for _, row in filtered_ledger.iterrows():
+                pdf.cell(40, 10, txt=str(row['Company Name']), border=1, align="C")
+                pdf.cell(40, 10, txt=str(row['SP/Station']), border=1, align="C")
+                pdf.cell(30, 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")
+                pdf.cell(50, 10, txt=str(row['Amount']), border=1, ln=True, align="C")
 
-            # Total amount
-            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")
+            # Total amount for the head
+            total_amount = filtered_ledger["Amount"].sum()
+            pdf.cell(140, 10, txt="Total Amount", border=1, align="R")
+            pdf.cell(50, 10, txt=str(total_amount), border=1, ln=True, align="C")
 
             # Save PDF file
             pdf.output(pdf_file_path)
@@ -85,19 +102,22 @@ def generate_invoice_pdf(company_name):
 
 # Gradio interface
 with gr.Blocks() as app:
-    # Section for adding new invoices with Company Name and Name fields
+    # Section for adding new invoices with Head and Company Name fields
     with gr.Row():
+        head = gr.Textbox(label="Head")
         company_name = gr.Textbox(label="Company Name")
-        name = gr.Textbox(label="Name")
+        sp_station = gr.Textbox(label="SP/Station")
         invoice_no = gr.Textbox(label="Invoice No")
         date = gr.Textbox(label="Date")
         amount = gr.Number(label="Amount")
         add_button = gr.Button("Add Invoice")
     
-    # Section for displaying records and total amount
+    # Section for searching and displaying ledger by head
     with gr.Row():
-        invoice_records = gr.DataFrame(label="Invoice Records")
-        total_amount = gr.Textbox(label="Total Amount", interactive=False)
+        search_head = gr.Textbox(label="Search by Head")
+        search_button = gr.Button("Search Ledger")
+        ledger_records = gr.DataFrame(label="Ledger Records")
+        total_display = gr.Textbox(label="Total Amount", interactive=False)
     
     # Section for generating and displaying PDF link
     with gr.Row():
@@ -105,16 +125,22 @@ with gr.Blocks() as app:
         pdf_output = gr.File(label="Download PDF")
 
     # Define button functionalities
-    add_button.click(add_invoice, [company_name, name, invoice_no, date, amount], [invoice_records, total_amount])
+    add_button.click(add_invoice, [head, company_name, sp_station, invoice_no, date, amount], [ledger_records, total_display])
     
-    def generate_pdf(company_name):
-        pdf_path = generate_invoice_pdf(company_name)
+    def search_ledger(head):
+        filtered_ledger, total = search_by_head(head)
+        return filtered_ledger, total
+
+    search_button.click(search_ledger, [search_head], [ledger_records, total_display])
+
+    def generate_pdf(head):
+        pdf_path = generate_ledger_pdf(head)
         if pdf_path:
             return pdf_path
         else:
             return None
 
-    generate_pdf_button.click(generate_pdf, [company_name], [pdf_output])
+    generate_pdf_button.click(generate_pdf, [search_head], [pdf_output])
 
 # Launch the app
 app.launch(share=True)