strength-studies/visualize.py at master · AnyBody/strength-studies · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
import marimo

__generated_with = "0.19.4"
app = marimo.App(
    width="medium",
    app_title="Strength Viewer",
    auto_download=["html"],
)


@app.cell(hide_code=True)
def _(df, fig, mo, ui_selected_second_dof, ui_selected_study):
    mo.vstack(
        [ui_selected_study, ui_selected_second_dof, mo.ui.plotly(fig), mo.ui.table(df, page_size=5)]
    )
    return


@app.cell
def _():
    import marimo as mo
    import plotly.express as px
    import plotly.graph_objects as go
    import polars as pd
    import polars as pl
    return mo, pd, pl, px


@app.cell
def _(pl):
    source = "abfs://strength-data@anybodydatasets.dfs.core.windows.net/"
    storage_options = {
        "sas_token": r"sp=r&st=2026-01-22T09:04:53Z&se=2028-08-10T16:19:53Z&spr=https&sv=2024-11-04&sr=c&sig=DSoRzE2CnJShYZmVs%2BhbWXPSyVk36XnNmdqswjtkFBI%3D"
    }


    # def get_data():
    #     df = pl.scan_parquet(
    #         source + f"*.parquet", storage_options=storage_options, include_file_paths="file"
    #     ).filter(pl.col("file") == pl.col("file").max())
    #     return df.collect()


    # df_full = get_data()
    df_full = pd.read_parquet("joint-strength-full.parquet")

    # df_full
    return (df_full,)


@app.cell
def _(df_full, mo):
    ui_selected_study = mo.ui.dropdown(
        options=df_full["measureObject"].unique().sort().to_list(),
        value=df_full["measureObject"].unique().sort().to_list()[0],
        label="Select Joint Measurement:",
    )
    return (ui_selected_study,)


@app.cell
def _(df_study, mo):
    ui_selected_second_dof = mo.ui.dropdown(
        options=df_study["measureSecondDoF"].unique().sort().to_list(),
        value=0,
        label=f"Second DOF: {df_study['secondaryDoF'].first()}",
    )
    ui_selected_second_dof = mo.ui.dropdown(
        options=df_study["measureSecondDoF"].unique().sort().to_list(),
        value=0,
        label=f"Second DOF: {df_study['secondaryDoF'].first()}",
    )
    return (ui_selected_second_dof,)


@app.cell(hide_code=True)
def _(df_full, pl, ui_selected_study):
    df_study = df_full.filter(pl.col("measureObject") == ui_selected_study.value)
    return (df_study,)


@app.cell
def _(df_study, pl, ui_selected_second_dof):
    df = df_study.filter(pl.col("measureSecondDoF") == ui_selected_second_dof.value)
    # df
    return (df,)


@app.cell
def _(df, px):
    fig = px.line(df, x="measurePrimaryDoF", y="measureValue", color="AnyBodyMuscleType")

    fig.update_layout(
        xaxis_title=df["primaryDoF"].unique()[0],
        yaxis_title="Strength",
        title=f"{df['measureObject'].first()} Strength",
    )
    return (fig,)


@app.cell
def _(df_measured_all):
    df_measured_all
    return


@app.cell
def _(mo, pd):
    df_measured_all = pd.read_parquet("convert-matlab-data/isometric.parquet")
    ui_selected_measured = mo.ui.dropdown(
        options=df_measured_all["measureObject"].unique().sort().to_list(),
        value=df_measured_all["measureObject"].first(),
        label="Select Joint Measurement:",
    )
    # ui_selected_measured
    # df_measured_all
    return df_measured_all, ui_selected_measured


@app.cell
def _(df_measured_all, pl, ui_selected_measured):
    df_measured = df_measured_all.filter(pl.col("measureObject") == ui_selected_measured.value)
    # df_measured
    return


if __name__ == "__main__":
    app.run()