Update app.py

app.py

@@ -333,57 +333,20 @@ def empty_df():
                                "Credit", "Closing Balance", "Category"])
 
 # Gradio Interface
-# 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]
-    )
+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()