import gradio as gr
from daggr import GradioNode, Graph

# Background removal: API returns (original, processed). Hide first with None, show final_image.
# hf-applications/background-removal runs on GPU (Zero); use merve/background-removal for CPU.
bg_remover = GradioNode(
    "hf-applications/background-removal",
    api_name="/image",
    run_locally=False,
    inputs={"image": gr.Image()},
    outputs={
        "original_image": None,
        "final_image": gr.Image(label="Processed Image"),
    },
)

# Add more nodes here (Video Gen, Watermark Remover, etc.) as needed
graph = Graph(name="Imageat Workflow", nodes=[bg_remover])
graph.launch()