blue-assistant 4.97.1__tar.gz → 4.121.1__tar.gz

{blue_assistant-4.97.1/blue_assistant.egg-info → blue_assistant-4.121.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: blue_assistant
-Version: 4.97.1
+Version: 4.121.1
 Summary: 🧠 An AI Assistant.
 Home-page: https://github.com/kamangir/blue-assistant
 Author: Arash Abadpour (Kamangir)
@@ -67,9 +67,13 @@ graph LR
     classDef folder fill:#999,stroke:#333,stroke-width:2px;
 ```
 
+|   |
+| --- |
+| [`blue-amo`](https://raw.githubusercontent.com/kamangir/blue-assistant/main/blue_assistant/script/repository/blue_amo/README.md) [![image](https://github.com/kamangir/assets/raw/main/blue-amo-2025-02-03-nswnx6/stitching_the_frames-2.png?raw=true)](https://raw.githubusercontent.com/kamangir/blue-assistant/main/blue_assistant/script/repository/blue_amo/README.md) story-telling with AI |
+
 ---
 
 
 [![pylint](https://github.com/kamangir/blue-assistant/actions/workflows/pylint.yml/badge.svg)](https://github.com/kamangir/blue-assistant/actions/workflows/pylint.yml) [![pytest](https://github.com/kamangir/blue-assistant/actions/workflows/pytest.yml/badge.svg)](https://github.com/kamangir/blue-assistant/actions/workflows/pytest.yml) [![bashtest](https://github.com/kamangir/blue-assistant/actions/workflows/bashtest.yml/badge.svg)](https://github.com/kamangir/blue-assistant/actions/workflows/bashtest.yml) [![PyPI version](https://img.shields.io/pypi/v/blue-assistant.svg)](https://pypi.org/project/blue-assistant/) [![PyPI - Downloads](https://img.shields.io/pypi/dd/blue-assistant)](https://pypistats.org/packages/blue-assistant)
 
-built by 🌀 [`blue_options-4.207.1`](https://github.com/kamangir/awesome-bash-cli), based on 🧠 [`blue_assistant-4.97.1`](https://github.com/kamangir/blue-assistant).
+built by 🌀 [`blue_options-4.207.1`](https://github.com/kamangir/awesome-bash-cli), based on 🧠 [`blue_assistant-4.121.1`](https://github.com/kamangir/blue-assistant).

{blue_assistant-4.97.1 → blue_assistant-4.121.1}/README.md

@@ -23,9 +23,13 @@ graph LR
     classDef folder fill:#999,stroke:#333,stroke-width:2px;
 ```
 
+|   |
+| --- |
+| [`blue-amo`](./blue_assistant/script/repository/blue_amo/README.md) [![image](https://github.com/kamangir/assets/raw/main/blue-amo-2025-02-03-nswnx6/stitching_the_frames-2.png?raw=true)](./blue_assistant/script/repository/blue_amo/README.md) story-telling with AI |
+
 ---
 
 
 [![pylint](https://github.com/kamangir/blue-assistant/actions/workflows/pylint.yml/badge.svg)](https://github.com/kamangir/blue-assistant/actions/workflows/pylint.yml) [![pytest](https://github.com/kamangir/blue-assistant/actions/workflows/pytest.yml/badge.svg)](https://github.com/kamangir/blue-assistant/actions/workflows/pytest.yml) [![bashtest](https://github.com/kamangir/blue-assistant/actions/workflows/bashtest.yml/badge.svg)](https://github.com/kamangir/blue-assistant/actions/workflows/bashtest.yml) [![PyPI version](https://img.shields.io/pypi/v/blue-assistant.svg)](https://pypi.org/project/blue-assistant/) [![PyPI - Downloads](https://img.shields.io/pypi/dd/blue-assistant)](https://pypistats.org/packages/blue-assistant)
 
-built by 🌀 [`blue_options-4.207.1`](https://github.com/kamangir/awesome-bash-cli), based on 🧠 [`blue_assistant-4.97.1`](https://github.com/kamangir/blue-assistant).
+built by 🌀 [`blue_options-4.207.1`](https://github.com/kamangir/awesome-bash-cli), based on 🧠 [`blue_assistant-4.121.1`](https://github.com/kamangir/blue-assistant).

{blue_assistant-4.97.1 → blue_assistant-4.121.1}/blue_assistant/.abcli/tests/script_run.sh

@@ -17,7 +17,8 @@ function test_blue_assistant_script_run() {
             script=$script_name \
             test_blue_assistant_script_run-$(abcli_string_timestamp_short) \
             "${@:2}" \
-            --test_mode 1
+            --test_mode 1 \
+            --verbose 1
 
         [[ $? -ne 0 ]] && return 1
 

blue_assistant-4.121.1/blue_assistant/README.py

@@ -0,0 +1,41 @@
+import os
+
+from blue_objects import file, README
+
+from blue_assistant import NAME, VERSION, ICON, REPO_NAME
+
+
+items = [
+    "[`{}`]({}) [![image]({})]({}) {}".format(
+        item["title"],
+        item["url"],
+        item["marquee"],
+        item["url"],
+        item["description"],
+    )
+    for item in [
+        {
+            "title": "blue-amo",
+            "url": "./blue_assistant/script/repository/blue_amo/README.md",
+            "marquee": "https://github.com/kamangir/assets/raw/main/blue-amo-2025-02-03-nswnx6/stitching_the_frames-2.png?raw=true",
+            "description": "story-telling with AI",
+        }
+    ]
+]
+
+
+def build():
+    return all(
+        README.build(
+            items=readme.get("items", []),
+            path=os.path.join(file.path(__file__), readme["path"]),
+            ICON=ICON,
+            NAME=NAME,
+            VERSION=VERSION,
+            REPO_NAME=REPO_NAME,
+        )
+        for readme in [
+            {"items": items, "path": ".."},
+            {"path": "script/repository/blue_amo/README.md"},
+        ]
+    )

{blue_assistant-4.97.1 → blue_assistant-4.121.1}/blue_assistant/__init__.py

@@ -4,7 +4,7 @@ ICON = "🧠"
 
 DESCRIPTION = f"{ICON} An AI Assistant."
 
-VERSION = "4.97.1"
+VERSION = "4.121.1"
 
 REPO_NAME = "blue-assistant"
 

blue_assistant-4.121.1/blue_assistant/script/actions/__init__.py

@@ -0,0 +1,16 @@
+from typing import Dict, Callable, Tuple
+
+from blue_assistant.script.repository.base.classes import BaseScript
+from blue_assistant.script.actions.generic import generic_action
+from blue_assistant.script.actions.generate_image import generate_image
+from blue_assistant.script.actions.generate_text import generate_text
+from blue_assistant.script.actions.skip import skip_action
+from blue_assistant.logger import logger
+
+
+dict_of_actions: Dict[str, Callable[[BaseScript, str], bool]] = {
+    "generic_action": generic_action,
+    "generate_image": generate_image,
+    "generate_text": generate_text,
+    "skip": skip_action,
+}

blue_assistant-4.121.1/blue_assistant/script/actions/generate_image.py

@@ -0,0 +1,52 @@
+from blueness import module
+from blue_objects import objects
+from openai_commands.image_generation.api import OpenAIImageGenerator
+
+from blue_assistant import NAME
+from blue_assistant.env import (
+    BLUE_ASSISTANT_IMAGE_DEFAULT_MODEL,
+    BLUE_ASSISTANT_IMAGE_DEFAULT_SIZE,
+    BLUE_ASSISTANT_IMAGE_DEFAULT_QUALITY,
+)
+from blue_assistant.script.repository.base.classes import BaseScript
+from blue_assistant.logger import logger
+
+NAME = module.name(__file__, NAME)
+
+
+# https://platform.openai.com/docs/guides/images
+def generate_image(
+    script: BaseScript,
+    node_name: str,
+) -> bool:
+    logger.info(f"{NAME}: {script} @ {node_name} ...")
+
+    generator = OpenAIImageGenerator(
+        model=BLUE_ASSISTANT_IMAGE_DEFAULT_MODEL,
+        verbose=script.verbose,
+    )
+
+    filename = f"{node_name}.png"
+
+    success = generator.generate(
+        prompt=script.nodes[node_name]["prompt"],
+        filename=objects.path_of(
+            filename=filename,
+            object_name=script.object_name,
+            create=True,
+        ),
+        quality=(BLUE_ASSISTANT_IMAGE_DEFAULT_QUALITY if script.test_mode else "hd"),
+        size=(BLUE_ASSISTANT_IMAGE_DEFAULT_SIZE if script.test_mode else "1792x1024"),
+        sign_with_prompt=False,
+        footer=[
+            script.nodes[node_name].get(
+                "summary_prompt",
+                script.nodes[node_name]["prompt"],
+            )
+        ],
+    )[0]
+
+    if success:
+        script.nodes[node_name]["filename"] = filename
+
+    return success

blue_assistant-4.121.1/blue_assistant/script/actions/generate_text.py

@@ -0,0 +1,85 @@
+from typing import List
+from openai import OpenAI
+import pprint
+
+from blueness import module
+from openai_commands.env import OPENAI_API_KEY
+
+from blue_assistant import NAME
+from blue_assistant.script.repository.base.classes import BaseScript
+from blue_assistant.env import (
+    BLUE_ASSISTANT_TEXT_DEFAULT_MODEL,
+    BLUE_ASSISTANT_TEXT_MAX_TOKEN,
+)
+from blue_assistant.logger import logger
+
+NAME = module.name(__file__, NAME)
+
+
+# https://platform.openai.com/docs/guides/text-generation
+def generate_text(
+    script: BaseScript,
+    node_name: str,
+) -> bool:
+    if not OPENAI_API_KEY:
+        logger.error("OPENAI_API_KEY is not set.")
+        return False
+
+    logger.info(f"{NAME}: {script} @ {node_name} ...")
+
+    messages: List = []
+    list_of_context_nodes = script.get_history(node_name)
+    logger.info("node context: {}".format(" <- ".join(list_of_context_nodes)))
+    for context_node in reversed(list_of_context_nodes):
+        messages += [
+            {
+                "role": "user",
+                "content": [
+                    {
+                        "type": "text",
+                        "text": script.apply_vars(script.nodes[context_node]["prompt"]),
+                    }
+                ],
+            }
+        ]
+
+        if script.nodes[context_node].get("completed", False):
+            messages += [
+                {
+                    "role": "assistant",
+                    "content": [
+                        {
+                            "type": "text",
+                            "text": script.nodes[context_node]["output"],
+                        }
+                    ],
+                }
+            ]
+
+    if script.verbose:
+        logger.info(f"messages: {pprint.pformat(messages)}")
+
+    client = OpenAI(api_key=OPENAI_API_KEY)
+
+    try:
+        response = client.chat.completions.create(
+            messages=messages,
+            model=BLUE_ASSISTANT_TEXT_DEFAULT_MODEL,
+            max_tokens=BLUE_ASSISTANT_TEXT_MAX_TOKEN,
+        )
+    except Exception as e:
+        logger.error(str(e))
+        return False
+
+    if script.verbose:
+        logger.info("response: {}".format(response))
+
+    if not response.choices:
+        logger.error("no choice.")
+        return False
+
+    output = response.choices[0].message.content
+    logger.info(f"🗣️ output: {output}")
+    script.nodes[node_name]["output"] = output
+
+    return True

blue_assistant-4.121.1/blue_assistant/script/actions/generic.py

@@ -0,0 +1,15 @@
+from blueness import module
+
+from blue_assistant import NAME
+from blue_assistant.script.repository.base.classes import BaseScript
+from blue_assistant.logger import logger
+
+NAME = module.name(__file__, NAME)
+
+
+def generic_action(
+    script: BaseScript,
+    node_name: str,
+) -> bool:
+    logger.info(f"{NAME}: {script} @ {node_name} ...")
+    return True

blue_assistant-4.121.1/blue_assistant/script/actions/skip.py

@@ -0,0 +1,15 @@
+from blueness import module
+
+from blue_assistant import NAME
+from blue_assistant.script.repository.base.classes import BaseScript
+from blue_assistant.logger import logger
+
+NAME = module.name(__file__, NAME)
+
+
+def skip_action(
+    script: BaseScript,
+    node_name: str,
+) -> bool:
+    logger.info(f"{NAME}: {script} @ {node_name} ...")
+    return True

blue_assistant-4.121.1/blue_assistant/script/repository/blue_amo/actions/__init__.py

@@ -0,0 +1,13 @@
+from typing import Dict, Callable
+
+from blue_assistant.script.repository.base.classes import BaseScript
+from blue_assistant.script.repository.blue_amo.actions import (
+    slicing_into_frames,
+    stitching_the_frames,
+)
+
+
+dict_of_actions: Dict[str, Callable[[BaseScript, str], bool]] = {
+    "slicing_into_frames": slicing_into_frames.slicing_into_frames,
+    "stitching_the_frames": stitching_the_frames.stitching_the_frames,
+}

blue_assistant-4.97.1/blue_assistant/script/repository/blue_amo/actions/slice_into_frames.py → blue_assistant-4.121.1/blue_assistant/script/repository/blue_amo/actions/slicing_into_frames.py

@@ -7,7 +7,7 @@ from blue_assistant.logger import logger
 NAME = module.name(__file__, NAME)
 
 
-def slice_into_frames(
+def slicing_into_frames(
     script: BaseScript,
     node_name: str,
 ) -> bool:
@@ -25,11 +25,14 @@ def slice_into_frames(
     list_of_frame_prompts += script.vars["frame_count"] * [""]
 
     for index in range(script.vars["frame_count"]):
-        node_name = f"generating-frame-{index+1:03d}"
+        node_name = f"generating_frame_{index+1:03d}"
 
-        script.nodes[node_name]["prompt"] = script.nodes[node_name]["prompt"].replace(
-            ":::input",
-            list_of_frame_prompts[index],
+        script.nodes[node_name]["summary_prompt"] = list_of_frame_prompts[index]
+
+        script.nodes[node_name]["prompt"] = (
+            script.nodes[node_name]["prompt"]
+            .replace(":::story", " ".join(list_of_frame_prompts[:index]))
+            .replace(":::input", list_of_frame_prompts[index])
         )
 
     return True

blue_assistant-4.121.1/blue_assistant/script/repository/blue_amo/actions/stitching_the_frames.py

@@ -0,0 +1,128 @@
+from typing import List
+import numpy as np
+import cv2
+from tqdm import trange
+import math
+
+from blueness import module
+from blue_objects import file, objects
+from blue_options import string
+
+from blue_assistant import NAME
+from blue_assistant.script.repository.base.classes import BaseScript
+from blue_assistant.logger import logger
+
+NAME = module.name(__file__, NAME)
+
+
+def stitching_the_frames(
+    script: BaseScript,
+    node_name: str,
+) -> bool:
+    list_of_frames_filenames: List[str] = [
+        filename
+        for filename in [
+            script.nodes[node_name_].get("filename", "")
+            for node_name_ in [
+                f"generating_frame_{index+1:03d}"
+                for index in range(script.vars["frame_count"])
+            ]
+        ]
+        if filename
+    ]
+    if not list_of_frames_filenames:
+        return True
+
+    logger.info(
+        "{} frames to stitch: {}".format(
+            len(list_of_frames_filenames),
+            ", ".join(list_of_frames_filenames),
+        )
+    )
+
+    list_of_frames: List[np.ndarray] = []
+    for filename in list_of_frames_filenames:
+        success, frame = file.load_image(
+            objects.path_of(
+                filename=filename,
+                object_name=script.object_name,
+            )
+        )
+
+        if success:
+            list_of_frames += [frame]
+
+    if not list_of_frames:
+        return True
+
+    common_height = list_of_frames[0].shape[0]
+    common_width = list_of_frames[0].shape[1]
+    for index in trange(1, len(list_of_frames)):
+        frame_height = list_of_frames[index].shape[0]
+        frame_width = list_of_frames[index].shape[1]
+
+        if frame_height != common_height or frame_width != common_width:
+            list_of_frames[index] = cv2.resize(
+                list_of_frames[index],
+                (common_width, common_height),
+                interpolation=cv2.INTER_AREA,
+            )
+
+    width_count = int(math.ceil(math.sqrt(len(list_of_frames))))
+    height_count = int(math.ceil(len(list_of_frames) / width_count))
+    logger.info(
+        "{} x {} -> {} x {}".format(
+            len(list_of_frames),
+            string.pretty_shape_of_matrix(list_of_frames[0]),
+            height_count,
+            width_count,
+        )
+    )
+
+    list_of_frames += (height_count * width_count - len(list_of_frames)) * [
+        np.zeros_like(list_of_frames[0])
+    ]
+
+    full_frame: np.ndarray = None
+    for height_index in trange(height_count):
+        row = np.concatenate(
+            list_of_frames[
+                height_index * width_count : (height_index + 1) * width_count
+            ],
+            axis=1,
+        )
+
+        if not height_index:
+            full_frame = row
+        else:
+            full_frame = np.concatenate([full_frame, row], axis=0)
+    logger.info(f"full frame: {string.pretty_shape_of_matrix(full_frame)}")
+
+    for scale in [1, 2, 4]:
+        scaled_full_frame = (
+            full_frame
+            if scale == 1
+            else cv2.resize(
+                full_frame,
+                (
+                    int(full_frame.shape[1] / scale),
+                    int(full_frame.shape[0] / scale),
+                ),
+                interpolation=cv2.INTER_AREA,
+            )
+        )
+
+        if not file.save_image(
+            objects.path_of(
+                filename="{}{}.png".format(
+                    node_name,
+                    "" if scale == 1 else f"-{scale}",
+                ),
+                object_name=script.object_name,
+            ),
+            scaled_full_frame,
+            log=True,
+        ):
+            return False
+
+    return True

{blue_assistant-4.97.1 → blue_assistant-4.121.1}/blue_assistant/script/repository/blue_amo/classes.py

@@ -5,12 +5,7 @@ from blue_objects import file, path
 
 from blue_assistant import NAME
 from blue_assistant.script.repository.generic.classes import GenericScript
-from blue_assistant.script.repository.blue_amo.actions.slice_into_frames import (
-    slice_into_frames,
-)
-from blue_assistant.script.repository.blue_amo.actions.stitch_the_frames import (
-    stitch_the_frames,
-)
+from blue_assistant.script.repository.blue_amo.actions import dict_of_actions
 from blue_assistant.logger import logger
 
 NAME = module.name(__file__, NAME)
@@ -34,7 +29,7 @@ class BlueAmoScript(GenericScript):
         if self.test_mode:
             self.vars["frame_count"] = 1
 
-        holder_node_name = "generating-the-frames"
+        holder_node_name = "generating_the_frames"
         logger.info(
             "{}: expanding {} X {}...".format(
                 NAME,
@@ -47,19 +42,19 @@ class BlueAmoScript(GenericScript):
         del self.nodes[holder_node_name]
         self.G.remove_node(holder_node_name)
 
-        reduce_node = "stitching-the-frames"
+        reduce_node = "stitching_the_frames"
         self.G.add_node(reduce_node)
         self.nodes[reduce_node] = {"action": "skip"}
 
         for index in range(self.vars["frame_count"]):
-            node_name = f"generating-frame-{index+1:03d}"
+            node_name = f"generating_frame_{index+1:03d}"
 
             self.nodes[node_name] = copy.deepcopy(holder_node)
 
             self.G.add_node(node_name)
             self.G.add_edge(
                 node_name,
-                "slicing-into-frames",
+                "slicing_into_frames",
             )
             self.G.add_edge(
                 reduce_node,
@@ -75,14 +70,8 @@ class BlueAmoScript(GenericScript):
         if not super().perform_action(node_name=node_name):
             return False
 
-        if node_name == "slicing-into-frames":
-            return slice_into_frames(
-                script=self,
-                node_name=node_name,
-            )
-
-        if node_name == "stitching-the-frames":
-            return stitch_the_frames(
+        if node_name in dict_of_actions:
+            return dict_of_actions[node_name](
                 script=self,
                 node_name=node_name,
             )

{blue_assistant-4.97.1 → blue_assistant-4.121.1}/blue_assistant/script/repository/generic/classes.py

@@ -9,7 +9,7 @@ from blue_objects.metadata import post_to_object
 
 from blue_assistant import NAME
 from blue_assistant.script.repository.base.classes import BaseScript
-from blue_assistant.script.actions.functions import get_action_class
+from blue_assistant.script.actions import dict_of_actions
 from blue_assistant.logger import logger
 
 
@@ -24,25 +24,15 @@ class GenericScript(BaseScript):
         node_name: str,
     ) -> bool:
         action_name = self.nodes[node_name].get("action", "unknown")
-        if action_name == "skip":
-            return True
-
-        success, action_class = get_action_class(action_name=action_name)
-        if not success:
-            return success
-
-        logger.info(
-            "{}.perform_action: {} == {} on {}".format(
-                NAME,
-                action_name,
-                action_class.__name__,
-                node_name,
-            )
-        )
 
-        action_object = action_class(script=self)
+        if action_name not in dict_of_actions:
+            logger.error(f"{action_name}: action not found.")
+            return False
 
-        return action_object.perform(node_name=node_name)
+        return dict_of_actions[action_name](
+            script=self,
+            node_name=node_name,
+        )
 
     def run(
         self,

{blue_assistant-4.97.1 → blue_assistant-4.121.1/blue_assistant.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: blue_assistant
-Version: 4.97.1
+Version: 4.121.1
 Summary: 🧠 An AI Assistant.
 Home-page: https://github.com/kamangir/blue-assistant
 Author: Arash Abadpour (Kamangir)
@@ -67,9 +67,13 @@ graph LR
     classDef folder fill:#999,stroke:#333,stroke-width:2px;
 ```
 
+|   |
+| --- |
+| [`blue-amo`](https://raw.githubusercontent.com/kamangir/blue-assistant/main/blue_assistant/script/repository/blue_amo/README.md) [![image](https://github.com/kamangir/assets/raw/main/blue-amo-2025-02-03-nswnx6/stitching_the_frames-2.png?raw=true)](https://raw.githubusercontent.com/kamangir/blue-assistant/main/blue_assistant/script/repository/blue_amo/README.md) story-telling with AI |
+
 ---
 
 
 [![pylint](https://github.com/kamangir/blue-assistant/actions/workflows/pylint.yml/badge.svg)](https://github.com/kamangir/blue-assistant/actions/workflows/pylint.yml) [![pytest](https://github.com/kamangir/blue-assistant/actions/workflows/pytest.yml/badge.svg)](https://github.com/kamangir/blue-assistant/actions/workflows/pytest.yml) [![bashtest](https://github.com/kamangir/blue-assistant/actions/workflows/bashtest.yml/badge.svg)](https://github.com/kamangir/blue-assistant/actions/workflows/bashtest.yml) [![PyPI version](https://img.shields.io/pypi/v/blue-assistant.svg)](https://pypi.org/project/blue-assistant/) [![PyPI - Downloads](https://img.shields.io/pypi/dd/blue-assistant)](https://pypistats.org/packages/blue-assistant)
 
-built by 🌀 [`blue_options-4.207.1`](https://github.com/kamangir/awesome-bash-cli), based on 🧠 [`blue_assistant-4.97.1`](https://github.com/kamangir/blue-assistant).
+built by 🌀 [`blue_options-4.207.1`](https://github.com/kamangir/awesome-bash-cli), based on 🧠 [`blue_assistant-4.121.1`](https://github.com/kamangir/blue-assistant).

{blue_assistant-4.97.1 → blue_assistant-4.121.1}/blue_assistant.egg-info/SOURCES.txt

@@ -40,17 +40,16 @@ blue_assistant/script/__init__.py
 blue_assistant/script/__main__.py
 blue_assistant/script/load.py
 blue_assistant/script/actions/__init__.py
-blue_assistant/script/actions/functions.py
 blue_assistant/script/actions/generate_image.py
 blue_assistant/script/actions/generate_text.py
 blue_assistant/script/actions/generic.py
-blue_assistant/script/actions/wip.py
+blue_assistant/script/actions/skip.py
 blue_assistant/script/repository/__init__.py
 blue_assistant/script/repository/blue_amo/__init__.py
 blue_assistant/script/repository/blue_amo/classes.py
 blue_assistant/script/repository/blue_amo/actions/__init__.py
-blue_assistant/script/repository/blue_amo/actions/slice_into_frames.py
-blue_assistant/script/repository/blue_amo/actions/stitch_the_frames.py
+blue_assistant/script/repository/blue_amo/actions/slicing_into_frames.py
+blue_assistant/script/repository/blue_amo/actions/stitching_the_frames.py
 blue_assistant/script/repository/generic/__init__.py
 blue_assistant/script/repository/generic/classes.py
 blue_assistant/script/repository/moon_datasets/__init__.py

blue_assistant-4.97.1/blue_assistant/README.py

@@ -1,33 +0,0 @@
-import os
-
-from blue_objects import file, README
-
-from blue_assistant import NAME, VERSION, ICON, REPO_NAME
-
-
-items = [
-    "{}[`{}`](#) [![image]({})](#) {}".format(
-        ICON,
-        f"feature {index}",
-        "https://github.com/kamangir/assets/raw/main/blue-assistant/marquee.png?raw=true",
-        f"description of feature {index} ...",
-    )
-    for index in range(1, 4)
-]
-
-
-def build():
-    return all(
-        README.build(
-            items=readme.get("items", []),
-            path=os.path.join(file.path(__file__), readme["path"]),
-            ICON=ICON,
-            NAME=NAME,
-            VERSION=VERSION,
-            REPO_NAME=REPO_NAME,
-        )
-        for readme in [
-            {"items": items, "path": ".."},
-            {"path": "docs/blue-amo-01.md"},
-        ]
-    )

blue_assistant-4.97.1/blue_assistant/script/actions/__init__.py

@@ -1,14 +0,0 @@
-from typing import List
-
-from blue_assistant.script.actions.generic import GenericAction
-from blue_assistant.script.actions.generate_image import GenerateImageAction
-from blue_assistant.script.actions.generate_text import GenerateTextAction
-from blue_assistant.script.actions.wip import WorkInProgressAction
-from blue_assistant.logger import logger
-
-list_of_actions: List[GenericAction] = [
-    GenericAction,
-    GenerateImageAction,
-    GenerateTextAction,
-    WorkInProgressAction,
-]

blue_assistant-4.97.1/blue_assistant/script/actions/functions.py

@@ -1,22 +0,0 @@
-from typing import List, Dict, Tuple, Type
-
-from blueness import module
-from blue_assistant.script.actions.generic import GenericAction
-from blue_assistant.script.repository.base.classes import BaseScript
-from blue_assistant.script.actions import list_of_actions
-
-from blue_assistant import NAME
-from blue_assistant.logger import logger
-
-NAME = module.name(__file__, NAME)
-
-
-def get_action_class(
-    action_name: str,
-) -> Tuple[bool, Type[GenericAction]]:
-    for action_class in list_of_actions:
-        if action_class.name == action_name:
-            return True, action_class
-
-    logger.error(f"{action_name}: action not found.")
-    return False, GenericAction

blue_assistant-4.97.1/blue_assistant/script/actions/generate_image.py

@@ -1,58 +0,0 @@
-from blue_objects import file, objects
-from openai_commands.image_generation.api import OpenAIImageGenerator
-
-from blue_assistant.env import (
-    BLUE_ASSISTANT_IMAGE_DEFAULT_MODEL,
-    BLUE_ASSISTANT_IMAGE_DEFAULT_SIZE,
-    BLUE_ASSISTANT_IMAGE_DEFAULT_QUALITY,
-)
-from blue_assistant.script.actions.generic import GenericAction
-from blue_assistant.script.repository.base.classes import BaseScript
-from blue_assistant.logger import logger
-
-
-class GenerateImageAction(GenericAction):
-    name = file.name(__file__)
-
-    def __init__(
-        self,
-        script: BaseScript,
-    ):
-        super().__init__(script=script)
-
-        self.generator = OpenAIImageGenerator(
-            model=BLUE_ASSISTANT_IMAGE_DEFAULT_MODEL,
-            verbose=self.script.verbose,
-        )
-
-    # https://platform.openai.com/docs/guides/images
-    def perform(
-        self,
-        node_name: str,
-    ) -> bool:
-        if not super().perform(node_name=node_name):
-            return False
-
-        filename = f"{node_name}.png"
-
-        success = self.generator.generate(
-            prompt=self.script.nodes[node_name]["prompt"],
-            filename=objects.path_of(
-                filename=filename,
-                object_name=self.script.object_name,
-                create=True,
-            ),
-            quality=(
-                BLUE_ASSISTANT_IMAGE_DEFAULT_QUALITY if self.script.test_mode else "hd"
-            ),
-            size=(
-                BLUE_ASSISTANT_IMAGE_DEFAULT_SIZE
-                if self.script.test_mode
-                else "1792x1024"
-            ),
-        )[0]
-
-        if success:
-            self.script.nodes[node_name]["filename"] = filename
-
-        return success

blue_assistant-4.97.1/blue_assistant/script/actions/generate_text.py

@@ -1,97 +0,0 @@
-from typing import Dict, Tuple, List
-from openai import OpenAI
-import pprint
-
-from blueness import module
-from blue_objects import file
-from openai_commands.env import OPENAI_API_KEY
-
-from blue_assistant import NAME
-from blue_assistant.env import (
-    BLUE_ASSISTANT_TEXT_DEFAULT_MODEL,
-    BLUE_ASSISTANT_TEXT_MAX_TOKEN,
-)
-from blue_assistant.script.actions.generic import GenericAction
-from blue_assistant.logger import logger
-
-NAME = module.name(__file__, NAME)
-
-
-class GenerateTextAction(GenericAction):
-    name = file.name(__file__)
-
-    # https://platform.openai.com/docs/guides/text-generation
-    def perform(
-        self,
-        node_name: str,
-    ) -> bool:
-        if not OPENAI_API_KEY:
-            logger.error("OPENAI_API_KEY is not set.")
-            return False
-
-        if not super().perform(node_name=node_name):
-            return False
-
-        messages: List = []
-        list_of_context_nodes = self.script.get_history(node_name)
-        logger.info("node context: {}".format(" <- ".join(list_of_context_nodes)))
-        for context_node in reversed(list_of_context_nodes):
-            messages += [
-                {
-                    "role": "user",
-                    "content": [
-                        {
-                            "type": "text",
-                            "text": self.script.apply_vars(
-                                self.script.nodes[context_node]["prompt"]
-                            ),
-                        }
-                    ],
-                }
-            ]
-
-            if self.script.nodes[context_node].get("completed", False):
-                messages += [
-                    {
-                        "role": "assistant",
-                        "content": [
-                            {
-                                "type": "text",
-                                "text": self.script.nodes[context_node]["output"],
-                            }
-                        ],
-                    }
-                ]
-
-        if self.script.verbose:
-            logger.info(f"messages: {pprint.pformat(messages)}")
-
-        client = OpenAI(api_key=OPENAI_API_KEY)
-
-        try:
-            response = client.chat.completions.create(
-                messages=messages,
-                model=BLUE_ASSISTANT_TEXT_DEFAULT_MODEL,
-                max_tokens=BLUE_ASSISTANT_TEXT_MAX_TOKEN,
-            )
-        except Exception as e:
-            logger.error(str(e))
-            return False
-
-        if self.script.verbose:
-            logger.info("response: {}".format(response))
-
-        if not response.choices:
-            logger.error("no choice.")
-            return False
-
-        output = response.choices[0].message.content
-        logger.info(f"🗣️ output: {output}")
-
-        self.script.nodes[node_name]["output"] = output
-
-        var_name = self.script.nodes[node_name].get("output", "")
-        if var_name:
-            self.script.vars[var_name] = output
-
-        return True

blue_assistant-4.97.1/blue_assistant/script/actions/generic.py

@@ -1,27 +0,0 @@
-from blue_objects import file
-
-from blue_assistant.script.repository.base.classes import BaseScript
-from blue_assistant.logger import logger
-
-
-class GenericAction:
-    name = file.name(__file__)
-
-    def __init__(
-        self,
-        script: BaseScript,
-    ):
-        self.script = script
-
-    def perform(
-        self,
-        node_name: str,
-    ) -> bool:
-        logger.info(
-            "{}.perform on {}.{} ...".format(
-                self.__class__.__name__,
-                self.script.name,
-                node_name,
-            ),
-        )
-        return True

blue_assistant-4.97.1/blue_assistant/script/actions/wip.py

@@ -1,7 +0,0 @@
-from blue_objects import file
-
-from blue_assistant.script.actions.generic import GenericAction
-
-
-class WorkInProgressAction(GenericAction):
-    name = file.name(__file__)

blue_assistant-4.97.1/blue_assistant/script/repository/blue_amo/actions/stitch_the_frames.py

@@ -1,81 +0,0 @@
-from typing import List
-import numpy as np
-import cv2
-from tqdm import trange
-
-from blueness import module
-from blue_objects import file, objects
-from blue_options import string
-
-from blue_assistant import NAME
-from blue_assistant.script.repository.base.classes import BaseScript
-from blue_assistant.logger import logger
-
-NAME = module.name(__file__, NAME)
-
-
-def stitch_the_frames(
-    script: BaseScript,
-    node_name: str,
-) -> bool:
-    list_of_frames_filenames: List[str] = [
-        filename
-        for filename in [
-            script.nodes[node_name_].get("filename", "")
-            for node_name_ in [
-                f"generating-frame-{index+1:03d}"
-                for index in range(script.vars["frame_count"])
-            ]
-        ]
-        if filename
-    ]
-    if not list_of_frames_filenames:
-        return True
-
-    logger.info(
-        "{} frames to stitch: {}".format(
-            len(list_of_frames_filenames),
-            ", ".join(list_of_frames_filenames),
-        )
-    )
-
-    list_of_frames: List[np.ndarray] = []
-    for filename in list_of_frames_filenames:
-        success, frame = file.load_image(
-            objects.path_of(
-                filename=filename,
-                object_name=script.object_name,
-            )
-        )
-
-        if success:
-            list_of_frames += [frame]
-
-    if not list_of_frames:
-        return True
-
-    common_height = list_of_frames[0].shape[0]
-    for index in trange(len(list_of_frames)):
-        if list_of_frames[index].shape[0] != common_height:
-            aspect_ratio = (
-                list_of_frames[index].shape[1] / list_of_frames[index].shape[0]
-            )
-            new_width = int(common_height * aspect_ratio)
-
-            list_of_frames[index] = cv2.resize(
-                list_of_frames[index],
-                (new_width, common_height),
-                interpolation=cv2.INTER_AREA,
-            )
-
-    full_frame = np.concatenate(list_of_frames, axis=1)
-    logger.info(f"full_frame: {string.pretty_shape_of_matrix(full_frame)}")
-
-    return file.save_image(
-        objects.path_of(
-            filename=f"{node_name}.png",
-            object_name=script.object_name,
-        ),
-        full_frame,
-        log=True,
-    )