Create app.py

app.py

@@ -0,0 +1,28 @@
+import gradio as gr
+import pandas as pd
+from utils import create_docs
+
+def process_invoices(file_list):
+    if not file_list:
+        return "No files uploaded", None
+
+    df = create_docs(file_list)
+    if df.empty:
+        return "No data extracted from the PDFs.", None
+    else:
+        return "Data extraction completed! 🎉", df
+
+demo = gr.Interface(
+    fn=process_invoices,
+    inputs=gr.File(file_types=[".pdf"], file_count="multiple", label="Upload PDF invoices"),
+    outputs=[
+        gr.Textbox(label="Status"),
+        gr.Dataframe(label="Extracted Invoice Data")
+    ],
+    title="Invoice Extraction Bot 🤖",
+    description="Upload your PDF invoices to extract key information like invoice number, date, items, totals, and contact info.",
+    allow_flagging="never"
+)
+
+if __name__ == "__main__":
+    demo.launch()