Quick Start

import asyncio
from nadoo_flow import (
    ChainableNode,
    FunctionNode,
    NodeResult,
    WorkflowContext
)

# Step 1: Define custom nodes
class TextTransformNode(ChainableNode):
    """Transform text to uppercase"""

    def __init__(self):
        super().__init__(
            node_id="transform",
            node_type="transform",
            name="Text Transformer"
        )

    async def execute(self, node_context, workflow_context):
        text = node_context.input_data.get("text", "")

        # Transform the text
        transformed = text.upper()

        # Store in workflow context for later use
        workflow_context.set_global_variable("original_text", text)

        return NodeResult(
            success=True,
            output={"text": transformed}
        )

class SentimentAnalyzerNode(ChainableNode):
    """Analyze sentiment of text"""

    def __init__(self):
        super().__init__(
            node_id="sentiment",
            node_type="analysis",
            name="Sentiment Analyzer"
        )

    async def execute(self, node_context, workflow_context):
        text = node_context.input_data.get("text", "")

        # Simple sentiment analysis (in production, use a real model)
        positive_words = ["GOOD", "GREAT", "HAPPY", "LOVE", "EXCELLENT"]
        negative_words = ["BAD", "HATE", "TERRIBLE", "AWFUL", "HORRIBLE"]

        score = 0
        for word in positive_words:
            if word in text:
                score += 1
        for word in negative_words:
            if word in text:
                score -= 1

        sentiment = "positive" if score > 0 else "negative" if score < 0 else "neutral"

        return NodeResult(
            success=True,
            output={
                "text": text,
                "sentiment": sentiment,
                "score": score
            }
        )

# Step 2: Build the workflow chain
def create_workflow():
    """Create and return the workflow chain"""

    # Chain nodes together using pipe operator
    workflow = (
        TextTransformNode()
        | SentimentAnalyzerNode()
        | FunctionNode(lambda x: {
            "summary": f"Sentiment: {x['sentiment']} (score: {x['score']})",
            "processed_text": x['text']
        })
    )

    return workflow

# Step 3: Run the workflow
async def main():
    # Create the workflow
    workflow = create_workflow()

    # Test inputs
    test_inputs = [
        {"text": "I love this framework, it's excellent!"},
        {"text": "This is a neutral statement"},
        {"text": "I hate bugs, they're terrible"}
    ]

    print("🚀 Nadoo Flow Core - Quick Start Demo\n")
    print("=" * 50)

    for input_data in test_inputs:
        print(f"\nInput: {input_data['text']}")
        print("-" * 30)

        # Run the workflow
        result = await workflow.run(input_data)

        # Display results
        print(f"Result: {result['summary']}")
        print(f"Processed: {result['processed_text']}")

if __name__ == "__main__":
    asyncio.run(main())