Merge pull request #955 from roboflow/block-gallery-refactor

Block gallery refactor
roboflow · Jan 24, 2025 · cf7d4bb · cf7d4bb
2 parents 61aab85 + 9a5b45b
commit cf7d4bb
Show file tree

Hide file tree

Showing 10 changed files with 354 additions and 132 deletions.
diff --git a/development/docs/build_block_docs.py b/development/docs/build_block_docs.py
diff --git a/development/docs/templates/blocks_index.md b/development/docs/templates/blocks_index.md
@@ -0,0 +1,79 @@
+---
+hide:
+  - toc
+---
+<script src="https://cdnjs.cloudflare.com/ajax/libs/dompurify/3.0.8/purify.min.js"></script>
+<link rel="stylesheet" href="/styles/workflows.css">
+
+<script src="https://kit.fontawesome.com/c9c3956d39.js" crossorigin="anonymous"></script>
+
+
+<section class="mdx-container portfolio-section">
+  <div class="md-grid md-typeset">
+    <div class="text-center">
+      <h1>Workflow Blocks</h1>
+    </div>
+
+      {% for section in block_sections %}
+      <div class="section">
+        <h2>{{ section.title | capitalize }}</h2>
+        <div class="blocks">
+            <div class="custom-grid">
+            {% for block in blocks_by_section[section.id] %}
+
+
+
+            <a href="{{block.url}}">
+              <div class="block">
+                
+                <div class="block_name"> <i class="{{block.icon}}" > </i>  {{ block.name }}</div>
+                <div class="block_description">{{ block.description }}</div>
+                <!-- <div class="block_license">{{ block.license }}</div> -->
+              </div>
+            </a>
+
+
+          
+          {% endfor %}
+          </div>
+        </div> 
+      </div>
+      {% endfor %}
+    </div>
+  </div>
+</section>
+
+<style>
+/* hide edit button for generated pages */
+article > a.md-content__button.md-icon:first-child {
+    display: none;
+}
+
+.block {
+  border: 1px solid black;
+  border-radius: 4px;
+  padding: 10px;
+  height: 100px;
+}
+
+.block_name {
+  font-size: large;
+  color: black;
+}
+
+.block_description {
+    font-size: 0.65em;
+    padding-top: 5px;
+    color: #444;
+}
+
+.block_license {
+  background-color: #14b8a6; 
+  color: #fff; 
+  padding: 2px 4px;
+  border-radius: 4px; 
+  font-size: small;
+}
+
+
+</style>
diff --git a/docs/api.md b/docs/api.md
@@ -1,3 +1,7 @@
+---
+hide:
+  - toc
+---
 When the Inference Server is running, it provides OpenAPI documentation at the `/docs` endpoint for use in development.
 
 Below is the OpenAPI specification for the Inference Server for the current release version.

diff --git a/inference/core/workflows/core_steps/classical_cv/dominant_color/v1.py b/inference/core/workflows/core_steps/classical_cv/dominant_color/v1.py
@@ -44,6 +44,12 @@ class DominantColorManifest(WorkflowBlockManifest):
             "long_description": LONG_DESCRIPTION,
             "license": "Apache-2.0",
             "block_type": "classical_computer_vision",
+            "ui_manifest": {
+                "section": "classical_cv",
+                "icon": "far fa-palette",
+                "blockPriority": 1,
+                "opencv": True,
+            },
         }
     )
     image: Selector(kind=[IMAGE_KIND]) = Field(

diff --git a/inference/core/workflows/core_steps/classical_cv/image_blur/v1.py b/inference/core/workflows/core_steps/classical_cv/image_blur/v1.py
@@ -40,6 +40,12 @@ class ImageBlurManifest(WorkflowBlockManifest):
             "long_description": LONG_DESCRIPTION,
             "license": "Apache-2.0",
             "block_type": "classical_computer_vision",
+            "ui_manifest": {
+                "section": "classical_cv",
+                "icon": "far fa-droplet",
+                "blockPriority": 5,
+                "opencv": True,
+            },
         }
     )
 

diff --git a/inference/core/workflows/core_steps/models/foundation/ocr/v1.py b/inference/core/workflows/core_steps/models/foundation/ocr/v1.py
@@ -67,6 +67,13 @@ class BlockManifest(WorkflowBlockManifest):
             "long_description": LONG_DESCRIPTION,
             "license": "Apache-2.0",
             "block_type": "model",
+            "ui_manifest": {
+                "section": "model",
+                "icon": "far fa-text",
+                "blockPriority": 11,
+                "inDevelopment": True,
+                "inference": True,
+            },
         }
     )
     type: Literal["roboflow_core/ocr_model@v1", "OCRModel"]

diff --git a/inference/core/workflows/core_steps/sinks/email_notification/v1.py b/inference/core/workflows/core_steps/sinks/email_notification/v1.py
@@ -172,7 +172,7 @@ class BlockManifest(WorkflowBlockManifest):
             "license": "Apache-2.0",
             "block_type": "sink",
             "ui_manifest": {
-                "section": "model",
+                "section": "notifications",
                 "icon": "far fa-envelope",
                 "blockPriority": 0,
                 "popular": True,

diff --git a/inference/core/workflows/core_steps/sinks/local_file/v1.py b/inference/core/workflows/core_steps/sinks/local_file/v1.py
@@ -74,7 +74,7 @@ class BlockManifest(WorkflowBlockManifest):
             "license": "Apache-2.0",
             "block_type": "sink",
             "ui_manifest": {
-                "section": "model",
+                "section": "data_storage",
                 "icon": "fal fa-file",
                 "blockPriority": 3,
                 "popular": True,

diff --git a/inference/core/workflows/core_steps/transformations/absolute_static_crop/v1.py b/inference/core/workflows/core_steps/transformations/absolute_static_crop/v1.py
@@ -39,6 +39,11 @@ class BlockManifest(WorkflowBlockManifest):
             "long_description": LONG_DESCRIPTION,
             "license": "Apache-2.0",
             "block_type": "transformation",
+            "ui_manifest": {
+                "section": "transformation",
+                "icon": "far fa-crop-alt",
+                "blockPriority": 1,
+            },
         }
     )
     type: Literal["roboflow_core/absolute_static_crop@v1", "AbsoluteStaticCrop"]

diff --git a/inference/core/workflows/core_steps/visualizations/dot/v1.py b/inference/core/workflows/core_steps/visualizations/dot/v1.py
@@ -38,6 +38,12 @@ class DotManifest(ColorableVisualizationManifest):
             "long_description": LONG_DESCRIPTION,
             "license": "Apache-2.0",
             "block_type": "visualization",
+            "ui_manifest": {
+                "section": "classical_cv",
+                "icon": "far fa-palette",
+                "blockPriority": 1,
+                "opencv": True,
+            },
         }
     )