Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import requests
 import os
+import zipfile
 
 API_URL = "https://api-inference.huggingface.co/models/stabilityai/TripoSR"
 HEADERS = {"Authorization": f"Bearer {os.environ.get('HF_TOKEN')}"}
@@ -10,12 +11,21 @@ def generate_3d_from_image(image):
         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:
+        output_zip = "model_output.zip"
+        with open(output_zip, "wb") as out_file:
             out_file.write(response.content)
-        return output_path
+
+        # Intentar extraer un archivo .glb o .ply desde el zip
+        with zipfile.ZipFile(output_zip, 'r') as zip_ref:
+            zip_ref.extractall("output_model")
+
+        for filename in zip_ref.namelist():
+            if filename.endswith(".glb") or filename.endswith(".ply"):
+                return os.path.join("output_model", filename)
+
+        raise gr.Error("El modelo no generó un archivo .glb o .ply compatible.")
     else:
-        return f"Error: {response.status_code} - {response.text}"
+        raise gr.Error(f"Error {response.status_code}: {response.text}")
 
 with gr.Blocks() as demo:
     gr.Markdown("## 🧠 Genera un modelo 3D real desde una imagen usando TripoSR")