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 tempfile  # Use for temporary file handling
+import tempfile
 import os
 
 # Initialize an empty DataFrame to store all records of invoices (acting as our ledger)
@@ -23,19 +23,16 @@ def add_invoice(head, company_name, sp_station, invoice_no, date, amount):
         }
         ledger = pd.concat([ledger, pd.DataFrame([new_row])], ignore_index=True)
         
-        # Debugging output
-        print("New row added:", new_row)
-        print("Updated ledger:", ledger)
-
         # Filter the ledger for the selected head
         filtered_ledger = ledger[ledger["Head"] == head]
-
-        # Calculate the total for the selected head
+        
+        # Return filtered ledger and the sum of amounts for this head
         total_amount = filtered_ledger["Amount"].sum()
-        return filtered_ledger, f"Total for {head}: {total_amount}"
+        
+        # Return only the relevant rows and total to speed up processing
+        return filtered_ledger.to_dict('records'), 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 search and display all previous records by head
@@ -45,16 +42,12 @@ def search_by_head(head):
         # Filter the ledger for the selected head
         filtered_ledger = ledger[ledger["Head"] == head]
 
-        # Debugging output
-        print(f"Records for {head}:", filtered_ledger)
-
         # Calculate the total for the selected head
         total_amount = filtered_ledger["Amount"].sum()
 
-        return filtered_ledger, f"Total for {head}: {total_amount}"
+        return filtered_ledger.to_dict('records'), 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
@@ -64,17 +57,12 @@ def generate_ledger_pdf(head):
         # Filter records by head
         filtered_ledger = ledger[ledger["Head"] == head]
 
-        # Check if there are any records for the specified head
         if filtered_ledger.empty:
-            print(f"No records found for {head}")
             return None
 
-        # Debugging output
-        print(f"Generating PDF for {head}")
-
         # Use a temporary directory to save the file
         with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as tmp_file:
-            pdf_file_path = tmp_file.name  # Get the temporary file path
+            pdf_file_path = tmp_file.name
 
             # Create a PDF
             pdf = FPDF()
@@ -101,25 +89,21 @@ def generate_ledger_pdf(head):
                 pdf.cell(30, 10, txt=str(row['Date']), border=1, align="C")
                 pdf.cell(50, 10, txt=str(row['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
+            # Save the PDF
             pdf.output(pdf_file_path)
 
-            # Return the file path for download
-            print(f"PDF generated at: {pdf_file_path}")
+            # Return the path of the generated PDF
             return pdf_file_path
 
     except Exception as e:
-        print(f"Error during PDF generation: {e}")
         return None
 
 # Gradio interface
 with gr.Blocks() as app:
-    # 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")
@@ -128,15 +112,13 @@ with gr.Blocks() as app:
         date = gr.Textbox(label="Date")
         amount = gr.Number(label="Amount")
         add_button = gr.Button("Add Invoice")
-    
-    # Section for searching and displaying ledger by head
+
     with gr.Row():
         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():
         generate_pdf_button = gr.Button("Generate PDF")
         pdf_output = gr.File(label="Download PDF")
@@ -144,20 +126,9 @@ with gr.Blocks() as app:
     # Define button functionalities
     add_button.click(add_invoice, [head, company_name, sp_station, invoice_no, date, amount], [ledger_records, total_display])
     
-    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
+    search_button.click(search_by_head, [search_head], [ledger_records, total_display])
 
-    generate_pdf_button.click(generate_pdf, [search_head], [pdf_output])
+    generate_pdf_button.click(generate_ledger_pdf, [search_head], [pdf_output])
 
 # Launch the app
 app.launch(share=True)