Update app.py

app.py

@@ -2,81 +2,73 @@ import streamlit as st
 import pandas as pd
 import matplotlib.pyplot as plt
 import seaborn as sns
-import requests
+
 # Create an empty DataFrame for expenses
 columns = ['Date', 'Category', 'Description', 'Amount']
-expenses_df = pd.DataFrame(columns=columns)
-
-# Streamlit app layout
-st.title('Expense Tracker')
-
-# Add new expense entry
-st.subheader("Add New Expense")
+if 'expenses_df' not in st.session_state:
+    st.session_state['expenses_df'] = pd.DataFrame(columns=columns)
 
-date = st.date_input("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")
+# Initialize a list for storing messages between user and bot
+if 'messages' not in st.session_state:
+    st.session_state['messages'] = []
 
-if st.button("Add Expense"):
-    new_expense = pd.DataFrame([[date, category, description, amount]], columns=expenses_df.columns)
-    expenses_df = pd.concat([expenses_df, new_expense], ignore_index=True)
-    st.success(f"Added expense: {description} - ${amount:.2f}")
+# 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)
 
-# Display all expenses
-st.subheader("Expenses List")
-st.write(expenses_df)
+# Streamlit UI
+st.title("Daily Expense Tracker")
 
-# Ensure 'Amount' is numeric (convert if necessary)
-expenses_df['Amount'] = pd.to_numeric(expenses_df['Amount'], errors='coerce')
+# 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"])
 
-# Show category summary
-st.subheader("Category Summary")
-category_summary = expenses_df.groupby('Category')['Amount'].sum().reset_index()
-st.write(category_summary)
+# User input for chatbot
+user_input = st.text_input("You:", key="user_input")
 
-# Generate individual bar and pie charts per Description
-st.subheader("Bar Charts and Pie Charts per Description")
+if user_input:
+    st.session_state['messages'].append({"role": "user", "content": user_input})
 
-# Get all unique descriptions
-unique_descriptions = expenses_df['Description'].unique()
+    # 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."
 
-# Loop through each unique description to plot bar and pie charts
-for description in unique_descriptions:
-    description_df = expenses_df[expenses_df['Description'] == description]
+    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'])
     
-    # Plot bar chart for each description
-    fig, ax = plt.subplots(figsize=(10, 6))
-    sns.barplot(x='Category', y='Amount', data=description_df, ax=ax)
-    ax.set_title(f'Total Expenses for: {description}')
-    ax.set_xlabel('Category')
-    ax.set_ylabel('Amount Spent ($)')
-    ax.set_xticklabels(ax.get_xticklabels(), rotation=45)
-    st.pyplot(fig)
+    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)
 
-    # Plot pie chart for each description
-    fig, ax = plt.subplots(figsize=(8, 8))
-    description_summary = description_df.groupby('Category')['Amount'].sum().reset_index()
-    description_summary.set_index('Category')['Amount'].plot(kind='pie', autopct='%1.1f%%', legend=False, ax=ax)
-    ax.set_title(f'Expense Distribution for: {description}')
-    ax.set_ylabel('')  # Hide ylabel
-    st.pyplot(fig)
+    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."
 
-# Save the expenses to a CSV file
-st.download_button(
-    label="Download Expenses as CSV",
-    data=expenses_df.to_csv(index=False),
-    file_name="expenses.csv",
-    mime="text/csv"
-)
+    st.session_state['messages'].append({"role": "assistant", "content": response})
 
-# Example of making an API call using the API Key (optional, depending on your use case)
-api_url = "https://example.com/api/data"
-headers = {"Authorization": f"Bearer {API_KEY}"}
+# 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")
 
-response = requests.get(api_url, headers=headers)
+    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}")
 
-if response.status_code == 200:
-    st.write("API Data:", response.json())  # Display data from API response
-else:
-    st.error("Failed to fetch data from API")