Balloons

videogenic.py

@@ -564,8 +564,9 @@ if st.session_state.progress == 2:
   # st.write(df_all)
   # highlight_length = 7.033
   # st.write(st.session_state.fps)
-  selection = altair_component(draw_chart(df_all, st.session_state.mode))
-  print(selection)
+  with st.spinner('Visualizing highlight scores...'):
+    selection = altair_component(draw_chart(df_all, st.session_state.mode))
+    print(selection)
   if 'selection' not in st.session_state:
     st.session_state.selection = selection
   st.session_state.selection = selection
@@ -595,4 +596,5 @@ if st.session_state.progress == 2:
       left = st.session_state.selection['x'][0]
       right = st.session_state.selection['x'][1]
       with st.spinner('Generating highlight video...'):
-        crop_video(st.session_state.df_all, left, right)
+        crop_video(st.session_state.df_all, left, right)
+  st.balloons()