Update app.py

app.py

@@ -3,6 +3,40 @@ import pandas as pd
 import matplotlib.pyplot as plt
 import seaborn as sns
 
+# Initialize an empty DataFrame to hold expenses data
+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'] = []
+
+# 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)
+
+# Function to generate response for the chatbot
+def chatbot_response(user_input):
+    # If the user wants to add an expense, we can extract the details from the input
+    if "add" in user_input.lower() and "expense" in user_input.lower():
+        # Ask the user for details about the expense
+        st.session_state['messages'].append({"role": "assistant", "content": "Please enter the date of the expense."})
+        return "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'])})
+        return "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)})
+        return "Here is the expense suimport streamlit as st
+import pandas as pd
+import matplotlib.pyplot as plt
+import seaborn as sns
+
 # Initialize an empty DataFrame to hold expenses data
 columns = ['Date', 'Category', 'Description', 'Amount']
 if 'expenses_df' not in st.session_state:
@@ -69,7 +103,8 @@ if user_input:
     st.session_state['messages'].append({"role": "assistant", "content": response})
 
 # Expense Addition Flow
-if "Please enter the date of the expense." in st.session_state['messages'][-1].get("content", ""):
+# Ensure there is at least one message and check for the assistant's message asking for date input
+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")