From c2ce42121b727d0aebb9f1fcd5973bafcd4a5100 Mon Sep 17 00:00:00 2001
From: "Brian E. Granger" <ellisonbg@gmail.com>
Date: Thu, 11 Aug 2011 14:51:53 -0700
Subject: [PATCH] %loadpy works in the notebook and bug with inline plotting
 fixed.

---
 .../html/notebook/static/js/notebook.js        | 18 +++++++++++-------
 1 file changed, 11 insertions(+), 7 deletions(-)

diff --git a/IPython/frontend/html/notebook/static/js/notebook.js b/IPython/frontend/html/notebook/static/js/notebook.js
index f415dde45..70c40e630 100644
--- a/IPython/frontend/html/notebook/static/js/notebook.js
+++ b/IPython/frontend/html/notebook/static/js/notebook.js
@@ -477,18 +477,22 @@ var IPython = (function (IPython) {
             cell.finish_completing(content.matched_text, content.matches);
         };
         var payload = content.payload || [];
-        this.handle_payload(payload);
+        this.handle_payload(cell, payload);
     };
 
 
-    Notebook.prototype.handle_payload = function (payload) {
+    Notebook.prototype.handle_payload = function (cell, payload) {
         var l = payload.length;
-        if (l > 0) {
-            IPython.pager.clear();
-            IPython.pager.expand();
-        };
         for (var i=0; i<l; i++) {
-            IPython.pager.append_text(payload[i].text);
+            if (payload[i].source === 'IPython.zmq.page.page') {
+                IPython.pager.clear();
+                IPython.pager.expand();
+                IPython.pager.append_text(payload[i].text);
+            } else if (payload[i].source === 'IPython.zmq.zmqshell.ZMQInteractiveShell.set_next_input') {
+                var index = this.find_cell_index(cell);
+                var new_cell = this.insert_code_cell_after(index);
+                new_cell.set_code(payload[i].text);
+            }
         };
     };