Update app.py

app.py

@@ -1,4 +1,3 @@
-
 import gradio as gr
 import pandas as pd
 from fpdf import FPDF
@@ -71,7 +70,7 @@ def generate_pdf(company_name):
         # Check if file exists
         if os.path.exists(pdf_file_path):
             print(f"PDF successfully created at {pdf_file_path}")
-            return pdf_file_path  # Returning the correct file path for Gradio to download
+            return pdf_file_path
         else:
             return "Error: PDF not generated."
     
@@ -95,15 +94,24 @@ with gr.Blocks() as app:
         invoice_records = gr.DataFrame(label="Invoice Records")
         total_amount = gr.Textbox(label="Total Amount", interactive=False)
     
-    # Section for generating and downloading PDF
+    # Section for generating and displaying PDF link
     with gr.Row():
         generate_pdf_button = gr.Button("Generate PDF")
-        pdf_output = gr.File(label="Download PDF")
+        pdf_link = gr.HTML()  # To manually display the download link
 
     # Define button functionalities
     add_button.click(add_invoice, [company_name, name, invoice_no, date, amount], [invoice_records, total_amount])
-    generate_pdf_button.click(generate_pdf, [company_name], [pdf_output])
+    
+    def show_pdf_link(company_name):
+        pdf_path = generate_pdf(company_name)
+        if pdf_path != "Error: PDF generation failed.":
+            return f'<a href="{pdf_path}" target="_blank">Download PDF</a>'
+        else:
+            return "Error generating PDF."
+
+    generate_pdf_button.click(show_pdf_link, [company_name], pdf_link)
 
 # Launch the app
 app.launch(share=True)
 
+