fixes

app.py

@@ -30,6 +30,7 @@ SANDBOXES = {}
 SANDBOX_TIMEOUT = 300
 TMP_DIR = './tmp/'
 model="Qwen/Qwen3-Coder-480B-A35B-Instruct:cerebras"
+init_notebook = JupyterNotebook()
 
 if not os.path.exists(TMP_DIR):
     os.makedirs(TMP_DIR)
@@ -48,6 +49,7 @@ The following files are available (if any):
 def execute_jupyter_agent(
     user_input, files, message_history, request: gr.Request
 ):
+
     if request.session_hash not in SANDBOXES:
         SANDBOXES[request.session_hash] = Sandbox(api_key=E2B_API_KEY, timeout=SANDBOX_TIMEOUT)
     sbx = SANDBOXES[request.session_hash]
@@ -56,6 +58,10 @@ def execute_jupyter_agent(
     os.makedirs(save_dir, exist_ok=True)
     save_dir = os.path.join(save_dir, 'jupyter-agent.ipynb')
 
+    with open(save_dir, 'w', encoding='utf-8') as f:
+        json.dump(init_notebook.data, f, indent=2)
+    yield init_notebook.render(), message_history, save_dir
+
     client = OpenAI(
         base_url="https://router.huggingface.co/v1",
         api_key=HF_TOKEN,
@@ -99,9 +105,13 @@ def execute_jupyter_agent(
         json.dump(notebook_data, f, indent=2)
     yield notebook_html, message_history, save_dir
 
-def clear(msg_state):
+def clear(msg_state, request: gr.Request):
+    if request.session_hash in SANDBOXES:
+        SANDBOXES[request.session_hash].kill()
+        SANDBOXES.pop(request.session_hash)
+
     msg_state = []
-    return JupyterNotebook().render(), msg_state
+    return init_notebook.render(), msg_state
 
 
 css = """
@@ -142,7 +152,7 @@ with gr.Blocks() as demo:
 
     powered_html = gr.HTML("""\
         <p align="center">
-        <img style="height:100px;"src="https://huggingface.co/spaces/lvwerra/jupyter-agent-2/resolve/main/powered-by.png" alt="Powered by" />
+             <img style="max-height:100px; max-width:100%; height:auto;"src="https://huggingface.co/spaces/lvwerra/jupyter-agent-2/resolve/main/powered-by.png" alt="Powered by" />
         </p>""")
     
 
@@ -150,6 +160,7 @@ with gr.Blocks() as demo:
         fn=execute_jupyter_agent,
         inputs=[user_input, files, msg_state],
         outputs=[html_output, msg_state, file],
+        show_progress="hidden",
     )
 
     clear_btn.click(fn=clear, inputs=[msg_state], outputs=[html_output, msg_state])

jupyter_handler.py

@@ -56,10 +56,10 @@ assistant_final_answer_template = """<div class="alert alert-block alert-warning
 """
 
 header_message = """<p align="center">
-  <img style="height:120px;"src="https://huggingface.co/spaces/lvwerra/jupyter-agent-2/resolve/main/jupyter-agent-2.png" alt="Jupyter Agent Logo" />
+  <img style="max-height:120px; max-width:100%; height:auto;" 
+       src="https://huggingface.co/spaces/lvwerra/jupyter-agent-2/resolve/main/jupyter-agent-2.png" 
+       alt="Jupyter Agent Logo" />
 </p>
-
-
 <p style="text-align:center;">Running Qwen3-Coder-480B-A35B-Instruct in a Jupyter notebook powered by E2B, Cerebras and Hugging Face Inference Providers.</p>"""
 
 bad_html_bad = """input[type="file"] {