cleanup

README.md

@@ -5,7 +5,7 @@ colorFrom: purple
 colorTo: indigo
 sdk: streamlit
 sdk_version: 1.39.0
-app_file: app_v2.py
+app_file: app.py
 pinned: false
 license: mit
 short_description: Demo of masking tasks from the PIANO dataset

app.py

@@ -1,54 +1,5 @@
-import streamlit as st
-import streamlit_pianoroll
-from datasets import load_dataset
-from piano_dataset import PianoTasks
-from fortepyan import MidiPiece
-
-
-def main():
-    st.write("# PIANO dataset task review")
-
-    available_tasks = PianoTasks.list_tasks()
-    task_name = st.selectbox(
-        label="Select PIANO task",
-        options=available_tasks,
-    )
-    piano_task = PianoTasks.get_task(task_name=task_name)
-
-    dataset = load_dataset("epr-labs/maestro-sustain-v2", split="test+validation")
-    record_idx = st.number_input(
-        label="Record ID",
-        min_value=0,
-        max_value=len(dataset) - 1,
-        value=43,
-    )
-
-    record = dataset[record_idx]
-    piece = MidiPiece.from_huggingface(record)
-    piece.time_shift(-piece.df.start.min())
-
-    st.json(piece.source)
-
-    piece_split = piano_task.prompt_target_split(piece.df)
-
-    source_piece = MidiPiece(df=piece_split.source_df)
-    target_piece = MidiPiece(df=piece_split.target_df)
-
-    cols = st.columns(2)
-    with cols[0]:
-        st.write("## Prompt")
-        streamlit_pianoroll.from_fortepyan(source_piece)
-
-    with cols[1]:
-        st.write("## Target")
-        streamlit_pianoroll.from_fortepyan(target_piece)
-
-    st.write("## Combined")
-    streamlit_pianoroll.from_fortepyan(
-        piece=source_piece,
-        secondary_piece=target_piece,
-    )
+from piano_dataset.dashboards import task_review
 
 
 if __name__ == "__main__":
-    main()
+    task_review.main()

app_v2.py

@@ -1,5 +0,0 @@
-from piano_dataset.dashboards import task_review
-
-
-if __name__ == "__main__":
-    task_review.main()