Widget integration for Python tools

from ibm_watsonx_orchestrate.agent_builder.tools import tool
from ibm_watsonx_orchestrate.run.widgets.forms import FormWidget, TextInput, ToolEvent
from ibm_watsonx_orchestrate.run.tool_result import ToolResult, TextContent, Annotations, Role

@tool()
def show_contact_form():
    """Display contact information form."""
    form = FormWidget(
        title="Contact Information",
        inputs=[
            TextInput(
                name="name",
                title="Full Name",
                required=True
            ),
            TextInput(
                name="email",
                title="Email",
                required=True
            ),
            TextInput(
                name="phone",
                title="Phone Number",
                required=False
            )
        ],
        on_event=[
            ToolEvent(
                tool="save_contact",
                parameters={
                    "name": "",
                    "email": "",
                    "phone": ""
                },
                map_input_to="submit"
            )
        ]
    )
    
    return ToolResult(
        content=[
            TextContent(
                text="Please provide your contact information",
                annotations=Annotations(
                    audience=[Role.USER]
                )
            )
        ],
        widget=form
    )

@tool()
def save_contact(name: str, email: str, phone: str = ""):
    """Save contact information."""
    # Process the contact data
    return ToolResult(
        content=[
            TextContent(
                text=f"Contact saved: {name} ({email})",
                annotations=Annotations(
                    audience=[Role.USER]
                )
            )
        ]
    )

@tool()
def show_registration_form():
    """Display registration form."""
    return {
        "content": [
            {
                "type": "text",
                "annotations": {
                    "audience": ["user"]
                },
                "text": "Please complete the registration form"
            }
        ],
        "_meta": {
            "com.ibm.orchestrate/widget": {
                "response_type": "forms",
                "json_schema": {
                    "title": "User Registration",
                    "type": "object",
                    "properties": {
                        "firstName": {"type": "string", "title": "First Name"},
                        "email": {"type": "string", "title": "Email"}
                    }
                },
                "ui_schema": {
                    "firstName": {"ui:widget": "TextWidget"},
                    "email": {"ui:widget": "TextWidget"}
                },
                "form_data": {"firstName": "", "email": ""},
                "on_event": [
                    {
                        "type": "tool",
                        "tool": "process_registration",  # Name of the tool to call when the event is triggered.
                        "parameters": {
                            # Each key must match a parameter name in process_registration(...).
                            # With map_input_to set to "submit", these fields are populated from the form submission.
                            "firstName": "",
                            "email": ""
                        },
                        "map_input_to": "submit"  # Maps submitted form values to the parameters above.
                    }
                ]
            }
        },
        "structuredContent": None,
        "context_updates": {}
    }

@tool()
def process_registration(firstName: str, email: str):
    """Process the registration form submission."""
    return {
        "content": [
            {
                "type": "text",
                "annotations": {
                    "audience": ["user"]
                },
                "text": f"Registration successful! Welcome, {firstName}. Confirmation sent to {email}."
            }
        ],
        "_meta": {},
        "structuredContent": None,
        "context_updates": {}
    }

from ibm_watsonx_orchestrate.agent_builder.tools import tool
from ibm_watsonx_orchestrate.run.widgets.forms import FormWidget, TextArea, RadioButton, MessageEvent
from ibm_watsonx_orchestrate.run.tool_result import ToolResult, TextContent, Annotations, Role

@tool()
def show_survey_form():
    """Display a simple survey form."""
    form = FormWidget(
        title="Quick Survey",
        description="Help us improve by sharing your feedback",
        inputs=[
            RadioButton(
                name="satisfaction",
                title="How satisfied are you?",
                options=["very_satisfied", "satisfied", "neutral", "dissatisfied"],
                option_labels=["Very Satisfied", "Satisfied", "Neutral", "Dissatisfied"],
                required=True
            ),
            TextArea(
                name="feedback",
                title="Additional Comments",
                placeholder="Share your thoughts...",
                required=False
            )
        ],
        submit_text="Submit Survey",
        on_event=[
            MessageEvent(
                message="Thank you for completing the survey! Your feedback helps us improve.",
                map_input_to="submit"
            )
        ]
    )
    
    return ToolResult(
        content=[
            TextContent(
                text="Please take a moment to complete our survey",
                annotations=Annotations(
                    audience=[Role.USER]
                )
            )
        ],
        widget=form
    )

@tool()
def show_feedback_form():
    """Display feedback form that sends a message to the agent."""
    return {
        "content": [
            {
                "type": "text",
                "annotations": {
                    "audience": ["user"]
                },
                "text": "Please provide your feedback"
            }
        ],
        "_meta": {
            "com.ibm.orchestrate/widget": {
                "response_type": "forms",
                "json_schema": {
                    "title": "Feedback Form",
                    "type": "object",
                    "properties": {
                        "rating": {"type": "string", "title": "Rating"},
                        "comments": {"type": "string", "title": "Comments"}
                    }
                },
                "ui_schema": {
                    "rating": {"ui:widget": "RadioButtonWidget"},
                    "comments": {"ui:widget": "TextAreaWidget"}
                },
                "form_data": {"rating": "", "comments": ""},
                "on_event": [
                    {
                        "type": "message",
                        "message": "Thank you for your feedback! Your response has been recorded.",
                        "map_input_to": "submit"
                    }
                ]
            }
        },
        "structuredContent": None,
        "context_updates": {}
    }

from ibm_watsonx_orchestrate.agent_builder.tools import tool
from datetime import datetime, timezone

@tool
def get_custom_widget(
    widget_type: str = "custom_component",
    title: str = "Custom Widget",
    message: str = "This is a custom user-defined widget",
    action_url: str = "https://ibm.com"
):
    """
    Generate a custom widget with user_defined response type.
    
    This tool creates a custom widget that can be rendered in the watsonx Orchestrate
    embedded chat interface using custom rendering logic.
    """
    
    return {
        "content": [
            {
                "type": "text",
                "annotations": {
                    "audience": ["user"]
                },
                "text": f"{title}: {message}"
            }
        ],
        "_meta": {
            "com.ibm.orchestrate/widget": {
                "response_type": "user_defined",
                "user_defined": {
                    "user_defined_type": widget_type,
                    "title": title,
                    "text": message,
                    "timestamp": datetime.now(timezone.utc).isoformat(),
                    "action_url": action_url,
                    "custom_field": "You can add any custom data here",
                    "interactive": True
                }
            }
        },
        "structuredContent": None,
        "context_updates": {}
    }