Update app.py

app.py

@@ -333,21 +333,57 @@ def empty_df():
                                "Credit", "Closing Balance", "Category"])
 
 # Gradio Interface
-interface = gr.Interface(
-    fn=process_file,
-    inputs=[
-        gr.File(label="Upload Bank Statement (PDF/Excel)"),
-        gr.Checkbox(label="Is Scanned PDF? (Use OCR)")
-    ],
-    outputs=gr.Dataframe(
-        label="Parsed Transactions",
-        headers=["Date", "Description", "Amount", "Debit", "Credit", "Closing Balance", "Category"],
-        datatype=["date", "str", "number", "number", "number", "number", "str"]
-    ),
-    title="AI Bank Statement Parser",
-    description="Extract structured transaction data from PDF/Excel bank statements",
-    allow_flagging="never"
-)
+# interface = gr.Interface(
+#     fn=process_file,
+#     inputs=[
+#         gr.File(label="Upload Bank Statement (PDF/Excel)")
+#     ],
+#     outputs=gr.Dataframe(
+#         label="Parsed Transactions",
+#         headers=["Date", "Description", "Amount", "Debit", "Credit", "Closing Balance", "Category"],
+#         datatype=["date", "str", "number", "number", "number", "number", "str"]
+#     ),
+#     title="AI Bank Statement Parser",
+#     description="Extract structured transaction data from PDF/Excel bank statements",
+#     allow_flagging="never"
+# )
+
+# if __name__ == "__main__":
+#     interface.launch()
+with gr.Blocks(title="AI Bank Statement Parser") as interface:
+    gr.Markdown("# AI Bank Statement Parser")
+    gr.Markdown("Extract structured transaction data from PDF/Excel bank statements")
+    
+    with gr.Row():
+        file_input = gr.File(label="Upload Bank Statement (PDF/Excel)")
+    
+    with gr.Row():
+        submit_btn = gr.Button("Process Statement", variant="primary")
+    
+    with gr.Row():
+        dataframe_output = gr.Dataframe(
+            label="Parsed Transactions",
+            headers=["Date", "Description", "Amount", "Debit", "Credit", "Closing Balance", "Category"],
+            datatype=["date", "str", "number", "number", "number", "number", "str"]
+        )
+    
+    with gr.Row():
+        download_pdf = gr.File(label="Download Report", interactive=False)
+        download_btn = gr.Button("Generate PDF Report")
+    
+    # Processing function
+    submit_btn.click(
+        fn=process_file,
+        inputs=[file_input],
+        outputs=[dataframe_output, download_pdf]
+    )
+    
+    # Download button handler
+    download_btn.click(
+        fn=lambda df: generate_pdf_report(df) if df is not None and not df.empty else None,
+        inputs=[dataframe_output],
+        outputs=[download_pdf]
+    )
 
 if __name__ == "__main__":
     interface.launch()