Run

from typing import List, Literal

from metaflow import FlowSpec, JSONType, Parameter, step
from pydantic import BaseModel


class ImagePath(BaseModel):
    img_path: str


class SampleFlowResult(ImagePath):
    """
    Sample Flow workflow response
    """

    result: Literal["success", "failed", "in_progress"]


class SampleFlowResponse(BaseModel):
    """
    Sample flow response
    """

    operation: Literal["sample_flow"]
    response: List[SampleFlowResult]


class SampleFlowWorkflowInput(BaseModel):
    request_id: str
    images: list[str]


class SimpleFlow(FlowSpec):
    input_payload = Parameter(
        "input_payload", help="The payload containing the images to be processed.", required=True, type=JSONType
    )

    some_key_1 = Parameter(
        "some_key_1", help="Some Extra Parameter provided to the workflow", type=str, default="key_1"
    )

    some_key_2 = Parameter(
        "some_key_2",
        help="Some Extra Parameter provided to the workflow",
        type=str,
        default="key_2",
    )

    @step
    def start(self) -> None:
        print("This is the start step.")
        self.next(self.process_data)

    @step
    def process_data(self) -> None:
        print("This is the data processing step.")
        self.next(self.end)

    @step
    def end(self) -> None:
        
        self.output_payload = SampleFlowResponse(
            operation="sample_flow",
            response=[
                SampleFlowResult(
                    img_path="path/to/image.jpg",
                    result="success",
                )
            ],
        ).model_dump_json()
        print("This is the end step.")


if __name__ == '__main__':
    SimpleFlow()

self.process = subprocess.Popen(
                    shlex.join(self.command),
                    cwd=self.cwd,
                    env=self.env,
                    stdout=subprocess.PIPE,
                    stderr=subprocess.PIPE,
                    bufsize=1,
                    universal_newlines=True,
                    shell=True
                )

shell=True