show prompt and target separately

app.py

@@ -20,22 +20,32 @@ def main():
         label="Record ID",
         min_value=0,
         max_value=len(dataset) - 1,
-        value=123,
+        value=43,
     )
 
     record = dataset[record_idx]
     piece = MidiPiece.from_huggingface(record)
+    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,
     )
-    st.json(piece.source)
 
 
 if __name__ == "__main__":