code: def screenshot(
      width=None,
      height=None,
      format="png",
      fig=None,
      timeout_seconds=10,
      output_widget=None,
      headless=False,
      devmode=False,
  ):
      
      assert format in ['png', 'jpeg', 'svg'], "image format must be png, jpeg or svg"
      data = _screenshot_data(
          timeout_seconds=timeout_seconds,
          output_widget=output_widget,
          format=format,
          width=width,
          height=height,
          fig=fig,
          headless=headless,
          devmode=devmode,
      )
      f = StringIO(data)
      return PIL.Image.open(f)
docs: Save the figure to a PIL.Image object.
  
      :param int width: the width of the image in pixels
      :param int height: the height of the image in pixels
      :param format: format of output data (png, jpeg or svg)
      :type fig: ipyvolume.widgets.Figure or None
      :param fig: if None use the current figure
      :type timeout_seconds: int
      :param timeout_seconds: maximum time to wait for image data to return
      :type output_widget: ipywidgets.Output
      :param output_widget: a widget to use as a context manager for capturing the data
      :param bool headless: if True, use headless chrome to take screenshot
      :param bool devmode: if True, attempt to get index.js from local js/dist folder
      :return: PIL.Image