Code toggle button for IPython notebook

code_toggle.py

'''Creates a button to toggle code inputs on and off.'''
from IPython.core.display import display_html

di.display_html('''
    <script>
      'use strict';
      function code_toggle() {
        let code = $('div.input');
        code.css('display') == 'none' ? code.show() : code.hide();
      }
      $(document).ready(function() {
        $('div.input').hide();
      });
    </script>
''', raw=True)

di.display_html('<button onclick="code_toggle()">Toggle Code</button>', raw=True)