ivoryos 0.1.6__py3-none-any.whl → 0.1.7__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', None)
+    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/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__())

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

@@ -255,7 +255,7 @@ def create_form_from_module(sdl_module, autofill: bool, script=None, design=True
         attr = getattr(sdl_module, attr_name)
         if inspect.ismethod(attr) and not attr_name.startswith('_'):
             form_class = create_add_form(attr, attr_name, autofill, script, design)
-            method_forms[attr_name] = form_class()
+            method_forms[format_name(attr_name)] = form_class()
     return method_forms
 
 

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.7.dist-info}/METADATA

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

@@ -1,4 +1,4 @@
-ivoryos/__init__.py,sha256=prqANp0lD5gqHOsvVEhnuAgaCvKg2BcC3zWiXZbJS_U,3535
+ivoryos/__init__.py,sha256=HuYhPOMFGkQ_f9s1dZXaknAylKXds0jYK0nmdf1I6Z8,3932
 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
@@ -14,7 +14,7 @@ ivoryos/routes/database/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG
 ivoryos/routes/database/database.py,sha256=e1OpmQayM0KYUTPiYQzXHo-zLVd5yHtS0EYNce7vcmQ,4357
 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/design.py,sha256=FUuO-1iSeKhrLQ5Qz4zQwZiFRC5rax7RwhJBH9Wt42U,16895
 ivoryos/routes/design/templates/design/experiment_builder.html,sha256=IAU9XvtbcEWDlnCqf2Z81gcBB7bNDFOfod3xLJfRbQc,27371
 ivoryos/routes/design/templates/design/experiment_run.html,sha256=xoEHH8CC83KCWTPavwP9JWUI8SE5HX9bkEfJN6vMg5s,22845
 ivoryos/routes/main/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -23,24 +23,25 @@ ivoryos/routes/main/templates/main/help.html,sha256=TY6uD-v8MXpiZlGucchko1Bas5jH
 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=mqNDuaxniaGnKcJ26N9dcXpHQ6TabXTqPVbcrcDvM4s,12239
 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/llm_agent.py,sha256=z0DIpZzc-z09p-diUZIOE5L9zfFW8RwseFjbfUvEqoQ,6596
+ivoryos/utils/script_runner.py,sha256=gtqiHy4-40j5FMERXrmGb4jb9RAPzjCR345PMPduDno,7120
 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/utils.py,sha256=BMmvyBNo8PYs-MiBiiHjYPvSwrHORofbNwhPYpaVnfI,15249
+ivoryos-0.1.7.dist-info/LICENSE,sha256=psyqat4GJkzi42551i0kH-bXLbEzrQEnjPDwy2TVhv8,1105
+ivoryos-0.1.7.dist-info/METADATA,sha256=SgV11aX0JSQjJRsn_1vDlt15DrdAEWKM7DSs5dU6cjE,6101
+ivoryos-0.1.7.dist-info/WHEEL,sha256=yQN5g4mg4AybRjkgi-9yy4iQEFibGQmlz78Pik5Or-A,92
+ivoryos-0.1.7.dist-info/top_level.txt,sha256=FRIWWdiEvRKqw-XfF_UK3XV0CrnNb6EmVbEgjaVazRM,8
+ivoryos-0.1.7.dist-info/RECORD,,