Update app.py

app.py

@@ -13,6 +13,7 @@ from langgraph.graph import StateGraph, START, END
 from langchain_core.messages import SystemMessage, HumanMessage
 from langchain_community.document_loaders import PyPDFLoader
 
+from markdown_pdf import MarkdownPdf, Section
 from dotenv import load_dotenv
 
 load_dotenv()
@@ -270,7 +271,13 @@ def claim_generator(input_bill_path, progress=gr.Progress()):
     progress(0.05)
     output = chain.invoke({'image_path': input_bill_path})
     progress(0.25)
-    return output['final_output']
+
+    pdf = MarkdownPdf(toc_level=3)
+
+    pdf.add_section(Section(output['final_output']))
+    pdf.save("expense-claim-generator.pdf")
+    
+    return output['final_output'], 'expense-claim-generator.pdf'
 
 agentic_workflow_representation = 'The agentic workflow used to generate an expense claim document is represented below: \n <img src="https://cdn-uploads.huggingface.co/production/uploads/64118e60756b9e455c7eddd6/iqdGjUG7POKJXLItzWb-K.png">'
 agentic_workflow_description = """
@@ -290,7 +297,8 @@ Similarly, the verifier is a search agent that extracts relevant portions of the
 demo = gr.Interface(
     fn=claim_generator,
     inputs=gr.Image(type="filepath", label="Upload an image of the bill"),
-    outputs=gr.Markdown(label="Expense Claim Document", show_copy_button=True, container=True),
+    outputs=[gr.Markdown(label="Expense Claim Document", show_copy_button=True, container=True),
+             gr.File(label="Download your claim document", show_label=True)],
     title="Expense Claim Document Generator",
     description=agentic_workflow_description,
     article=agentic_workflow_representation,