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 this for temporary file handling
+import tempfile  # Use for temporary file handling
 import os
 
 # Initialize an empty DataFrame to store all invoice records
@@ -60,7 +60,6 @@ def generate_invoice_pdf(company_name):
 
             # Filter records by company name
             filtered_data = invoice_data[invoice_data["Company Name"] == company_name]
-            print(f"Processing {len(filtered_data)} records for {company_name}...")
 
             for _, row in filtered_data.iterrows():
                 pdf.cell(50, 10, txt=str(row['Company Name']), border=1, align="C")
@@ -82,7 +81,7 @@ def generate_invoice_pdf(company_name):
 
     except Exception as e:
         print(f"Error during PDF generation: {e}")
-        return "Error: PDF generation failed."
+        return None
 
 # Gradio interface
 with gr.Blocks() as app:
@@ -103,19 +102,19 @@ with gr.Blocks() as app:
     # Section for generating and displaying PDF link
     with gr.Row():
         generate_pdf_button = gr.Button("Generate PDF")
-        pdf_link = gr.HTML()  # To manually display the download link
+        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])
     
-    def show_pdf_link(company_name):
+    def generate_pdf(company_name):
         pdf_path = generate_invoice_pdf(company_name)
-        if pdf_path != "Error: PDF generation failed.":
-            return f'<a href="{pdf_path}" target="_blank">Download PDF</a>'
+        if pdf_path:
+            return pdf_path
         else:
-            return "Error generating PDF."
+            return None
 
-    generate_pdf_button.click(show_pdf_link, [company_name], pdf_link)
+    generate_pdf_button.click(generate_pdf, [company_name], [pdf_output])
 
 # Launch the app
 app.launch(share=True)