Update app.py

app.py

@@ -54,6 +54,10 @@ def search_by_head(head):
 def generate_ledger_pdf(head):
     global ledger
     try:
+        # Ensure the head is not empty
+        if head.strip() == "":
+            return None
+
         # 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
@@ -80,21 +84,24 @@ def generate_ledger_pdf(head):
 
             # Filter records by head
             filtered_ledger = ledger[ledger["Head"] == head]
-            pdf.set_font("Arial", size=12)
-            
-            for _, row in filtered_ledger.iterrows():
-                pdf.cell(50, 10, str(row['Company Name']), border=1, align="C")
-                pdf.cell(40, 10, str(row['SP/Station']), border=1, align="C")
-                pdf.cell(30, 10, str(row['Invoice No']), border=1, align="C")
-                pdf.cell(30, 10, str(row['Date']), border=1, align="C")
-                pdf.cell(40, 10, f"{row['Amount']:.2f}", border=1, ln=True, align="C")
-
-            # Total amount for the head
-            total_amount = filtered_ledger["Amount"].sum()
-            pdf.ln(10)  # Add some space before the total
-            pdf.set_font("Arial", 'B', 12)
-            pdf.cell(150, 10, "Total Amount", border=1, align="R")
-            pdf.cell(40, 10, f"{total_amount:.2f}", border=1, ln=True, align="C")
+            if filtered_ledger.empty:
+                pdf.cell(0, 10, f"No records found for {head}", ln=True, align="C")
+            else:
+                pdf.set_font("Arial", size=12)
+                
+                for _, row in filtered_ledger.iterrows():
+                    pdf.cell(50, 10, str(row['Company Name']), border=1, align="C")
+                    pdf.cell(40, 10, str(row['SP/Station']), border=1, align="C")
+                    pdf.cell(30, 10, str(row['Invoice No']), border=1, align="C")
+                    pdf.cell(30, 10, str(row['Date']), border=1, align="C")
+                    pdf.cell(40, 10, f"{row['Amount']:.2f}", border=1, ln=True, align="C")
+
+                # Total amount for the head
+                total_amount = filtered_ledger["Amount"].sum()
+                pdf.ln(10)  # Add some space before the total
+                pdf.set_font("Arial", 'B', 12)
+                pdf.cell(150, 10, "Total Amount", border=1, align="R")
+                pdf.cell(40, 10, f"{total_amount:.2f}", border=1, ln=True, align="C")
 
             # Save PDF file
             pdf.output(pdf_file_path)