Update app.py

app.py

@@ -8,67 +8,56 @@ columns = ['Date', 'Category', 'Description', 'Amount']
 if 'expenses_df' not in st.session_state:
     st.session_state['expenses_df'] = pd.DataFrame(columns=columns)
 
-# Initialize a list for storing messages between user and bot
-if 'messages' not in st.session_state:
-    st.session_state['messages'] = []
+# Initialize the UI
+st.title("Daily Expense Tracker")
 
 # Function to add an expense
 def add_expense(date, category, description, amount):
     new_expense = pd.DataFrame([[date, category, description, amount]], columns=st.session_state['expenses_df'].columns)
     st.session_state['expenses_df'] = pd.concat([st.session_state['expenses_df'], new_expense], ignore_index=True)
 
-# Streamlit UI
-st.title("Daily Expense Tracker")
-
-# Display messages (chat history)
-for message in st.session_state['messages']:
-    if message["role"] == "assistant":
-        st.chat_message("assistant").markdown(message["content"])
-    else:
-        st.chat_message("user").markdown(message["content"])
-
-# User input for chatbot
-user_input = st.text_input("You:", key="user_input")
-
-if user_input:
-    st.session_state['messages'].append({"role": "user", "content": user_input})
-
-    # Get response from chatbot
-    response = "I'm sorry, I didn't understand that."
-    if "add" in user_input.lower() and "expense" in user_input.lower():
-        st.session_state['messages'].append({"role": "assistant", "content": "Please enter the date of the expense."})
-        response = "Please enter the date of the expense."
-
-    elif "view" in user_input.lower():
-        st.session_state['messages'].append({"role": "assistant", "content": str(st.session_state['expenses_df'])})
-        response = "Here are all your expenses:\n" + str(st.session_state['expenses_df'])
-    
-    elif "summary" in user_input.lower():
-        category_summary = st.session_state['expenses_df'].groupby('Category')['Amount'].sum().reset_index()
-        st.session_state['messages'].append({"role": "assistant", "content": str(category_summary)})
-        response = "Here is the expense summary by category:\n" + str(category_summary)
-
-    elif "visualize" in user_input.lower():
-        fig, ax = plt.subplots(figsize=(10, 6))
-        sns.barplot(x='Category', y='Amount', data=st.session_state['expenses_df'], ax=ax)
-        ax.set_title('Total Expenses by Category')
-        ax.set_xlabel('Category')
-        ax.set_ylabel('Amount Spent ($)')
-        ax.set_xticklabels(ax.get_xticklabels(), rotation=45)
-        st.pyplot(fig)
-        response = "Here is the bar chart showing total expenses by category."
-
-    st.session_state['messages'].append({"role": "assistant", "content": response})
-
-# Expense Addition Flow
-if st.session_state['messages'] and "Please enter the date of the expense." in st.session_state['messages'][-1].get("content", ""):
-    date = st.date_input("Enter Expense Date")
-    category = st.selectbox("Category", ['Food', 'Transport', 'Entertainment', 'Other'])
-    description = st.text_input("Description")
-    amount = st.number_input("Amount", min_value=0.0, format="%.2f")
+# Add some sample expenses (for testing purposes)
+if st.button("Add Sample Expenses"):
+    add_expense('2024-12-01', 'Food', 'Lunch at restaurant', 15.50)
+    add_expense('2024-12-02', 'Transport', 'Uber ride', 20.00)
+    add_expense('2024-12-03', 'Entertainment', 'Movie ticket', 12.00)
+    add_expense('2024-12-04', 'Food', 'Grocery shopping', 35.75)
+    st.success("Sample expenses added.")
+
+# Display expenses list
+st.subheader("Expenses List")
+st.write(st.session_state['expenses_df'])
+
+# Generate summary by category
+category_summary = st.session_state['expenses_df'].groupby('Category')['Amount'].sum().reset_index()
+
+st.subheader("Category Summary")
+st.write(category_summary)
+
+# Visualize the expenses by category (Bar Chart)
+st.subheader("Expense Visualization (Bar Chart)")
+fig, ax = plt.subplots(figsize=(10, 6))
+sns.barplot(x='Category', y='Amount', data=category_summary, ax=ax)
+ax.set_title('Total Expenses by Category')
+ax.set_xlabel('Category')
+ax.set_ylabel('Amount Spent ($)')
+ax.set_xticklabels(ax.get_xticklabels(), rotation=45)
+st.pyplot(fig)
+
+# Pie chart for category-wise expenses
+st.subheader("Expense Distribution (Pie Chart)")
+fig2, ax2 = plt.subplots(figsize=(8, 8))
+category_summary.set_index('Category')['Amount'].plot(kind='pie', autopct='%1.1f%%', legend=False, ax=ax2)
+ax2.set_title('Expense Distribution by Category')
+ax2.set_ylabel('')  # Hide ylabel
+st.pyplot(fig2)
+
+# Option to save the expenses to a CSV file
+st.download_button(
+    label="Download Expenses as CSV",
+    data=st.session_state['expenses_df'].to_csv(index=False),
+    file_name="expenses.csv",
+    mime="text/csv"
+)
 
-    if st.button("Submit Expense"):
-        add_expense(date, category, description, amount)
-        st.session_state['messages'].append({"role": "assistant", "content": f"Expense added: {description} - ${amount:.2f}"})
-        st.success(f"Added expense: {description} - ${amount:.2f}")