Update app.py

app.py

@@ -6,7 +6,7 @@ st.title("Stock Portfolio and Tax Analysis")
 
 # Initialize empty dataframes for stock sales and portfolio stocks
 if 'stock_sales' not in st.session_state:
-    st.session_state.stock_sales = pd.DataFrame(columns=['Name', 'Sale Price', 'Purchase Price', 'Holding Period (Months)'])
+    st.session_state.stock_sales = pd.DataFrame(columns=['Name', 'Sale Price', 'Purchase Price', 'Quantity', 'Holding Period (Months)'])
 if 'portfolio_stocks' not in st.session_state:
     st.session_state.portfolio_stocks = pd.DataFrame(columns=['Name', 'Purchase Price', 'Current Price', 'Quantity'])
 
@@ -25,11 +25,11 @@ def add_stock_sale():
         with col5:
             holding_period_months = st.number_input("Holding Period (Months)", min_value=1, key="stock_sale_holding_period")
         
-        submit_button = st.form_submit_button("Add Stock Sale")
+        submit_button = st.form_submit_button("Add Sold Stock")
         if submit_button:
             new_stock = pd.DataFrame([{'Name': name, 'Sale Price': sale_price, 'Purchase Price': purchase_price, 'Quantity': sell_quantity, 'Holding Period (Months)': holding_period_months}])
             st.session_state.stock_sales = pd.concat([st.session_state.stock_sales, new_stock], ignore_index=True)
-            st.success("Stock Sale Added!")
+            st.success("Sold Stock Added!")
 
 # Define function to add portfolio stock
 def add_portfolio_stock():
@@ -53,7 +53,7 @@ def add_portfolio_stock():
 # Define functions to edit and delete stocks
 def delete_stock_sale(index):
     st.session_state.stock_sales = st.session_state.stock_sales.drop(index).reset_index(drop=True)
-    st.success("Stock Sale Deleted!")
+    st.success("Sold Stock Deleted!")
 
 def delete_portfolio_stock(index):
     st.session_state.portfolio_stocks = st.session_state.portfolio_stocks.drop(index).reset_index(drop=True)
@@ -65,7 +65,7 @@ def edit_stock_sale(index, name, sale_price, purchase_price, holding_period_mont
     st.session_state.stock_sales.at[index, 'Purchase Price'] = purchase_price
     st.session_state.stock_sales.at[index, 'Quantity'] = sell_quantity
     st.session_state.stock_sales.at[index, 'Holding Period (Months)'] = holding_period_months
-    st.success("Stock Sale Updated!")
+    st.success("Sold Stock Updated!")
 
 def edit_portfolio_stock(index, name, purchase_price, current_price, quantity):
     st.session_state.portfolio_stocks.at[index, 'Name'] = name