Update app.py

app.py

@@ -2,6 +2,7 @@ import pandas as pd
 import plotly.express as px
 import streamlit as st
 from transformers import pipeline
+import matplotlib.pyplot as plt
 
 # Function to add custom background image from a URL
 def add_bg_from_url(image_url):
@@ -19,6 +20,10 @@ def add_bg_from_url(image_url):
         unsafe_allow_html=True
     )
 
+# Add the background image using the provided URL
+background_image_url = 'https://huggingface.co/spaces/engralimalik/Smart-Expense-Tracker/resolve/main/colorful-abstract-textured-background-design.jpg'
+add_bg_from_url(background_image_url)
+
 # File upload
 uploaded_file = st.file_uploader("Upload your expense CSV file", type=["csv"])
 if uploaded_file:
@@ -27,18 +32,18 @@ if uploaded_file:
     # Display Dataframe
     st.write(df.head())
 
-    # Initialize Hugging Face model for zero-shot classification
-    classifier = pipeline('zero-shot-classification', model='distilbert-base-uncased')
+    # Initialize Hugging Face model for zero-shot classification (using a better model like roberta-large-mnli)
+    classifier = pipeline('zero-shot-classification', model='roberta-large-mnli')
     categories = ["Groceries", "Rent", "Utilities", "Entertainment", "Dining", "Transportation"]
 
-    # Function to categorize
+    # Function to categorize expenses based on the description
     def categorize_expense(description):
         result = classifier(description, candidate_labels=categories)
         return result['labels'][0]  # Most probable category
 
     # Apply categorization
     df['Category'] = df['Description'].apply(categorize_expense)
-    
+
     # Display categorized data
     st.write("Categorized Data", df)
 
@@ -89,9 +94,9 @@ if uploaded_file:
         'Budget': [sum(budgets.values())] * len(monthly_expenses)  # Same budget for simplicity
     })
 
-    fig3 = monthly_expenses_df.plot(kind='bar', figsize=(10, 6))
+    # Create a matplotlib figure for the bar chart
+    fig3, ax = plt.subplots(figsize=(10, 6))
+    monthly_expenses_df.plot(kind='bar', ax=ax)
+    ax.set_title('Monthly Spending vs Budget')
+    ax.set_ylabel('Amount ($)')
     st.pyplot(fig3)
-
-# Add the background image using the provided URL
-background_image_url = 'https://huggingface.co/spaces/engralimalik/Smart-Expense-Tracker/resolve/main/colorful-abstract-textured-background-design.jpg'
-add_bg_from_url(background_image_url)