from IPython.display import HTML

HTML('''<script>
     code_show=true; 
     function code_toggle() {
      if (code_show){
       {% math_inline %}('div.input').hide();
       } else {
        {% endmath_inline %}('div.input').show();
        }
      code_show = !code_show
     } 
     $( document ).ready(code_toggle);
     </script>
     <form action="javascript:code_toggle()"><input type="submit" value="Click here to toggle on/off the raw code."></form>''')