import gradio as gr
from gradio_client import Client

def generate_3d(image):
    try:
        client = Client("stabilityai/TripoSR")
        result = client.predict(
            image,
            True,
            0.85,
            api_name="/generate"
        )
        return result[0]
    except Exception as e:
        raise gr.Error(f"Error: {str(e)}")

demo = gr.Interface(
    fn=generate_3d,
    inputs=gr.Image(type="filepath", label="Image Upload karo"),
    outputs=gr.Model3D(label="3D Model"),
    title="PixelForge AI - Image to 3D",
    description="Game developers ke liye AI powered 3D model generator",
)

demo.launch()