Update app.py

app.py

@@ -54,6 +54,14 @@ def search_by_head(head):
 def generate_ledger_pdf(head):
     global ledger
     try:
+        # 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
+
         # 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
@@ -75,9 +83,7 @@ def generate_ledger_pdf(head):
             pdf.cell(30, 10, txt="Date", border=1, align="C")
             pdf.cell(50, 10, txt="Amount", border=1, ln=True, align="C")
 
-            # Filter records by head
-            filtered_ledger = ledger[ledger["Head"] == head]
-
+            # Add table rows
             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")
@@ -144,4 +150,3 @@ with gr.Blocks() as app:
 
 # Launch the app
 app.launch(share=True)
-