Update app.py

app.py

@@ -2,9 +2,40 @@ import pandas as pd
 import plotly.express as px
 import streamlit as st
 from transformers import pipeline
+import base64
+
+# Function to add custom background image
+def add_bg_from_url(image_path):
+    # Convert image to base64
+    with open(image_path, "rb") as image_file:
+        encoded_string = base64.b64encode(image_file.read()).decode()
+
+    # HTML for background image
+    st.markdown(
+        f"""
+        <style>
+        .stApp {{
+            background-image: url("data:image/png;base64,{encoded_string}");
+            background-size: cover;
+            background-position: center center;
+            background-repeat: no-repeat;
+        }}
+        </style>
+        """,
+        unsafe_allow_html=True
+    )
+
+# Add background image (replace with your image path)
+add_bg_from_url('path_to_your_image.jpg')  # Replace with your image path
+
+# Title and header styling
+st.markdown("""
+    <h1 style="color: #00f7b7; font-family: 'Arial', sans-serif; text-align: center;">Smart Expense Tracker</h1>
+""", unsafe_allow_html=True)
 
 # File upload
 uploaded_file = st.file_uploader("Upload your expense CSV file", type=["csv"])
+
 if uploaded_file:
     df = pd.read_csv(uploaded_file)
 
@@ -17,9 +48,7 @@ if uploaded_file:
 
     # Function to categorize
     def categorize_expense(description):
-        # Check raw output from the model for debugging
         result = classifier(description, candidate_labels=categories)
-        st.write(f"Raw classification result for '{description}': {result}")  # Debugging line
         return result['labels'][0]  # Most probable category
 
     # Apply categorization
@@ -69,11 +98,11 @@ if uploaded_file:
     fig2 = px.line(monthly_expenses, x=monthly_expenses.index, y=monthly_expenses.values, title="Monthly Expenses", labels={"x": "Month", "y": "Amount ($)"})
     st.plotly_chart(fig2)
 
-    # 3. Monthly Spending vs Budget (Bar Chart with Plotly)
+    # 3. Monthly Spending vs Budget (Bar Chart)
     monthly_expenses_df = pd.DataFrame({
         'Actual': monthly_expenses,
         'Budget': [sum(budgets.values())] * len(monthly_expenses)  # Same budget for simplicity
     })
 
-    fig3 = px.bar(monthly_expenses_df, x=monthly_expenses_df.index, y=["Actual", "Budget"], title="Monthly Spending vs Budget", labels={"x": "Month", "y": "Amount ($)"})
-    st.plotly_chart(fig3)
+    fig3 = monthly_expenses_df.plot(kind='bar', figsize=(10, 6))
+    st.pyplot(fig3)