Minor

app.py

@@ -165,7 +165,12 @@ if __name__ == "__main__":
         num_foley_segments = np.sum(num_foley_per_clip)
         data = [d for d, n in zip(data, num_foley_per_clip) if n > 0]
         
+        # get movie titles
+        df = pd.read_csv(csv_path)
+        titles = df["title"].values
+        
         # store variables
+        st.session_state.titles = titles
         st.session_state.video_ids = video_ids
         st.session_state.data = data
         st.session_state.num_foley_segments = num_foley_segments
@@ -190,7 +195,7 @@ if __name__ == "__main__":
         ) for i in range(len(annot["non_speech_segments"]))
     ]
     segments, durations = annot["non_speech_segments"], annot["duration"]
-    movie = annot["title"]
+    movie = st.session_state.titles[index]
 
 
     st.markdown(f"Showing Foley segments from a clip in movie: **{movie}**")