import streamlit as st
from IPython.display import display
from solidipes.viewers import backends as viewer_backends
from solidipes.viewers.viewer import Viewer
[docs]
class Video(Viewer):
"""Viewer for videos"""
def __init__(self, data=None):
#: Image to display
self.video = None
super().__init__(data)
[docs]
def add(self, data_container):
"""Replace the viewer's image"""
self.check_data_compatibility(data_container)
self.video = data_container.video
[docs]
def show(self):
if viewer_backends.current_backend == "jupyter notebook":
display(self.image)
elif viewer_backends.current_backend == "streamlit":
with st.container():
st.video(self.video)
else: # python
self.image.show()