"""
A file that will visualize salary data from an incoming CSV.
"""

import locale
import pandas as pd
import plotly.graph_objs as go

locale.setlocale(locale.LC_ALL, 'en_US.UTF-8')

df = pd.read_csv('~/git/cleberg.net/static/salary.csv')

# Function to format salary as US currency
def format_currency(value):
    """
    Format values in USD currency format.
    """
    return f"${value:,.2f}"

# Reverse the order of the DataFrame
df = df.iloc[::-1].reset_index(drop=True)

# Calculate the percentage increase
df['Percentage Increase'] = df['Salary'].pct_change() * 100

# Initialize the plot
fig = go.Figure()

# Adding each data point as a separate trace to display the text
for index, row in df.iterrows():
    TITLE_COMPANY = f"{row['Title']} ({row['Company']})"
    salary_formatted = format_currency(row['Salary'])
    if pd.notna(row['Percentage Increase']):
        text = f"{salary_formatted} ({row['Percentage Increase']:.2f}%)"
    else:
        text = salary_formatted
    fig.add_trace(go.Scatter(
        x=[row['Start'], row['End']],
        y=[row['Salary'], row['Salary']],
        text=[text],
        mode='lines+text',
        name=TITLE_COMPANY,
        textposition='top center'
    ))

# Update visual styles of the figure
fig.update_layout(
    title="Salary Data Over Time (annualized)",
    xaxis_title="Time",
    yaxis_title="Salary",
    font={"family": 'monospace', "size": 16},
    margin={"l": 50, "r": 50, "t": 50, "b": 100},
    legend={
	"orientation": 'h',
	"yanchor": 'top',
	"y": -0.3,
	"xanchor": 'center',
	"x": 0.5
    },
    height=800
)

fig.show()