Fix script output

Author: shuds13

agentic/web_ui/gradio_chat.py

@@ -367,7 +367,7 @@ def reset_ui():
                  settings_visible, settings_modal]
     )
 
-    # Run button: start script → stream output → refresh versions
+    # Run button: start script → stream output → refresh versions → load scripts
     run_btn.click(
         start_run,
         inputs=[agent_dropdown, scripts_dropdown, chatbot, agent_dir_state, scripts_dir_state],
@@ -376,6 +376,9 @@ def reset_ui():
         stream_output, inputs=[chatbot], outputs=[chatbot]
     ).then(
         refresh_versions, inputs=[agent_dir_state], outputs=[version_dropdown]
+    ).then(
+        load_version_scripts, inputs=[version_dropdown, agent_dir_state],
+        outputs=[scripts_dict, script_file_dropdown, output_script]
     )
 
     # Chat input: send to stdin → stream continued output
@@ -385,13 +388,19 @@ def reset_ui():
         stream_output, inputs=[chatbot], outputs=[chatbot]
     ).then(
         refresh_versions, inputs=[agent_dir_state], outputs=[version_dropdown]
+    ).then(
+        load_version_scripts, inputs=[version_dropdown, agent_dir_state],
+        outputs=[scripts_dict, script_file_dropdown, output_script]
     )
     chat_input.submit(
         send_user_input, inputs=[chat_input, chatbot], outputs=[chat_input, chatbot]
     ).then(
         stream_output, inputs=[chatbot], outputs=[chatbot]
     ).then(
         refresh_versions, inputs=[agent_dir_state], outputs=[version_dropdown]
+    ).then(
+        load_version_scripts, inputs=[version_dropdown, agent_dir_state],
+        outputs=[scripts_dict, script_file_dropdown, output_script]
     )
 
     # Reset