app.py

from dash import Dash, dcc, html, Input, Output, State
import plotly.express as px
from cache_data import get_data_from_file, get_cached_data
from datetime import datetime as dt
import dash_bootstrap_components as dbc
from dash_bootstrap_templates import load_figure_template
from dash.exceptions import PreventUpdate

"""
Bitcoin DCA calculator
x axis = date, y axis = total amount dca'd
total = sum date start - date end of [amt * frequency]
unit rate = fiat amt / btc rate in fiat
frequency = (daily, weekly, bi-weekly, monthly)
currency = hkd or usd
Inputs: amount, currency, date range, frequency
"""

LOGO = "https://rates.bitcoin.org.hk/static/images/BAHK_black_square.svg"

# stylesheet with the .dbc class
dbc_css = "https://cdn.jsdelivr.net/gh/AnnMarieW/dash-bootstrap-templates/dbc.min.css"
app = Dash(external_stylesheets=[dbc.themes.VAPOR, dbc_css])
app.title = "DCA"
server = app.server

# datafile = "./btc_historical"
# df = get_data_from_file(datafile)
df = get_cached_data()
# Find the latest date in the index
latest_date = df.index.max()
print(latest_date)

# Check for duplicates
duplicates = df.index.duplicated(keep="first")
# Remove duplicate rows
df = df[~duplicates]

# df = get_cached_data()
df_weekly = df.resample("W").last()
df_biweekly = df.resample("2W").last()
df_monthly = df.resample("M").last()


load_figure_template(
    ["sketchy", "cyborg", "minty", "darkly", "vapor", "slate", "superhero", "quartz"]
)
template_type = "vapor"


items_bar = dbc.Row(
    [
        dbc.Col(
            dbc.NavItem(dbc.NavLink("Rates", href="https://rates.bitcoin.org.hk/"))
        ),
        dbc.Col(dbc.NavItem(dbc.NavLink("Sats", href="https://sats.bitcoin.org.hk/"))),
        dbc.Col(
            dbc.NavItem(dbc.NavLink("Blocks", href="https://blocks.bitcoin.org.hk/"))
        ),
    ],
    className="text-white ms-auto flex-nowrap mt-3 mt-md-0",
    align="center",
)

navbar = dbc.Navbar(
    dbc.Container(
        [
            html.A(
                # Use row and col to control vertical alignment of logo / brand
                dbc.Row(
                    [
                        dbc.Col(html.Img(src=LOGO, height="60px")),
                        dbc.Col(dbc.NavbarBrand("Bitcoin HK", className="ms-2")),
                    ],
                    align="center",
                ),
                href="https://bitcoin.org.hk",
                style={"textDecoration": "none"},
            ),
            dbc.NavbarToggler(id="navbar-toggler", n_clicks=0),
            dbc.Collapse(
                items_bar,
                id="navbar-collapse",
                is_open=False,
                navbar=True,
            ),
        ],
    ),
    color="dark",
    dark=True,
)


# add callback for toggling the collapse on small screens
@app.callback(
    Output("navbar-collapse", "is_open"),
    [Input("navbar-toggler", "n_clicks")],
    [State("navbar-collapse", "is_open")],
)
def toggle_navbar_collapse(n, is_open):
    if n:
        return not is_open
    return is_open


currency_type = html.Div(
    [
        dbc.Label("Currency: ", className="ms-2"),
        dbc.RadioItems(
            options=[
                {"label": "HKD", "value": "HKD"},
                {"label": "USD", "value": "USD"},
            ],
            value="HKD",
            id="currency",
            inline=True,
            className="mb-3",
        ),
    ],
    className="mt-3 mb-4 mt-md-0",
)


amount_input = html.Div(
    [
        dbc.Label("Enter Amount: ", className="ms-2"),
        dbc.Input(
            size="lg",
            value=100,
            className="mb-3",
            type="number",
            id="amount",
            min=0,
            max=1000000,
            step=1,
        ),
    ]
)


inline_radioitems = html.Div(
    [
        dbc.Label("Frequency: ", className="ms-2"),
        dbc.RadioItems(
            options=[
                {"label": "Daily", "value": "daily"},
                {"label": "Weekly", "value": "weekly"},
                {"label": "Bi-weekly", "value": "bi-weekly"},
                {"label": "Monthly", "value": "monthly"},
            ],
            value="monthly",
            id="freq",
            inline=True,
            className="mb-3",
        ),
    ],
    className="mt-3 mb-4 mt-md-0",
)

date_range = html.Div(
    [
        dbc.Label("Date Range: ", className="ms-2"),
        dcc.DatePickerRange(
            id="date-picker",
            min_date_allowed=dt(2010, 7, 28),
            max_date_allowed=latest_date,
            initial_visible_month=dt(2023, 1, 1),
            start_date=dt(2022, 1, 1),
            end_date=latest_date,
            className="ms-2",
        ),
    ],
    className="mt-3 mb-4 mt-md-0",
)

footer = html.Div(
    [
        html.A(
            "Source",
            href="https://github.com/bitkarrot/dca-calculator",
            style={"textDecoration": "none"},
        )
    ],
    className="mb-4",
)


collapse = html.Div(
    [
        dbc.Collapse([
                    html.P(
                        "Find out how many Sats you can Stack with this Dollar Cost Average (DCA) calculator.",
                        className="text-white",
                    ),
                    amount_input,
                    currency_type,
                    inline_radioitems,
                    date_range,
                ],
            id="collapse",
            is_open=True,
        ),
                dbc.Button(
            "Show/Hide Calculator",
            id="collapse-button",
            className="mb-1",
            color="primary",
            n_clicks=0,
        ),

    ], className="text-white",
)

             


app.layout = dbc.Container(
    [
        navbar,
        dbc.Card(
            [
                dbc.Container(
                    [
                        html.Div(
                            [
                                html.H1(
                                    "DCA Calculator", className="display-3 text-warning"
                                ),
                                # html.P(
                                #     "Find out how many Sats you can Stack with this Dollar Cost Average (DCA) calculator.",
                                #     className="text-white",
                                # ),
                            ],
                            className="mt-4 mb-4",
                        ),
                        html.Div(
                            [
                                collapse,
                            ],
                            className="text-white p-3 bg-primary bg-opacity-10",
                        ),
                        html.Div(
                            [
                                html.Div(id="stacked", className="text-warning"),
                                dcc.Markdown(),
                                dcc.Graph(id="graph", config={"displayModeBar": False}),
                            ],
                            className="p-3",
                        ),
                        footer,
                    ]
                )
            ],
            className="",
        ),
    ],
    fluid=True,
    className="dbc",
)


@app.callback(
    Output("collapse", "is_open"),
    [Input("collapse-button", "n_clicks")],
    [State("collapse", "is_open")],
)
def toggle_collapse(n, is_open):
    if n:
        return not is_open
    return is_open


@app.callback(
    Output("graph", "figure"),
    Output("stacked", "children"),
    Input("amount", "value"),
    Input("currency", "value"),
    Input("freq", "value"),
    Input("date-picker", "start_date"),
    Input("date-picker", "end_date"),
)
def display_area(amount, currency, freq, start_date, end_date):
    try:
        # print(amount, currency, freq, start_date, end_date)

        if amount is None:
            raise Exception("Amount is none")

        # date range
        f_df = df[(df.index >= start_date) & (df.index <= end_date)]

        # filter by frequency
        if freq == "weekly":
            f_df = f_df.resample("W").last()
        elif freq == "bi-weekly":
            f_df = f_df.resample("2W").last()
        elif freq == "monthly":
            f_df = f_df.resample("M").last()

        # currency type
        currency_col = "usdsat_rate"
        if currency == "HKD":
            currency_col = "sathkd_rate"

        dfs = f_df[[currency_col]].copy()
        dfs["Sats per freq"] = dfs[currency_col] * amount
        dfs["Sats Stacked"] = dfs["Sats per freq"].cumsum()

        total_value = dfs["Sats per freq"].sum()
        btc_total = total_value / 100000000

        fig = px.area(dfs, x=dfs.index, y="Sats Stacked", template=template_type)

        # update the line color
        # fig.update_traces(line_color="orange")

        # content info
        stacker_info = (
            "### You stacked a total of "
            + str(format(total_value, ","))
            + " sats or "
            + str(btc_total)
            + " BTC \n\n"
        )
        stacker_info = (
            stacker_info
            + "with "
            + str(amount)
            + " "
            + str(currency)
            + " "
            + str(freq)
            + " dollar cost averaging "
        )
        stacker_info = (
            stacker_info
            + " from "
            + start_date.split("T")[0]
            + " to "
            + end_date.split("T")[0]
        )
        return [fig, dcc.Markdown(stacker_info)]
    except Exception as e:
        raise PreventUpdate


if __name__ == "__main__":
    app.run_server(debug=True)