display code functionality is moved to the extension

Author: OutlyingWest

src/jumper/kernel.py

@@ -954,41 +954,7 @@ async def do_execute(
                 allow_stdin,
                 cell_id=cell_id,
             )
-        elif code.startswith("%%display_code_for_index"):
-            if len(code.split(" ")) == 1:
-                self.cell_output(
-                    "No index specified. Use: %%display_code_for_index index",
-                    "stdout",
-                )
-            index = int(code.split(" ")[1])
-            if index >= len(kernel_context.perfdata_handler.get_perfdata_history()):
-                self.cell_output(
-                    "Tracked only "
-                    + str(len(kernel_context.perfdata_handler.get_perfdata_history()))
-                    + " cells. This index is not available."
-                )
-            else:
-                self.cell_output(
-                    "Cell timestamp: "
-                    + str(kernel_context.perfdata_handler.get_code_history()[index][0])
-                    + "\n--\n",
-                    "stdout",
-                )
-                self.cell_output(
-                    kernel_context.perfdata_handler.get_code_history()[index][1],
-                    "stdout",
-                )
-            return self.standard_reply()
-        elif code.startswith("%%display_code_history"):
-            show(
-                pd.DataFrame(
-                    kernel_context.perfdata_handler.get_code_history(),
-                    columns=["timestamp", "code"],
-                ).reset_index(),
-                layout={"topStart": "search", "topEnd": None},
-                columnDefs=[{"className": "dt-left", "targets": 2}],
-            )
-            return self.standard_reply()
+
         elif code.startswith("%%perfdata_to_variable"):
             if len(code.split(" ")) == 1:
                 self.cell_output(

src/magic_extension/magic.py

@@ -1,9 +1,11 @@
 import sys
 
+import pandas as pd
+from itables import show
 from IPython.core.magic import (Magics, magics_class, line_magic, cell_magic)
 
-from jumper.context import kernel_context, KernelMode
 
+from jumper.context import kernel_context, KernelMode
 import jumper.visualization as perfvis
 
 
@@ -68,7 +70,7 @@ def display_graph_for_index(self, line):
         if time_indices:
             sub_idxs = [x[0] for x in time_indices[0]]
             self.cell_output(
-                f"📈 Cell seemed to be tracked in multi-cell mode. "
+                f"Cell seemed to be tracked in multi-cell mode. "
                 f"Got performance data for the following sub-cells: {sub_idxs}"
             )
 
@@ -99,6 +101,46 @@ def display_graph_for_last(self, line):
         )
 
 
+    @line_magic
+    def display_code_for_index(self, line):
+        """
+         Display stored source code of a previously executed cell by index.
+         Usage:
+             %%display_code_for_index 2
+         """
+        if not line.strip():
+            self.cell_output("No index specified. Use: %%display_code_for_index <index>", stream="stdout")
+            return
+
+        try:
+            index = int(line.strip())
+        except ValueError:
+            self.cell_output("Invalid index format. Provide an integer.", stream="stderr")
+            return
+
+        history = kernel_context.perfdata_handler.get_perfdata_history()
+        if index >= len(history):
+            self.cell_output(
+                f"Tracked only {len(history)} cells. This index is not available.",
+                stream="stdout"
+            )
+            return
+
+        timestamp, code = kernel_context.perfdata_handler.get_code_history()[index]
+        self.cell_output(f"Cell timestamp: {timestamp}\n--", stream="stdout")
+        self.cell_output(code, stream="stdout")
+
+    @line_magic
+    def display_code_history(self, line):
+        show(
+            pd.DataFrame(
+                kernel_context.perfdata_handler.get_code_history(),
+                columns=["timestamp", "code"],
+            ).reset_index(),
+            layout={"topStart": "search", "topEnd": None},
+            columnDefs=[{"className": "dt-left", "targets": 2}],
+        )
+
     @cell_magic
     def set_perfmonitor(self, line, code):
         """