Update app.py

app.py

@@ -1,14 +1,15 @@
-import gradio as gr
+import streamlit as st
 import pdfplumber
 import pandas as pd
 
 # Function to process PDF and classify transactions
-def process_pdf(file_path):
-    if file_path is None:
-        return "No file uploaded."
+def process_pdf(file):
+    if file is None:
+        st.error("No file uploaded.")
+        return None
     
     # Extract text from the uploaded PDF
-    with pdfplumber.open(file_path) as pdf:
+    with pdfplumber.open(file) as pdf:
         text = "\n".join([page.extract_text() for page in pdf.pages if page.extract_text()])
 
     # Extract transactions (Modify based on statement format)
@@ -35,9 +36,21 @@ def process_pdf(file_path):
     # Apply classification
     df["Category"] = df["Description"].apply(classify_transaction)
 
-    return df  # Display the table
+    return df  # Return DataFrame
 
-# Fix: Use "filepath" instead of "file"
-app = gr.Interface(fn=process_pdf, inputs=gr.File(type="filepath"), outputs="dataframe", title="Bank Statement Classifier")
-app.launch()
+# Streamlit UI
+st.title("📄 Credit Card Statement Classifier")
+st.write("Upload a PDF bank/credit card statement to categorize transactions.")
+
+uploaded_file = st.file_uploader("Upload PDF", type=["pdf"])
+
+if uploaded_file is not None:
+    st.success("File uploaded successfully!")
+    
+    # Process and display transactions
+    df_result = process_pdf(uploaded_file)
+    
+    if df_result is not None:
+        st.write("### Classified Transactions:")
+        st.dataframe(df_result)  # Display table