Update app.py

app.py

@@ -4,16 +4,39 @@ import streamlit as st
 from transformers import pipeline
 import datetime
 
+# Function to add background image to the app
+def add_bg_from_url(image_url):
+    st.markdown(
+        f"""
+        <style>
+        .stApp {{
+            background-image: url({image_url});
+            background-size: cover;
+            background-position: center center;
+            background-repeat: no-repeat;
+        }}
+        </style>
+        """,
+        unsafe_allow_html=True
+    )
+
+# Set background image (it will remain even after file upload)
+add_bg_from_url('https://huggingface.co/spaces/engralimalik/Smart-Expense-Tracker/resolve/main/top-view-finance-business-elements.jpg')
+
 # File upload
 uploaded_file = st.file_uploader("Upload your expense CSV file", type=["csv"])
 if uploaded_file:
     df = pd.read_csv(uploaded_file)
 
-    # Display Dataframe
+    # Display first few rows to the user for format verification
+    st.write("Here are the first few entries in your file for format verification:")
     st.write(df.head())
 
+    # Ensure 'Amount' is numeric
+    df['Amount'] = pd.to_numeric(df['Amount'], errors='coerce')
+
     # Initialize Hugging Face model for zero-shot classification
-    classifier = pipeline('zero-shot-classification', model='distilbert-base-uncased')
+    classifier = pipeline('zero-shot-classification', model='roberta-large-mnli')
     categories = ["Groceries", "Rent", "Utilities", "Entertainment", "Dining", "Transportation"]
 
     # Function to categorize