Update app.py

app.py

@@ -6,7 +6,7 @@ class SalaryTracker:
     def __init__(self, initial_salary):
         self.salary = initial_salary
         self.expenses = []
-        self.data = pd.DataFrame({'Month': [], 'Salary': []})
+        self.data = pd.DataFrame({'Transaction': [], 'Salary': []})
 
     def add_expense(self, amount):
         self.expenses.append(amount)
@@ -16,8 +16,8 @@ class SalaryTracker:
         self.expenses.append(-amount)
         self.salary += amount
 
-    def update_data(self, month):
-        self.data = self.data.append({'Month': month, 'Salary': self.salary}, ignore_index=True)
+    def update_data(self, transaction):
+        self.data = self.data.append({'Transaction': transaction, 'Salary': self.salary}, ignore_index=True)
 
     def generate_report(self):
         return self.data
@@ -36,14 +36,12 @@ deduct_expense = col2.number_input("Deduct Expense", min_value=0.0, key="deduct_
 if st.button("Add Expense"):
     if add_expense > 0:
         salary_tracker.add_expense(add_expense)
-        month = st.text_input("Enter the month and year (e.g., 'Oct 2023'):", key="month")
-        salary_tracker.update_data(month)
+        salary_tracker.update_data("Credit")
 
 if st.button("Deduct Expense"):
     if deduct_expense > 0:
         salary_tracker.deduct_expense(deduct_expense)
-        month = st.text_input("Enter the month and year (e.g., 'Oct 2023'):", key="month")
-        salary_tracker.update_data(month)
+        salary_tracker.update_data("Debit")
 
 # Display Salary Data
 st.subheader("Salary History")
@@ -53,9 +51,8 @@ if not salary_tracker.data.empty:
 # Generate and Display Graph
 if not salary_tracker.data.empty:
     plt.figure(figsize=(10, 5))
-    plt.plot(salary_tracker.data['Month'], salary_tracker.data['Salary'], marker='o')
-    plt.title("Salary History Over Time")
-    plt.xlabel("Month and Year")
+    plt.plot(salary_tracker.data['Transaction'], salary_tracker.data['Salary'], marker='o')
+    plt.title("Salary History Over Transactions")
+    plt.xlabel("Transactions")
     plt.ylabel("Salary")
     st.pyplot(plt)
-