import gradio as gr
from gradio_client import Client, handle_file

trellis = Client("microsoft/TRELLIS.2")

def generate_3d(image):
    result = trellis.predict(
        image=handle_file(image),
        api_name="/generate"
    )
    return result

demo = gr.Interface(
    fn=generate_3d,
    inputs=gr.Image(type="filepath", label="Sube tu imagen"),
    outputs=gr.Model3D(label="Modelo 3D generado"),
    title="🚀 Henry3D-AI",
    description="Sube una imagen y genera un modelo 3D con TRELLIS."
)

demo.launch()