Update app.py

app.py

@@ -1,19 +1,29 @@
 import gradio as gr
+import requests
+import os
 
-def generate_3d_model(image):
-    # Esta versión de prueba ignora la imagen y devuelve un modelo 3D GLB funcional
-    return "https://raw.githubusercontent.com/KhronosGroup/glTF-Sample-Models/master/2.0/Duck/glTF-Binary/Duck.glb"
+API_URL = "https://api-inference.huggingface.co/models/stabilityai/TripoSR"
+HEADERS = {"Authorization": f"Bearer {os.environ.get('HF_TOKEN')}"}
 
-with gr.Blocks() as demo:
-    gr.Markdown("## 🧠 Genera tu modelo 3D desde una imagen")
-    gr.Markdown("Sube una imagen y obtén un modelo 3D generado automáticamente.")
+def generate_3d_from_image(image):
+    with open(image, "rb") as f:
+        response = requests.post(API_URL, headers=HEADERS, files={"file": f})
+
+    if response.status_code == 200:
+        output_path = "output.glb"
+        with open(output_path, "wb") as out_file:
+            out_file.write(response.content)
+        return output_path
+    else:
+        return f"Error: {response.status_code} - {response.text}"
 
+with gr.Blocks() as demo:
+    gr.Markdown("## 🧠 Genera un modelo 3D real desde una imagen usando TripoSR")
     with gr.Row():
-        image_input = gr.Image(type="filepath", label="Sube tu imagen")
-    
-    btn = gr.Button("Generar modelo 3D")
-    output = gr.Model3D(label="Vista previa del modelo 3D")
+        image_input = gr.Image(type="filepath", label="Sube una imagen (.png o .jpg)")
+    generate_btn = gr.Button("Generar modelo 3D")
+    output_model = gr.Model3D(label="Vista previa del modelo 3D")
 
-    btn.click(fn=generate_3d_model, inputs=image_input, outputs=output)
+    generate_btn.click(fn=generate_3d_from_image, inputs=image_input, outputs=output_model)
 
 demo.launch()