Made improvements

app.py

@@ -4,6 +4,11 @@ import yfinance as yf
 import plotly.graph_objects as go
 from datetime import datetime, timedelta
 import numpy as np
+import base64
+from io import BytesIO
+import matplotlib.pyplot as plt
+import matplotlib
+matplotlib.use('Agg')
 
 # Import utility functions
 from utils.yfinance_utils import fetch_yfinance_daily
@@ -164,6 +169,12 @@ if not selected_assets:
 # Create the plot
 fig = go.Figure()
 
+# Add vertical lines for every 5 years
+start_year = returns_data.index[0].year
+end_year = returns_data.index[-1].year
+for year in range(start_year, end_year + 1, 5):
+    fig.add_vline(x=datetime(year, 1, 1), line_dash="dash", line_color="gray", opacity=0.3)
+
 for asset in selected_assets:
     fig.add_trace(go.Scatter(
         x=returns_data.index,
@@ -202,12 +213,27 @@ for asset in selected_assets:
         total_years = len(yearly_returns)
         positive_percentage = (positive_years / total_years) * 100
         
+        # Create sparkline using matplotlib
+        plt.figure(figsize=(2, 0.5))
+        plt.plot(valid_series.index, valid_series, linewidth=1)
+        plt.axis('off')
+        plt.margins(0)
+        plt.tight_layout(pad=0)
+        
+        # Convert plot to base64 string
+        buf = BytesIO()
+        plt.savefig(buf, format='png', bbox_inches='tight', pad_inches=0, transparent=True)
+        buf.seek(0)
+        sparkline = base64.b64encode(buf.read()).decode('utf-8')
+        plt.close()
+        
         summary_data.append({
             "Asset": asset,
             f"Final Value ({currency_symbol})": final_value,
             "Annualized Return (%)": annualized_return,
             "Positive Years": f"{positive_years}/{total_years}",
-            "Positive Years %": positive_percentage
+            "Positive Years %": positive_percentage,
+            "Performance": f'<img src="data:image/png;base64,{sparkline}" width="100" height="30">'
         })
     else:
         summary_data.append({
@@ -215,7 +241,8 @@ for asset in selected_assets:
             f"Final Value ({currency_symbol})": None,
             "Annualized Return (%)": None,
             "Positive Years": "N/A",
-            "Positive Years %": None
+            "Positive Years %": None,
+            "Performance": "N/A"
         })
 
 # Convert to DataFrame
@@ -239,6 +266,9 @@ st.dataframe(
         ),
         "Positive Years %": st.column_config.NumberColumn(
             format="%.1f"
+        ),
+        "Performance": st.column_config.ImageColumn(
+            "Performance"
         )
     }
 )