Fixed outputs index check

Author: 3coins

jupyter_server_documents/outputs/manager.py

@@ -120,17 +120,14 @@ def write(self, file_id, cell_id, output, display_id=None):
         return placeholder
 
     def write_output(self, file_id, cell_id, output, display_id=None):
-        print(f"display_id:{display_id} passed into write_output")
         self._ensure_path(file_id, cell_id)
         last_index = self._last_output_index.get(cell_id, -1)
         if display_id:
             index = self._output_index_by_display_id.get(display_id)
-            print(f"Found index: {index} for display_id {display_id}")
             if index is None:
                 index = last_index + 1
                 self._last_output_index[cell_id] = index
                 self._output_index_by_display_id[display_id] = index
-                print(f"Generated new index: {index} for display_id: {display_id}")
         else:
             index = last_index + 1
             self._last_output_index[cell_id] = index

jupyter_server_documents/outputs/output_processor.py

@@ -107,7 +107,6 @@ async def output_task(self, msg_type, cell_id, content):
         self._file_id = file_id
 
         display_id = content.get("transient", {}).get("display_id")
-        self.log.info(f"display_id: {display_id} found in content")
         output_index = self.outputs_manager.get_output_index(display_id) if display_id else None
         # Convert from the message spec to the nbformat output structure
         if self.use_outputs_service: 
@@ -128,7 +127,7 @@ async def output_task(self, msg_type, cell_id, content):
         # Write the outputs to the ydoc cell.
         _, target_cell = notebook.find_cell(cell_id)
         if target_cell is not None and output is not None:
-            if output_index:
+            if output_index is not None:
                 target_cell["outputs"][output_index] = output
             else:
                 target_cell["outputs"].append(output)

src/notebook-factory/notebook-factory.ts

@@ -38,7 +38,6 @@ const DIRTY_CLASS = 'jp-mod-dirty';
   }
 
   if (change.outputsChange) {
-    console.trace(change.outputsChange);
     globalModelDBMutex(() => {
       let retain = 0;
       for (const outputsChange of change.outputsChange!) {