Update app.py

app.py

@@ -80,7 +80,10 @@ async def callback(content: str, user: str, instance: pn.chat.ChatInterface):
     if llm_code.splitlines()[-1].strip() != "fig":
         llm_code += "\nfig"
     code_editor.value = llm_code
-    matplotlib_pane.object = exec_with_return(llm_code)
+
+
+def update_plot(event):
+    matplotlib_pane.object = exec_with_return(event.new)
 
 
 # instantiate widgets and panes
@@ -107,6 +110,9 @@ code_editor = pn.widgets.CodeEditor(
     sizing_mode="stretch_both",
 )
 
+# watch for code changes
+code_editor.param.watch(update_plot, "value")
+
 # lay them out
 tabs = pn.Tabs(
     ("Plot", matplotlib_pane),