fixed the download on refresh

Author: yankev

plotly/offline/offline.py

@@ -238,31 +238,27 @@ def iplot(figure_or_data, show_link=True, link_text='Export to plot.ly',
 
     display(HTML(plot_html))
 
-    # Use the plot id to download the image now:
-
-    script = ('<script id=\'download-script-{plot_id}\'>'
-              'var completed = document.getElementById(\'downloaded-plot-{plot_id}\');'
-              'alert(completed);'
-              'if(!completed) {{'
-              'console.log(completed);'
-              'var div_id = document.getElementById(\'{plot_id}\');'
-              'setTimeout(function() {{Plotly.downloadImage(div_id, {{format: \'{format}\', '
-              'height: {height}, width: {width}, filename: \'{filename}\'}})'
-              '.then(function(){{'
-              'var complete = document.createElement(\'span\');'
-              'complete.id = \'downloaded-plot-{plot_id}\';'
-              'var thisScript = document.getElementById(\'download-script-{plot_id}\');'
-              'console.log(complete);'
-              'thisScript.parentNode.insertBefore(complete, thisScript.nextSibling);'
-              '}})'
-              ';}}, 500);'
+    # Write script to download image of the plot
+    script = ('<script>'
+              'function downloadimage(format, height, width,'
+              ' filename) {{'
+              'var p = document.getElementById(\'{plot_id}\');'
+              'Plotly.downloadImage(p, {{format: format, height: height, '
+              'width: width, filename: filename}});'
+              '}};'
+              'if(document.readyState == \'complete\') {{'
+              'downloadimage(\'{format}\', {height}, {width}, \'{filename}\');'
               '}}'
               '</script>'
               ).format(format=format, width=_width, height=_height,
                        filename=filename, plot_id=plot_id)
-
+    # allow time for the plot to draw
+    import time
+    time.sleep(1)
+    #inject code to download an image of the plot
     display(HTML(script))
 
+
 def plot(figure_or_data,
          show_link=True, link_text='Export to plot.ly',
          validate=True, output_type='file',