Update app to support log scale

app.py

@@ -16,6 +16,7 @@ st.set_page_config(page_title="Asset Class Comparison", layout="wide")
 # Title and description
 st.title("Asset Class Performance Comparison")
 st.write("Compare the performance of different asset classes over time")
+st.write("Note: Bitcoin (BTC) is highly volatile and should be considered a high-risk investment")
 
 # Sidebar for user inputs
 st.sidebar.header("Investment Parameters")
@@ -24,6 +25,7 @@ initial_investment = st.sidebar.number_input(f"Initial Investment Amount ({curre
 start_date = st.sidebar.date_input("Start Date", value=datetime.now() - timedelta(days=365*10))
 user_end_date = st.sidebar.date_input("End Date", value=datetime.now())
 fd_rate = st.sidebar.number_input("Fixed Deposit Rate (%)", min_value=0.0, value=2.9, step=0.1) / 100
+use_log_scale = st.sidebar.checkbox("Use Log Scale", value=False)
 
 # Asset selection
 selected_assets = st.sidebar.multiselect(
@@ -46,6 +48,7 @@ selected_assets = st.sidebar.multiselect(
         "Tesla",
         "Netflix",
         "Meta",
+        "Bitcoin",
     ],
     default=["Fixed Deposit", "Microsoft", "Google", "Nvidia"]
 )
@@ -74,6 +77,7 @@ tickers = {
     "Tesla": "TSLA",
     "Netflix": "NFLX",
     "Meta": "META",
+    "Bitcoin": "BTC-USD",
 }
 
 # Determine the effective end date for each asset
@@ -152,7 +156,8 @@ fig.update_layout(
     xaxis_title="Date",
     yaxis_title=f"Investment Value ({currency_symbol})",
     hovermode="x unified",
-    height=600
+    height=600,
+    yaxis_type="log" if use_log_scale else "linear"
 )
 
 # Display the plot