Update app.py

app.py

@@ -240,7 +240,8 @@ def page_2():
             st.markdown("### Videos")
             frame = inputs["frame"]
             if frame:
-                video_path = f"videos_flattened/{os.path.basename(frame).split('/')[0]}.mp4"
+                directory_name = frame.split('/')[1] 
+                video_path = f"videos_flattened/{directory_name}.mp4"
                 if os.path.exists(video_path):
                     st.video(video_path)
                 else: