ivoryos 0.1.20__tar.gz → 0.1.21__tar.gz

{ivoryos-0.1.20/ivoryos.egg-info → ivoryos-0.1.21}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ivoryos
-Version: 0.1.20
+Version: 0.1.21
 Summary: an open-source Python package enabling Self-Driving Labs (SDLs) interoperability
 Home-page: https://gitlab.com/heingroup/ivoryos
 Author: Ivory Zhang

{ivoryos-0.1.20 → ivoryos-0.1.21}/ivoryos/__init__.py

@@ -1,10 +1,8 @@
-import importlib
-import inspect
 import os
 import sys
 from typing import Union
 
-from flask import Flask, redirect, url_for, Blueprint, g
+from flask import Flask, redirect, url_for, g
 
 from ivoryos.config import Config, get_config
 from ivoryos.routes.auth.auth import auth, login_manager
@@ -23,6 +21,11 @@ global_config = GlobalConfig()
 
 url_prefix = os.getenv('URL_PREFIX', "/ivoryos")
 app = Flask(__name__, static_url_path=f'{url_prefix}/static', static_folder='static')
+app.register_blueprint(main, url_prefix=url_prefix)
+app.register_blueprint(auth, url_prefix=url_prefix)
+app.register_blueprint(control, url_prefix=url_prefix)
+app.register_blueprint(design, url_prefix=url_prefix)
+app.register_blueprint(database, url_prefix=url_prefix)
 
 
 def create_app(config_class=None):
@@ -86,14 +89,6 @@ def run(module=None, host="0.0.0.0", port=None, debug=None, llm_server=None, mod
     """
     app = create_app(config_class=config or get_config())  # Create app instance using factory function
 
-    app.register_blueprint(main, url_prefix=url_prefix)
-    app.register_blueprint(auth, url_prefix=url_prefix)
-    app.register_blueprint(control, url_prefix=url_prefix)
-
-    if enable_design:
-        app.register_blueprint(design, url_prefix=url_prefix)
-        app.register_blueprint(database, url_prefix=url_prefix)
-
     plugins = load_plugins(app, socketio)
 
     def inject_nav_config():
@@ -156,3 +151,5 @@ def load_plugins(app, socketio):
         app.register_blueprint(getattr(plugin, entry_point.name), url_prefix=f"{url_prefix}/{entry_point.name}")
 
     return plugin_names
+
+

{ivoryos-0.1.20 → ivoryos-0.1.21}/ivoryos/routes/database/database.py

@@ -176,10 +176,12 @@ def save_as():
     """
     if request.method == "POST":
         run_name = request.form.get("run_name")
+        register_workflow = request.form.get("register_workflow")
         exist_script = Script.query.get(run_name)
         if not exist_script:
             script = get_script_file()
             script.save_as(run_name)
+            script.registered = register_workflow == "on"
             script.author = session.get('user')
             post_script_file(script)
             publish()

{ivoryos-0.1.20 → ivoryos-0.1.21}/ivoryos/routes/database/templates/database/experiment_database.html

@@ -26,10 +26,11 @@
             <tr>
                 <th scope="col">Workflow name</th>
                 <th scope="col">Deck </th>
-                <th scope="col">Current status</th>
+                <th scope="col">Editing</th>
                 <th scope="col">Time created</th>
                 <th scope="col">Last modified</th>
                 <th scope="col">Author</th>
+                <th scope="col">Registered</th>
                 <th scope="col"></th>
             </tr>
         </thead>
@@ -42,6 +43,7 @@
                     <td>{{ workflow.time_created }}</td>
                     <td>{{ workflow.last_modified }}</td>
                     <td>{{ workflow.author }}</td>
+                    <td>{{ workflow.registered }}</td>
                     <td>
                         {#not workflow.status == "finalized" or#}
                         {% if session['user'] == 'admin' or session['user'] == workflow.author %}

{ivoryos-0.1.20 → ivoryos-0.1.21}/ivoryos/routes/design/design.py

@@ -13,9 +13,10 @@ from werkzeug.utils import secure_filename
 from ivoryos.utils import utils
 from ivoryos.utils.global_config import GlobalConfig
 from ivoryos.utils.form import create_builtin_form, create_action_button, format_name, create_form_from_pseudo, \
-    create_form_from_action
+    create_form_from_action, create_all_builtin_forms
 from ivoryos.utils.db_models import Script
 from ivoryos.utils.script_runner import ScriptRunner
+# from ivoryos.utils.utils import load_workflows
 
 socketio = SocketIO()
 design = Blueprint('design', __name__, template_folder='templates/design')
@@ -92,6 +93,9 @@ def experiment_builder(instrument=None):
     """
     deck = global_config.deck
     script = utils.get_script_file()
+    # load_workflows(script)
+    # registered_workflows = global_config.registered_workflows
+
     if deck and script.deck is None:
         script.deck = os.path.splitext(os.path.basename(deck.__file__))[
             0] if deck.__name__ == "__main__" else deck.__name__
@@ -114,7 +118,10 @@ def experiment_builder(instrument=None):
 
     functions = {}
     if deck:
-        deck_variables = global_config.deck_snapshot.keys()
+        deck_variables = list(global_config.deck_snapshot.keys())
+        # deck_variables.insert(0, "registered_workflows")
+        deck_variables.insert(0, "flow_control")
+
     else:
         deck_variables = list(pseudo_deck.keys()) if pseudo_deck else []
         deck_variables.remove("deck_name") if len(deck_variables) > 0 else deck_variables
@@ -122,23 +129,31 @@ def experiment_builder(instrument=None):
     if edit_action_info:
         forms = create_form_from_action(edit_action_info, script=script)
     elif instrument:
-        if instrument in ['if', 'while', 'variable', 'wait', 'repeat']:
-            forms = create_builtin_form(instrument, script=script)
+        # if instrument in ['if', 'while', 'variable', 'wait', 'repeat']:
+        #     forms = create_builtin_form(instrument, script=script)
+        if instrument == 'flow_control':
+            forms = create_all_builtin_forms(script=script)
+        # elif instrument == 'registered_workflows':
+        #     functions = utils._inspect_class(registered_workflows)
+        #     # forms = create_workflow_forms(script=script)
+        #     forms = create_form_from_pseudo(pseudo=functions, autofill=autofill, script=script)
+        elif instrument in global_config.defined_variables.keys():
+            _object = global_config.defined_variables.get(instrument)
+            functions = utils._inspect_class(_object)
+            forms = create_form_from_pseudo(pseudo=functions, autofill=autofill, script=script)
         else:
             if deck:
                 functions = global_config.deck_snapshot.get(instrument, {})
             elif pseudo_deck:
                 functions = pseudo_deck.get(instrument, {})
-            # print(function_metadata)
-            # functions = {key: data.get('signature', {}) for key, data in function_metadata.items()}
             forms = create_form_from_pseudo(pseudo=functions, autofill=autofill, script=script)
         if request.method == 'POST' and "hidden_name" in request.form:
             # all_kwargs = request.form.copy()
             method_name = request.form.get("hidden_name", None)
             # if method_name is not None:
             form = forms.get(method_name)
+            insert_position = request.form.get("drop_target_id", None)
             kwargs = {field.name: field.data for field in form if field.name != 'csrf_token'}
-
             if form and form.validate_on_submit():
                 function_name = kwargs.pop("hidden_name")
                 save_data = kwargs.pop('return', '')
@@ -151,30 +166,57 @@ def experiment_builder(instrument=None):
                           "args": kwargs,
                           "return": save_data,
                           'arg_types': primitive_arg_types}
-                script.add_action(action=action)
+                script.add_action(action=action, insert_position=insert_position)
             else:
                 flash(form.errors)
 
         elif request.method == 'POST' and "builtin_name" in request.form:
-            kwargs = {field.name: field.data for field in forms if field.name != 'csrf_token'}
-            if forms.validate_on_submit():
+            function_name = request.form.get("builtin_name")
+            form = forms.get(function_name)
+            kwargs = {field.name: field.data for field in form if field.name != 'csrf_token'}
+            insert_position = request.form.get("drop_target_id", None)
+
+            if form.validate_on_submit():
                 # print(kwargs)
                 logic_type = kwargs.pop('builtin_name')
                 if 'variable' in kwargs:
                     try:
-                        script.add_variable(**kwargs)
+                        script.add_variable(**kwargs, insert_position=insert_position)
                     except ValueError:
                         flash("Invalid variable type")
                 else:
-                    script.add_logic_action(logic_type=logic_type, **kwargs)
+                    script.add_logic_action(logic_type=logic_type, **kwargs, insert_position=insert_position)
             else:
-                flash(forms.errors)
+                flash(form.errors)
+        elif request.method == 'POST' and "workflow_name" in request.form:
+            workflow_name = request.form.get("workflow_name")
+            form = forms.get(workflow_name)
+            kwargs = {field.name: field.data for field in form if field.name != 'csrf_token'}
+            insert_position = request.form.get("drop_target_id", None)
+
+            if form.validate_on_submit():
+                # workflow_name = kwargs.pop('workflow_name')
+                save_data = kwargs.pop('return', '')
+
+                primitive_arg_types = utils.get_arg_type(kwargs, functions[workflow_name])
+
+                script.eval_list(kwargs, primitive_arg_types)
+                kwargs = script.validate_variables(kwargs)
+                action = {"instrument": instrument, "action": workflow_name,
+                          "args": kwargs,
+                          "return": save_data,
+                          'arg_types': primitive_arg_types}
+                script.add_action(action=action, insert_position=insert_position)
+                script.add_workflow(**kwargs, insert_position=insert_position)
+            else:
+                flash(form.errors)
 
-        # toggle autofill
+        # toggle autofill, autofill doesn't apply to control flow ops
         elif request.method == 'POST' and "autofill" in request.form:
             autofill = not autofill
-            forms = create_form_from_pseudo(functions, autofill=autofill, script=script)
             session['autofill'] = autofill
+            if not instrument == 'flow_control':
+                forms = create_form_from_pseudo(functions, autofill=autofill, script=script)
 
     utils.post_script_file(script)
     design_buttons = create_action_button(script)
@@ -329,7 +371,7 @@ def experiment_run():
                            return_list=return_list, config_list=config_list, config_file_list=config_file_list,
                            config_preview=config_preview, data_list=data_list, config_type_list=config_type_list,
                            no_deck_warning=no_deck_warning, dismiss=dismiss, design_buttons=design_buttons,
-                           history=deck_list)
+                           history=deck_list, pause_status=runner.pause_status())
 
 
 @design.route("/toggle_script_type/<stype>")

{ivoryos-0.1.20 → ivoryos-0.1.21}/ivoryos/routes/design/templates/design/experiment_builder.html

@@ -2,13 +2,14 @@
 {% block title %}IvoryOS | Design{% endblock %}
 
 {% block body %}
-{#    <div id="overlay" class="overlay">Generating design, please wait...</div>#}
+    {# overlay block for text-to-code gen #}
     <div id="overlay" class="overlay">
         <div>
             <h3 id="overlay-text">Generating design, please wait...</h3>
             <div class="spinner-border" role="status"></div>
         </div>
     </div>
+
     <div class="row">
         <div class="col-md-3 scroll-column" >
 
@@ -30,7 +31,6 @@
 
             {# edit action #}
             {% if session["edit_action"] %}
-{#                {{ session["edit_action"] }}#}
                 {% with action = session["edit_action"] %}
                     <h5> {{ format_name(action['action']) }} </h5>
                     <form role="form" method='POST' name="{{instrument}}" action="{{ url_for('design.edit_action', uuid=session["edit_action"]['uuid']) }}">
@@ -69,91 +69,71 @@
                         <button class="btn btn-primary" type="submit" name="back" id="back" value="back">Back</button>
                     </form>
                 {% endwith %}
+
+
             {% elif instrument %}
                 <div>
                     <div class="d-flex justify-content-between align-items-center " style="margin-bottom: 1vh;margin-top: 1vh;">
-
                         <a class="btn btn-primary" role="button" type="button" href="{{url_for('design.experiment_builder')}}"><i class="bi bi-arrow-return-left"></i></a>
-
                         {{ format_name(instrument) }}
-
-
                     </div>
 
-<!--                    <div class="rounded flex-fill" style="height: 30px;background-color: aliceblue">-->
-<!--                        <h6  style=" text-align: center; ">{{ format_name(instrument) }}</h6>-->
-<!--                    </div>-->
-                    {% if instrument in ['if' , 'while' , 'variable' ,'wait', 'repeat'] %}
-                        {# form for builtin functions #}
-                        <form role="form" method='POST' name="{{instrument}}" action="{{url_for('design.experiment_builder',instrument=instrument)}}" >
-                            <div class="form-group">
-                                {{ forms.hidden_tag() }}
-                                {% for field in forms %}
-                                    {% if field.type not in ['CSRFTokenField', 'HiddenField'] %}
-                                        <div class="input-group mb-3">
-                                            <label class="input-group-text">{{ field.label.text }}</label>
-                                            {{ field(class="form-control") }}
-                                            <div class="form-text">{{ field.description }} </div>
-                                        </div>
-                                    {% endif %}
-                                {% endfor %}
-                            </div>
-                            <button type="submit" class="btn btn-dark">Add </button>
-                        </form>
-                    {% else %}
-                        {% if script.editing_type == "script" %}
-                            {# Auto Fill Toggle #}
-                            <div class="d-flex justify-content-between align-items-center " style="margin-bottom: 1vh;margin-top: 1vh;">
-
-                                <div></div>
-                                <form role="form" method='POST' name="autoFill" id="autoFill">
-                                    <div class="form-check form-switch">
-                                        <input type="hidden" id="autofill" name="autofill" value="temp_value">
-                                        <input class="form-check-input" type="checkbox" id="autoFillCheck" name="autoFillCheck" onchange="document.getElementById('autoFill').submit();"
-                                               value="temp_value"
-                                                {{ "checked" if session["autofill"] else "" }}>
-                                        <label class="form-check-label" for="autoFillCheck">Auto fill</label>
-                                    </div>
-                                    <button type="submit" class="btn btn-default" style="display: none;">Auto fill </button>
-                                </form>
-                            </div>
-                        {% endif %}
+                    {% if script.editing_type == "script" %}
+                        {# Auto Fill Toggle #}
+                        <div class="d-flex justify-content-between align-items-center " style="margin-bottom: 1vh;margin-top: 1vh;">
+                            <div></div>
+                            <form role="form" method='POST' name="autoFill" id="autoFill">
+                                <div class="form-check form-switch">
+                                    <input type="hidden" id="autofill" name="autofill" value="temp_value">
+                                    <input class="form-check-input" type="checkbox" id="autoFillCheck" name="autoFillCheck" onchange="document.getElementById('autoFill').submit();"
+                                           value="temp_value"
+                                            {{ "checked" if session["autofill"] else "" }}>
+                                    <label class="form-check-label" for="autoFillCheck">Auto fill</label>
+                                </div>
+                                <button type="submit" class="btn btn-default" style="display: none;">Auto fill </button>
+                            </form>
+                        </div>
+                    {% endif %}
 
                         {# according for instrument #}
                         <div class="accordion accordion-flush" id="accordionActions" >
-                            {% if use_llm %}
-                            <div class="accordion-item text-to-code">
-                                    <h2 class="accordion-header">
-                                        <button class="accordion-button text-to-code" type="button" data-bs-toggle="collapse" data-bs-target="#text-to-code" aria-expanded="false" aria-controls="collapseExample">
-                                            Text-to-Code
-                                        </button>
-                                    </h2>
-                                    <div id="text-to-code" class="accordion-collapse collapse show" data-bs-parent="#accordionActions">
-                                        <div class="accordion-body">
-                                            <form role="form" method='POST' name="generate" id="generate" action="{{url_for('design.generate_code')}}">
-                                                <input type="hidden" id="instrument" name="instrument" value="{{instrument}}">
-                                                <textarea class="form-control" id="prompt" name="prompt" rows="6" aria-describedby="promptHelpBlock">{{ session['prompt'][instrument] if instrument in session['prompt'] else '' }}</textarea>
-                                                <div id="promptHelpBlock" class="form-text">
-                                                    This will overwrite current design.
-                                                </div>
-<!--                                                <button type="submit" class="btn btn-dark" id="clear" name="clear" onclick="submitForm('generate')">Clear</button>-->
-                                                <button type="submit" class="btn btn-dark" id="gen" name="gen">Generate</button>
-                                            </form>
-                                        </div>
+                            {% if use_llm and not instrument == "flow_control" %}
+                                <div class="accordion-item text-to-code">
+                                <h2 class="accordion-header">
+                                    <button class="accordion-button text-to-code" type="button" data-bs-toggle="collapse" data-bs-target="#text-to-code" aria-expanded="false" aria-controls="collapseExample">
+                                        Text-to-Code
+                                    </button>
+                                </h2>
+                                <div id="text-to-code" class="accordion-collapse collapse show" data-bs-parent="#accordionActions">
+                                    <div class="accordion-body">
+                                        <form role="form" method='POST' name="generate" id="generate" action="{{url_for('design.generate_code')}}">
+                                            <input type="hidden" id="instrument" name="instrument" value="{{instrument}}">
+                                            <textarea class="form-control" id="prompt" name="prompt" rows="6" aria-describedby="promptHelpBlock">{{ session['prompt'][instrument] if instrument in session['prompt'] else '' }}</textarea>
+                                            <div id="promptHelpBlock" class="form-text">
+                                                This will overwrite current design.
+                                            </div>
+<!--                                               <button type="submit" class="btn btn-dark" id="clear" name="clear" onclick="submitForm('generate')">Clear</button>-->
+                                            <button type="submit" class="btn btn-dark" id="gen" name="gen">Generate</button>
+                                        </form>
                                     </div>
+                                </div>
                             </div>
                             {% endif %}
+
                             {% for name, form in forms.items() %}
-                                <div class="accordion-item  design-control">
+                                <div class="accordion-item design-control" draggable="true">
                                     <h2 class="accordion-header">
-                                        <button class="accordion-button collapsed" type="button" data-bs-toggle="collapse" data-bs-target="#{{name}}" aria-expanded="false" aria-controls="collapseExample">
+                                        <button class="accordion-button collapsed draggable-action"
+                                                type="button" data-bs-toggle="collapse"
+                                                data-bs-target="#{{name}}" aria-expanded="false"
+                                                aria-controls="collapseExample"
+                                                 data-action="{{ name }}">
                                             {{ format_name(name) }}
                                         </button>
                                     </h2>
-
                                     <div id="{{name}}" class="accordion-collapse collapse" data-bs-parent="#accordionActions">
                                         <div class="accordion-body">
-                                            <form role="form" method='POST' name="add" id="add">
+                                            <form role="form" method='POST' name="add" id="add-{{name}}">
                                                 <div class="form-group">
                                                     {{ form.hidden_tag() }}
                                                     {% for field in form %}
@@ -171,18 +151,21 @@
                                                         {% endif %}
                                                     {% endfor %}
                                                 </div>
-                                                <button type="submit" class="btn btn-dark">Add </button>
-                                                <i class="bi bi-info-circle ms-2" data-bs-toggle="tooltip" data-bs-placement="top" title='{{ form.hidden_name.description or "Docstring is not available" }}' ></i>
+                                                <button type="submit" class="btn btn-dark">Add</button>
+                                                {% if 'hidden_name' in form %}
+                                                    <i class="bi bi-info-circle ms-2" data-bs-toggle="tooltip" data-bs-placement="top"
+                                                       title='{{ form.hidden_name.description or "Docstring is not available" }}'>
+                                                    </i>
+                                                {% else %}
+                                                    <!-- handle info tooltip for flow control / workflows -->
+                                                {% endif %}
 
                                             </form>
-
-
                                         </div>
                                     </div>
                                 </div>
                             {% endfor %}
                         </div>
-                    {% endif %}
                 </div>
 
 
@@ -190,29 +173,11 @@
             {% else %}
                 <div style="margin-bottom: 4vh;"></div>
                 <div class="accordion accordion-flush">
-                    <div class="accordion-item  design-control">
-                        <h5 class="accordion-header">
-                            <button class="accordion-button" data-bs-toggle="collapse" data-bs-target="#advanced" role="button" aria-expanded="false" aria-controls="collapseExample">
-                                Builtin Operations:
-                            </button>
-                        </h5>
-                        <div class="accordion-collapse collapse show" id="advanced">
-                            <ul class="list-group">
-                                {% for instrument in ['if', 'while', 'variable', 'wait', 'repeat'] %}
-                                    <form role="form" method='GET' name="device" action="{{url_for('design.experiment_builder',instrument=instrument)}}">
-                                        <div class="form-group">
-                                            <button class="list-group-item list-group-item-action" aria-current="true" type="submit">{{instrument}}</button>
-                                        </div>
-                                    </form>
-                                {% endfor %}
-                            </ul>
-                        </div>
-                    </div>
 
                     <div class="accordion-item design-control">
                         <h5 class="accordion-header">
                             <button class="accordion-button" data-bs-toggle="collapse" data-bs-target="#deck" role="button" aria-expanded="false" aria-controls="collapseExample">
-                                Deck Actions:
+                                Operations
                             </button>
                         </h5>
                         <div class="accordion-collapse collapse show" id="deck">
@@ -232,7 +197,7 @@
                         <div class="accordion-item  design-control">
                             <h5 class="accordion-header">
                                 <button class="accordion-button" data-bs-toggle="collapse" data-bs-target="#local" role="button" aria-expanded="false" aria-controls="collapseExample">
-                                    Local Actions:
+                                    Local Operations
                                 </button>
                             </h5>
                             <div class="accordion-collapse collapse show" id="local">
@@ -279,7 +244,7 @@
                 </ul>
             </div>
 
-            <div class="canvas">
+            <div class="canvas" droppable="true">
                 <div class="collapse" id="info">
                     <table class="table script-table">
                         <tbody>
@@ -363,6 +328,10 @@
                         <label class="input-group-text" for="run_name">Run Name</label>
                         <input class="form-control" type="text" name="run_name" id="run_name" placeholder="{{script['name']}}" required="required">
                     </div>
+                    <div class="form-check form-switch">
+                        <input class="form-check-input" type="checkbox" name="register_workflow" id="register_workflow">
+                        <label class="input-group-label" for="register_workflow">Register this workflow</label>
+                    </div>
                 </div>
                 <div class="modal-footer">
                     <button type="button" class="btn btn-secondary" data-bs-dismiss="modal"> Close </button>
@@ -415,8 +384,29 @@
         </div>
     </div>
 </div>
+<!-- Bootstrap Modal -->
+<div class="modal fade" id="dropModal" tabindex="-1" aria-labelledby="dropModalLabel" aria-hidden="true">
+    <div class="modal-dialog">
+        <div class="modal-content">
+            <div class="modal-header">
+                <h5 class="modal-title" id="dropModalLabel">Configure Action</h5>
+                <button type="button" class="btn-close" data-bs-dismiss="modal" aria-label="Close"></button>
+            </div>
+            <div class="modal-body">
+                <p>Drop Position ID: <strong id="modalDropTarget"></strong></p>
+
+                <!-- Form will be dynamically inserted here -->
+                <div id="modalFormFields"></div>
+            </div>
+            <div class="modal-footer">
+                <button type="button" class="btn btn-secondary" data-bs-dismiss="modal">Cancel</button>
+            </div>
+        </div>
+    </div>
+</div>
+
 
-    {% if instrument and not instrument in ['if' , 'while' , 'variable' ,'wait', 'repeat'] and use_llm %}
+    {% if instrument and use_llm %}
         <script>
             const buttonIds = {{ ['generate'] | tojson }};
         </script>

{ivoryos-0.1.20 → ivoryos-0.1.21}/ivoryos/routes/design/templates/design/experiment_run.html

@@ -28,7 +28,7 @@
                 <div class="accordion-body">
                     {% if script['script'] or script['prep'] or script['cleanup'] %}
                         <div class="row">
-                            <div class="col-lg-6 col-sm-12" id="run-panel">
+                            <div class="col-lg-6 col-sm-12" id="run-panel"  style="{{ 'display: none;'  if pause_status else '' }}">
                                 <ul class="nav nav-tabs" id="myTabs" role="tablist">
                                     <li class="nav-item" role="presentation">
                                         <a class="nav-link {{ 'disabled' if config_list else '' }} {{ 'active' if not config_list else '' }}" id="tab1-tab" data-bs-toggle="tab" href="#tab1" role="tab" aria-controls="tab1" aria-selected="false">Repeat</a>
@@ -217,10 +217,11 @@
                                     </div>
                                 </div>
                             </div>
-                            <div class="col-lg-6 col-sm-12" id="code-panel" style="display: none;">
+                            <div class="col-lg-6 col-sm-12" id="code-panel" style="{{ '' if pause_status else 'display: none;'}}">
                                 <p>
                                     <h5>Progress：</h5>
                                     {% for stype, script in line_collection.items() %}
+                                        <h6>{{ stype }}：</h6>
                                         {% for code in script %}
                                             <pre style="margin: 0; padding: 0; line-height: 1;"><code class="python" id="{{ stype }}-{{ loop.index0 }}" >{{code}}</code></pre>
                                         {% endfor %}
@@ -233,7 +234,11 @@
                                         <h5>Progress：</h5>
                                         <div class="d-flex gap-2 ms-auto">
                                             <button id="pause-resume" class="btn btn-info text-white" data-bs-toggle="tooltip" title="Pause execution">
-                                                <i class="bi bi-pause-circle"></i> <!-- Icon for Pause -->
+                                                {% if pause_status %}
+                                                    <i class="bi bi-play-circle"></i> <!-- Icon for Pause -->
+                                                {% else %}
+                                                    <i class="bi bi-pause-circle"></i> <!
+                                                {% endif %}
                                             </button>
                                             <button id="abort-current" class="btn btn-danger text-white" data-bs-toggle="tooltip" title="Stop execution after current step">
                                                 <i class="bi bi-stop-circle"></i> <!-- Icon for Stop After This Step -->
@@ -326,6 +331,28 @@
             </div>
         </div>
     </div>
+
+    <!-- Error Modal -->
+    <div class="modal fade" id="error-modal" tabindex="-1" aria-labelledby="errorModalLabel" aria-hidden="true">
+      <div class="modal-dialog">
+        <div class="modal-content">
+          <div class="modal-header">
+            <h5 class="modal-title" id="errorModalLabel">Error Detected</h5>
+            <button type="button" class="btn-close" data-bs-dismiss="modal" aria-label="Close"></button>
+          </div>
+          <div class="modal-body">
+            <p id="error-message">An error has occurred.</p>
+            <p>Do you want to continue execution or stop?</p>
+          </div>
+          <div class="modal-footer">
+            <button type="button" class="btn btn-danger" id="stop-btn" data-bs-dismiss="modal">Stop Execution</button>
+            <button type="button" class="btn btn-success" id="continue-btn" data-bs-dismiss="modal">Continue</button>
+          </div>
+        </div>
+      </div>
+    </div>
+
+
     <script src="{{ url_for('static', filename='js/socket_handler.js') }}"></script>
     <script>
         var rowCount = 0; // Initialize row count
@@ -347,22 +374,4 @@
             }
         });
     </script>
-
-    <script>
-    var socket = io.connect(window.location.origin);
-
-    socket.on('execution', function(data) {
-        // Remove highlighting from all lines
-        document.querySelectorAll('pre code').forEach(el => el.style.backgroundColor = '');
-
-        // Get the currently executing line and highlight it
-        let executingLine = document.getElementById(data.section);
-        if (executingLine) {
-            executingLine.style.backgroundColor = '#cce5ff'; // Highlight
-            executingLine.style.transition = 'background-color 0.3s ease-in-out';
-
-        }
-    });
-</script>
-
 {% endblock %}