Update block.py

block.py

@@ -1,12 +1,13 @@
 from typing import List
 import torch
 
-from diffusers.pipelines.modular_pipeline import PipelineState, PipelineBlock
+from diffusers.pipelines.modular_pipeline import PipelineState, PipelineBlock, SequentialPipelineBlocks, AutoPipelineBlocks
 from diffusers.pipelines.modular_pipeline_utils import (
     InputParam,
     ComponentSpec,
     OutputParam,
 )
+from diffusers.utils import load_image
 from diffusers.image_processor import PipelineImageInput
 from image_gen_aux import DepthPreprocessor
 
@@ -28,11 +29,20 @@ class DepthProcessorBlock(PipelineBlock):
             InputParam(
                 "image",
                 PipelineImageInput,
-                required=True,
                 description="Image(s) to use to extract depth maps",
             )
         ]
 
+    @property
+    def intermediates_inputs(self) -> List[InputParam]:
+        return [
+            InputParam(
+                "image",
+                PipelineImageInput,
+                description="Image(s) to use to extract depth maps, can be output from LoadURL block",
+            )
+        ]
+
     @property
     def intermediates_outputs(self) -> List[OutputParam]:
         return [
@@ -53,3 +63,45 @@ class DepthProcessorBlock(PipelineBlock):
 
         self.add_block_state(state, block_state)
         return pipeline, state
+
+class LoadURL(PipelineBlock):
+
+    @property
+    def inputs(self) -> List[InputParam]:
+        return [
+            InputParam(
+                "url",
+                str,
+            )
+        ]
+
+    @property
+    def intermediates_outputs(self) -> List[OutputParam]:
+        return [
+            OutputParam(
+                "image",
+                type_hint=PipelineImageInput,
+                description="Image(s) to use to extract depth maps",
+            ),
+        ]
+    
+    def __call__(self, pipeline, state: PipelineState) -> PipelineState:
+        block_state = self.get_block_state(state)
+        block_state.image = load_image(blck_state.url)
+        self.add_block_state(state, block_state)
+        return pipeline, state
+
+class AutoLoadURL(AutoPipelineBlocks):
+    block_classes = [LoadURL]
+    block_name = ["url_to_image"]
+    block_trigger_inputs = ["url"]
+
+    @property
+    def description(self):
+        return "Run if `url` is provided."
+
+class DepthInput(SequentialPipelineBlocks)
+    block_classes = [AutoLoadURL, DepthProcessorBlock]
+    block_names = ["load_url", "depth_processor"]
+    
+