ivoryos 0.1.6__py3-none-any.whl → 0.1.8__py3-none-any.whl

ivoryos/__init__.py

@@ -2,7 +2,7 @@ import os
 import sys
 from typing import Union
 
-from flask import Flask
+from flask import Flask, redirect, url_for
 
 from ivoryos.config import Config, get_config
 from ivoryos.routes.auth.auth import auth, login_manager
@@ -19,7 +19,8 @@ global_config = GlobalConfig()
 
 
 def create_app(config_class=None):
-    app = Flask(__name__)
+    url_prefix = os.getenv('URL_PREFIX', "/ivoryos")
+    app = Flask(__name__, static_url_path=f'{url_prefix}/static', static_folder='static')
     app.config.from_object(config_class or 'config.get_config()')
 
     # Initialize extensions
@@ -45,11 +46,15 @@ def create_app(config_class=None):
         g.logger = logger
         g.socketio = socketio
 
-    app.register_blueprint(main)
-    app.register_blueprint(auth)
-    app.register_blueprint(design)
-    app.register_blueprint(database)
-    app.register_blueprint(control)
+    app.register_blueprint(main, url_prefix=url_prefix)
+    app.register_blueprint(auth, url_prefix=url_prefix)
+    app.register_blueprint(design, url_prefix=url_prefix)
+    app.register_blueprint(database, url_prefix=url_prefix)
+    app.register_blueprint(control, url_prefix=url_prefix)
+
+    @app.route('/')
+    def redirect_to_prefix():
+        return redirect(url_for('main.index'))  # Assuming 'index' is a route in your blueprint
 
     return app
 

ivoryos/routes/control/templates/control/controllers.html

@@ -37,7 +37,7 @@
                                     {% endfor %}
                                 </div>
                                 <div class="input-group mb-3">
-                                    <button type="submit" name="{{ function }}" id="{{ function }}" class="form-control" style="background-color: #a5cece;">{{function}} </button>
+                                    <button type="submit" name="{{ function }}" id="{{ function }}" class="form-control" style="background-color: #a5cece;">{{format_name(function)}} </button>
                                 </div>
                             </form>
                         </div>

ivoryos/routes/database/database.py

@@ -7,15 +7,6 @@ from ivoryos.utils.utils import get_script_file, post_script_file
 database = Blueprint('database', __name__, template_folder='templates/database')
 
 
-@database.route("/delete/<id>")
-@login_required
-def delete_action(id):
-    back = request.referrer
-    script = get_script_file()
-    script.delete_action(id)
-    post_script_file(script)
-    return redirect(back)
-
 
 @database.route("/edit_workflow/<workflow_name>")
 @login_required

ivoryos/routes/design/design.py

@@ -13,10 +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
+from ivoryos.utils.llm_agent import LlmAgent
 from ivoryos.utils.db_models import Script
 from ivoryos.utils.script_runner import ScriptRunner
 
-
 socketio = SocketIO()
 design = Blueprint('design', __name__, template_folder='templates/design')
 
@@ -165,7 +165,6 @@ def generate_code():
                 model = current_app.config["LLM_MODEL"]
                 server = current_app.config["LLM_SERVER"]
                 module = current_app.config["MODULE"]
-                from ivoryos.utils.llm_agent import LlmAgent
                 agent = LlmAgent(host=server, model=model, output_path=os.path.dirname(os.path.abspath(module)))
             except Exception as e:
                 flash(e.__str__())
@@ -395,3 +394,13 @@ def edit_action(uuid):
                 flash(e.__str__())
         session.pop('edit_action')
     return redirect(url_for('design.experiment_builder'))
+
+
+@design.route("/delete/<id>", methods=['GET', 'POST'])
+@login_required
+def delete_action(id):
+    back = request.referrer
+    script = utils.get_script_file()
+    script.delete_action(id)
+    utils.post_script_file(script)
+    return redirect(back)

ivoryos/routes/design/templates/design/experiment_builder.html

@@ -302,7 +302,7 @@
                     {% for button in buttons %}
                         <li id="{{ button['id'] }}" style="list-style-type: none;">
                         <a href="{{ url_for('design.edit_action', uuid=button['uuid']) }}" type="button" class="btn btn-light" style="{{ button['style'] }}">{{ button['label'] }}</a>
-                        <a href="/delete/{{ button['id'] }}" type="button" class="btn btn-light"><span class="bi bi-trash"></span></a>
+                        <a href="{{ url_for('design.delete_action', id=button['id']) }}" type="button" class="btn btn-light"><span class="bi bi-trash"></span></a>
                         </li>
                     {% endfor %}
                     </ul>

ivoryos/static/js/socket_handler.js

@@ -6,7 +6,16 @@ document.addEventListener("DOMContentLoaded", function() {
     socket.on('progress', function(data) {
         var progress = data.progress;
         console.log(progress);
-        $('#progress-bar-inner').css('width', progress + '%')
+        // Update the progress bar's width and appearance
+        var progressBar = document.getElementById('progress-bar-inner');
+        progressBar.style.width = progress + '%';
+        progressBar.setAttribute('aria-valuenow', progress);
+
+        if (progress === 100) {
+            // Remove animation and set green color when 100% is reached
+            progressBar.classList.remove('progress-bar-animated');
+            progressBar.classList.add('bg-success'); // Bootstrap class for green color
+        }
     });
     socket.on('log', function(data) {
         var logMessage = data.message;

ivoryos/templates/base.html

@@ -25,7 +25,7 @@
         <div class= "container">
 
             <a class="navbar-brand" href="{{ url_for('main.index') }}">
-                <img src="{{url_for('static', filename='logo.png')}}" alt="Logo" height="60" class="d-inline-block align-text-bottom">
+                <img src="{{url_for('static', filename='logo.webp')}}" alt="Logo" height="60" class="d-inline-block align-text-bottom">
             </a>
             <button class="navbar-toggler" type="button" data-bs-toggle="collapse" data-bs-target="#navbarSupportedContent" aria-controls="navbarSupportedContent" aria-expanded="false" aria-label="Toggle navigation">
                 <span class="navbar-toggler-icon"></span>
@@ -33,6 +33,9 @@
 
             <div class="collapse navbar-collapse" id="navbarSupportedContent">
                 <ul class="navbar-nav mr-auto">
+                    <li class="nav-item">
+                        <a class="nav-link" href="{{ url_for('main.index') }}" aria-current="page">Home</a>
+                    </li>
                     <li class="nav-item">
                         <a class="nav-link" href="{{ url_for('database.load_from_database') }}" aria-current="page">Library</a>
                     </li>

ivoryos/utils/form.py

@@ -220,7 +220,8 @@ def create_form_for_method(method, method_name, autofill, script=None, design=Tr
         formatted_param_name = format_name(param.name)
         field_kwargs = {
             "label": formatted_param_name,
-            "default": f'#{param.name}' if autofill else (param.default if param.default is not param.empty else ""),
+            "default": f'#{param.name}' if autofill else (param.default if param.default is not param.empty else None),
+            "validators": [InputRequired()] if param.default is param.empty else None,
             **({"script": script} if (autofill or design) else {})
         }
         field_class, placeholder_text = annotation_mapping.get(

ivoryos/utils/llm_agent.py

@@ -3,7 +3,7 @@ import json
 import os
 import re
 
-from openai import OpenAI, BaseModel
+from openai import OpenAI
 
 
 # from dotenv import load_dotenv

ivoryos/utils/script_runner.py

@@ -53,6 +53,7 @@ class ScriptRunner:
     def _run_with_stop_check(self, script: Script, repeat_count, run_name, logger, socketio, config, bo_args,
                              output_path):
         time.sleep(1)
+        self._emit_progress(socketio, 1)
         try:
             # Run "prep" section once
             script_dict = script.script_dict
@@ -75,6 +76,7 @@ class ScriptRunner:
         finally:
             with self.lock:
                 self.is_running = False  # Reset the running flag when done
+            self._emit_progress(socketio, 100)
 
     def _run_actions(self, actions, section_name="", run_name=None, logger=None):
         logger.info(f'Executing {section_name} steps') if actions else logger.info(f'No {section_name} steps')
@@ -104,7 +106,7 @@ class ScriptRunner:
                     break
                 logger.info(f'Executing {i + 1} of {len(config)} with kwargs = {kwargs}')
                 progress = (i + 1) * 100 / len(config)
-                socketio.emit('progress', {'progress': progress})
+                self._emit_progress(socketio, progress)
                 fname = f"{run_name}_script"
                 function = self.globals_dict[fname]
                 output = function(**kwargs)
@@ -121,8 +123,8 @@ class ScriptRunner:
                 logger.info(f'Stopping execution during {run_name}: {i + 1}/{int(repeat_count)}')
                 break
             logger.info(f'Executing {run_name} experiment: {i + 1}/{int(repeat_count)}')
-            progress = (i + 1) * 100 / int(repeat_count)
-            socketio.emit('progress', {'progress': progress})
+            progress = (i + 1) * 100 / int(repeat_count) - 0.1
+            self._emit_progress(socketio, progress)
             if bo_args:
                 try:
                     parameters, trial_index = ax_client.get_next_trial()
@@ -156,3 +158,7 @@ class ScriptRunner:
             writer.writeheader()
             writer.writerows(output_list)
         logger.info(f'Results saved to {file_path}')
+
+    @staticmethod
+    def _emit_progress(socketio, progress):
+        socketio.emit('progress', {'progress': progress})

ivoryos/utils/utils.py

@@ -1,3 +1,4 @@
+import ast
 import importlib
 import inspect
 import logging
@@ -124,23 +125,22 @@ def _get_type_from_parameters(arg, parameters):
     """get argument types from inspection"""
     arg_type = ''
     if type(parameters) is inspect.Signature:
-        p = parameters.parameters
+        annotation = parameters.parameters[arg].annotation
+    elif type(parameters) is dict:
+        annotation = parameters[arg]
+    if annotation is not inspect._empty:
         # print(p[arg].annotation)
-        if p[arg].annotation is not inspect._empty:
-            # print(p[arg].annotation)
-            if p[arg].annotation.__module__ == 'typing':
+        if annotation.__module__ == 'typing':
+            if hasattr(annotation, '_name') and annotation._name in ["Optional", "Union"]:
                 # print(p[arg].annotation.__args__)
-                arg_type = [i.__name__ for i in p[arg].annotation.__args__]
-            else:
-                arg_type = p[arg].annotation.__name__
-            # print(arg_type)
-    elif type(parameters) is dict:
-        if parameters[arg]:
-
-            if parameters[arg].__module__ == 'typing':
-                arg_type = [i.__name__ for i in parameters[arg].__args__]
+                arg_type = [i.__name__ for i in annotation.__args__]
+            elif hasattr(annotation, '__origin__'):
+                arg_type = annotation.__origin__.__name__
             else:
-                arg_type = parameters[arg].__name__
+                # TODO
+                pass
+        else:
+            arg_type = annotation.__name__
     return arg_type
 
 
@@ -168,17 +168,12 @@ def _convert_by_str(args, arg_types):
     if type(arg_types) is not list:
         arg_types = [arg_types]
     for arg_type in arg_types:
-        if arg_type == "any":
+        if not arg_type == "any":
             try:
-                args = eval(args)
+                args = eval(f'{arg_type}("{args}")') if type(args) is str else eval(f'{arg_type}({args})')
+                return args
             except Exception:
-                pass
-            return args
-        try:
-            args = eval(f'{arg_type}("{args}")')
-            return args
-        except Exception:
-            raise TypeError(f"Input type error: cannot convert '{args}' to {arg_type}.")
+                raise TypeError(f"Input type error: cannot convert '{args}' to {arg_type}.")
 
 
 def _convert_by_class(args, arg_types):
@@ -214,16 +209,20 @@ def convert_config_type(args, arg_types, is_class: bool = False):
                 raise ValueError("config file format not supported.")
             if args[arg] == '' or args[arg] == "None":
                 args[arg] = None
-            elif args[arg] == "True" or args[arg] == "False":
-                args[arg] = bool_dict[args[arg]]
+            # elif args[arg] == "True" or args[arg] == "False":
+            #     args[arg] = bool_dict[args[arg]]
             else:
                 arg_type = arg_types[arg]
-
-                if is_class:
-                    # if arg_type.__module__ == 'builtins':
-                    args[arg] = _convert_by_class(args[arg], arg_type)
-                else:
-                    args[arg] = _convert_by_str(args[arg], arg_type)
+                try:
+                    args[arg] = ast.literal_eval(args[arg])
+                except ValueError:
+                    pass
+                if type(args[arg]) is not arg_type and not type(args[arg]).__name__ == arg_type:
+                    if is_class:
+                        # if arg_type.__module__ == 'builtins':
+                        args[arg] = _convert_by_class(args[arg], arg_type)
+                    else:
+                        args[arg] = _convert_by_str(args[arg], arg_type)
     return args
 
 

{ivoryos-0.1.6.dist-info → ivoryos-0.1.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ivoryos
-Version: 0.1.6
+Version: 0.1.8
 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.6.dist-info → ivoryos-0.1.8.dist-info}/RECORD

@@ -1,4 +1,4 @@
-ivoryos/__init__.py,sha256=prqANp0lD5gqHOsvVEhnuAgaCvKg2BcC3zWiXZbJS_U,3535
+ivoryos/__init__.py,sha256=0bjHffTf3m8NWU3zwLz7sZMDqhwNP9FpNXJhxFRliEQ,3938
 ivoryos/config.py,sha256=K03jdGmbUfJ9o4kK6NOtDGJtydGHFq8-oU8nvCyq5zQ,1358
 ivoryos/routes/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ivoryos/routes/auth/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -7,40 +7,39 @@ ivoryos/routes/auth/templates/auth/login.html,sha256=1uxYU7NpxVaA4sfwkC6CuzZXJdy
 ivoryos/routes/auth/templates/auth/signup.html,sha256=QQ7n4OBnF8TNFS5_4s11n4BCqSePn429rZfA6vO8qw8,1497
 ivoryos/routes/control/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ivoryos/routes/control/control.py,sha256=YKM1T_bDIggbW_NF6Ld6-rb3d0KbAjyikJztN-ka_XM,11305
-ivoryos/routes/control/templates/control/controllers.html,sha256=CD1DEm9DuBBlKg_ltrEWgNxWhTIR9C2_a0-AkxwGre0,4146
+ivoryos/routes/control/templates/control/controllers.html,sha256=5LKKlhiGdg2fxs-akAAxtm1AMF5HAiuSlputDBMeTc0,4159
 ivoryos/routes/control/templates/control/controllers_home.html,sha256=IND1T3_mPZd-MzfuyodbedMnmsTowiTVdRp5ez6NoZM,2783
 ivoryos/routes/control/templates/control/controllers_new.html,sha256=Wqn9x9D6K7RWHkLFxvZkzbIJxHJR1zywQ6WDgySXOig,5010
 ivoryos/routes/database/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ivoryos/routes/database/database.py,sha256=e1OpmQayM0KYUTPiYQzXHo-zLVd5yHtS0EYNce7vcmQ,4357
+ivoryos/routes/database/database.py,sha256=W7axqxlRyIwqdLCuzYzER33obz6PCkevk6NTy-sCOLw,4133
 ivoryos/routes/database/templates/database/experiment_database.html,sha256=x9zf4u4KbG6BEICnH_TOVNNUkp5oAmGBB12OUX0PPl4,3506
 ivoryos/routes/design/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ivoryos/routes/design/design.py,sha256=g1V4HYLldD7O6K5brtdxV8uoyOVGTpM3rC1lOmw91AA,16913
-ivoryos/routes/design/templates/design/experiment_builder.html,sha256=IAU9XvtbcEWDlnCqf2Z81gcBB7bNDFOfod3xLJfRbQc,27371
+ivoryos/routes/design/design.py,sha256=sbGkd9GztrNsY0vvvC4wZ9O1_rRS1f5dqkAYCATgvdM,17156
+ivoryos/routes/design/templates/design/experiment_builder.html,sha256=HIkb-llvEy57agHuaiSN95tKj4Oh0T88GBfHxEsz8bI,27399
 ivoryos/routes/design/templates/design/experiment_run.html,sha256=xoEHH8CC83KCWTPavwP9JWUI8SE5HX9bkEfJN6vMg5s,22845
 ivoryos/routes/main/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ivoryos/routes/main/main.py,sha256=Zkagtw0E67FaspNJ87jkYpc-wqqoTq99dyyB7qdnOVM,655
 ivoryos/routes/main/templates/main/help.html,sha256=TY6uD-v8MXpiZlGucchko1Bas5jHRzARIgCgneBZ_ok,9511
 ivoryos/routes/main/templates/main/home.html,sha256=fjeSSXkuVDr2y-CQmskNHf1yYFFdrPPI4wn6_XPb6AY,3252
 ivoryos/static/favicon.ico,sha256=RhlrPtfITOkzC9BjP1UB1V5L9Oyp6NwNtWeMcGOnpyc,15406
-ivoryos/static/logo.png,sha256=7lNyToDllflGPUK2sj7IBR8FkHLC-6gi-OVSL9o4jrs,63464
+ivoryos/static/logo.webp,sha256=lXgfQR-4mHTH83k7VV9iB54-oC2ipe6uZvbwdOnLETc,14974
 ivoryos/static/style.css,sha256=rY6n2mbP_xNavtVin_yUqtcvNm6uqAF82t7ONE2Sx9E,3918
 ivoryos/static/gui_annotation/Slide1.png,sha256=Lm4gdOkUF5HIUFaB94tl6koQVkzpitKj43GXV_XYMMc,121727
 ivoryos/static/gui_annotation/Slide2.PNG,sha256=z3wQ9oVgg4JTWVLQGKK_KhtepRHUYP1e05XUWGT2A0I,118761
 ivoryos/static/js/overlay.js,sha256=44l9THVKYZfa7HX6siyqY7EdFWKBk5pyyKgN0_7ZnrM,495
-ivoryos/static/js/socket_handler.js,sha256=HduGphUyyZq8GE1-5ZmDgHGTzcWDRa6oXBNrWWMnxwM,989
+ivoryos/static/js/socket_handler.js,sha256=2936CldW6Po_saWh1aL_EV-VydJVIvikrNfTaSfU1sE,1449
 ivoryos/static/js/sortable_card.js,sha256=mDVd2YjhusLokUw3xL6YOZLXIzty9yKDsC1U5yR8aC8,831
 ivoryos/static/js/sortable_design.js,sha256=BxNXzqET_yY0xpS1Fc0iwPCnkkDwYMiuVqkgOPMb6JY,1156
-ivoryos/templates/base.html,sha256=cu6MZMlfOR0IhuNrMin_pWYtm1_TpJLHI6KH5NomopI,7763
+ivoryos/templates/base.html,sha256=KcKMjITaaC23yzIRN535uMhzv5x96nJl245Y2GaqdsM,7943
 ivoryos/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ivoryos/utils/db_models.py,sha256=umEATdxDP-sR_AMaLrOW2PBUcmBf1w38DLeFkBvVjJE,19280
-ivoryos/utils/form.py,sha256=W8fSGybeWSCfPqySEfH1mKJbdoYmRb3vwpy15w2rj_Q,12226
+ivoryos/utils/form.py,sha256=FcrexHxcykoQke2J78mDQf1PRAcXlMHwi-JZMgGBpOU,12316
 ivoryos/utils/global_config.py,sha256=JCQvmZB0pNC-EjveRu48Tp4uvcNwn9DP3Ema6Xd9fJY,1656
-ivoryos/utils/llm_agent.py,sha256=pxiaiIVHzbGA_Tq4yOZY3iyIDikx6-_chIvPEjlywLg,6607
-ivoryos/utils/script_runner.py,sha256=jlJMCFLmcS6GOlOfEmLXAzG47sU18nNtRPtojjdN018,6919
-ivoryos/utils/task_manager.py,sha256=xfQ1s9ywWDrCYYpdgliVvoWED0s2xARmo3LXvAy6fgY,2517
-ivoryos/utils/utils.py,sha256=8GMuMwKUKKMjz8hvilavNlV4RANcQWsJy8hZpbXanRg,15042
-ivoryos-0.1.6.dist-info/LICENSE,sha256=psyqat4GJkzi42551i0kH-bXLbEzrQEnjPDwy2TVhv8,1105
-ivoryos-0.1.6.dist-info/METADATA,sha256=QTek7bK8bAPqXCo0RIY6w_YTUeisTWFcg2ythkjqYc4,6101
-ivoryos-0.1.6.dist-info/WHEEL,sha256=yQN5g4mg4AybRjkgi-9yy4iQEFibGQmlz78Pik5Or-A,92
-ivoryos-0.1.6.dist-info/top_level.txt,sha256=FRIWWdiEvRKqw-XfF_UK3XV0CrnNb6EmVbEgjaVazRM,8
-ivoryos-0.1.6.dist-info/RECORD,,
+ivoryos/utils/llm_agent.py,sha256=z0DIpZzc-z09p-diUZIOE5L9zfFW8RwseFjbfUvEqoQ,6596
+ivoryos/utils/script_runner.py,sha256=gtqiHy4-40j5FMERXrmGb4jb9RAPzjCR345PMPduDno,7120
+ivoryos/utils/utils.py,sha256=BMmvyBNo8PYs-MiBiiHjYPvSwrHORofbNwhPYpaVnfI,15249
+ivoryos-0.1.8.dist-info/LICENSE,sha256=psyqat4GJkzi42551i0kH-bXLbEzrQEnjPDwy2TVhv8,1105
+ivoryos-0.1.8.dist-info/METADATA,sha256=8t3dDdrj62BKNKuOMKS1FOwlPur9rox_4F9_x9haIYk,6101
+ivoryos-0.1.8.dist-info/WHEEL,sha256=yQN5g4mg4AybRjkgi-9yy4iQEFibGQmlz78Pik5Or-A,92
+ivoryos-0.1.8.dist-info/top_level.txt,sha256=FRIWWdiEvRKqw-XfF_UK3XV0CrnNb6EmVbEgjaVazRM,8
+ivoryos-0.1.8.dist-info/RECORD,,

ivoryos/utils/task_manager.py

@@ -1,80 +0,0 @@
-import threading
-import queue
-import time
-
-
-# A task manager class to manage the queue and tasks
-class TaskManager:
-    def __init__(self):
-        self.task_queue = queue.Queue()
-        self.current_task = None
-        self.stop_event = threading.Event()
-
-    def add_task(self, func, **kwargs):
-        # Add the function and its kwargs to the task queue
-        self.task_queue.put((func, kwargs))
-
-    def run_tasks(self):
-        # Run the tasks from the queue
-        while not self.task_queue.empty():
-            func, kwargs = self.task_queue.get()
-            thread = threading.Thread(target=self.run_task, args=(func, kwargs))
-            self.current_task = thread
-            thread.start()
-            thread.join()  # Wait for the task to finish
-
-    def run_task(self, func, kwargs):
-        # Run the task function with control to stop in the middle
-        self.stop_event.clear()  # Reset the stop flag
-        func(**kwargs)
-        if self.stop_event.is_set():
-            print("Current task was stopped.")
-
-    def stop_current_task(self):
-        # Stop the current task by setting the stop flag
-        if self.current_task and self.current_task.is_alive():
-            print("Stopping current task...")
-            self.stop_event.set()  # Signal to stop the current task
-            self.current_task.join()  # Wait for the task to stop
-
-
-# Wrapping tasks to allow stopping between them
-def function_to_call(stop_event, **kwargs):
-    if stop_event.is_set():
-        return
-    task1(kwargs['arg1'])
-    if stop_event.is_set():
-        return
-    task2(kwargs['arg2'])
-
-
-# Dummy task functions as provided
-def task1(arg1):
-    for i in range(arg1):
-        print(f"Task 1 running: {i}")
-        time.sleep(1)
-
-
-def task2(arg2):
-    for i in range(arg2):
-        print(f"Task 2 running: {i}")
-        time.sleep(1)
-
-
-if __name__ == "__main__":
-    manager = TaskManager()
-
-    # Add tasks to the manager
-    manager.add_task(function_to_call, stop_event=manager.stop_event, arg1=3, arg2=5)
-    manager.add_task(function_to_call, stop_event=manager.stop_event, arg1=2, arg2=4)
-
-    # Run tasks in a separate thread
-    manager_thread = threading.Thread(target=manager.run_tasks)
-    manager_thread.start()
-
-    # Example: Stop the current workflow while task1 is running
-    time.sleep(2)  # Let task1 run for a bit
-    manager.stop_current_task()
-
-    # Wait for all tasks to finish
-    manager_thread.join()