Inject process pk to url on submission

qe.ipynb

@@ -56,8 +56,6 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from IPython.display import display\n",
-    "\n",
     "from aiidalab_qe.app.wrapper import AppWrapperContoller, AppWrapperModel, AppWrapperView\n",
     "\n",
     "model = AppWrapperModel()\n",

src/aiidalab_qe/app/submission/model.py

@@ -4,6 +4,7 @@
 
 import ipywidgets as ipw
 import traitlets as tl
+from IPython.display import Javascript, display
 
 from aiida import orm
 from aiida.engine import ProcessBuilderNamespace, submit
@@ -91,9 +92,13 @@ def update_process_label(self):
             {"properties": []},
         )
 
-        soc_parameters = self.input_parameters["advanced"]["pw"]["parameters"][
-            "SYSTEM"
-        ].get("lspinorb", False)
+        soc_parameters = (
+            self.input_parameters.get("advanced", {})
+            .get("pw", {})
+            .get("parameters", {})
+            .get("SYSTEM", {})
+            .get("lspinorb", False)
+        )
 
         soc_info = "spin-orbit coupling" if soc_parameters else ""
 
@@ -231,6 +236,12 @@ def _submit(self):
             )
             self.process_node = process_node
 
+            self._update_url()
+
+    def _update_url(self):
+        pk = self.process_node.pk
+        display(Javascript(f"window.history.pushState(null, '', '?pk={pk}');"))
+
     def _link_model(self, model: ResourceSettingsModel):
         for dependency in model.dependencies:
             dependency_parts = dependency.split(".")