state.py

from typing import TypedDict, List, Dict, Any, Optional


class AgentState(TypedDict):

    # -------------------------------
    # User Input
    # -------------------------------
    user_input: str

    # -------------------------------
    # Uploaded Images
    # -------------------------------
    uploaded_files: List[str]

    # Current working image
    current_image: Optional[str]

    # Latest edited image
    last_image: Optional[str]

    # -------------------------------
    # Image Analysis
    # -------------------------------
    analysis: Optional[Dict[str, Any]]

    # Natural language description
    image_context: str

    # -------------------------------
    # Planner
    # -------------------------------
    task: str

    steps: List[str]

    # -------------------------------
    # Conversation
    # -------------------------------
    history: List[Dict[str, str]]

    # -------------------------------
    # Final Output
    # -------------------------------
    result: Any