clean scene 13

app.py

@@ -1,3 +1,5 @@
+
+
 import os
 import subprocess
 
@@ -25,10 +27,6 @@ from inference_triposg import run_triposg
 from triposg.pipelines.pipeline_triposg import TripoSGPipeline
 from briarmbg import BriaRMBG
 
-from pygltflib import GLTF2, Scene, Node, Mesh, Buffer, BufferView, Accessor, BufferTarget, ComponentType, AccessorType
-import numpy as np
-import base64
-
 
 print("Trimesh version:", trimesh.__version__)
 
@@ -71,6 +69,7 @@ def generate(image_path, face_number=50000, guidance_scale=5.0, num_steps=25):
 
     temp_id = str(uuid.uuid4())
     output_path = f"/tmp/{temp_id}.glb"
+    print("[DEBUG] Generating mesh from:", image_path)
 
     try:
         mesh = run_triposg(
@@ -86,62 +85,21 @@ def generate(image_path, face_number=50000, guidance_scale=5.0, num_steps=25):
         if mesh is None or mesh.vertices.shape[0] == 0 or mesh.faces.shape[0] == 0:
             raise ValueError("Mesh generation returned an empty mesh")
 
-        vertices = mesh.vertices.astype(np.float32)
-        indices = mesh.faces.astype(np.uint32).flatten()
+        # Безопасная очистка визуала
+        if hasattr(mesh, "visual") and mesh.visual is not None:
+            try:
+                mesh.visual = None
+            except Exception:
+                print("[WARN] Failed to clear visual, skipping")
 
-        # Pack binary data
-        vertex_bytes = vertices.tobytes()
-        index_bytes = indices.tobytes()
-        total_bytes = vertex_bytes + index_bytes
+        mesh.metadata.clear()
+        mesh.name = "endless_tools_mesh"
 
-        buffer = Buffer(byteLength=len(total_bytes))
-        buffer_view_vert = BufferView(
-            buffer=0,
-            byteOffset=0,
-            byteLength=len(vertex_bytes),
-            target=BufferTarget.ARRAY_BUFFER.value
-        )
-        buffer_view_index = BufferView(
-            buffer=0,
-            byteOffset=len(vertex_bytes),
-            byteLength=len(index_bytes),
-            target=BufferTarget.ELEMENT_ARRAY_BUFFER.value
-        )
-
-        accessor_vert = Accessor(
-            bufferView=0,
-            byteOffset=0,
-            componentType=ComponentType.FLOAT.value,
-            count=len(vertices),
-            type=AccessorType.VEC3.value,
-            min=vertices.min(axis=0).tolist(),
-            max=vertices.max(axis=0).tolist()
-        )
-
-        accessor_index = Accessor(
-            bufferView=1,
-            byteOffset=0,
-            componentType=ComponentType.UNSIGNED_INT.value,
-            count=len(indices),
-            type=AccessorType.SCALAR.value
-        )
-
-        gltf = GLTF2(
-            buffers=[buffer],
-            bufferViews=[buffer_view_vert, buffer_view_index],
-            accessors=[accessor_vert, accessor_index],
-            meshes=[Mesh(primitives=[{
-                "attributes": {"POSITION": 0},
-                "indices": 1
-            }])],
-            scenes=[Scene(nodes=[0])],
-            nodes=[Node(mesh=0)],
-            scene=0
-        )
-
-        # Inject binary blob
-        gltf.set_binary_blob(total_bytes)
-        gltf.save_binary(output_path)
+        # Экспорт .glb
+        # glb_data = export_glb(mesh)
+        glb_data = mesh.export(file_type='glb')
+        with open(output_path, "wb") as f:
+            f.write(glb_data)
 
         print(f"[DEBUG] Mesh saved to {output_path}")
         return output_path if os.path.exists(output_path) else None
@@ -150,7 +108,7 @@ def generate(image_path, face_number=50000, guidance_scale=5.0, num_steps=25):
         print("[ERROR]", e)
         traceback.print_exc()
         return f"Error: {e}"
-    
+
 # Интерфейс Gradio
 demo = gr.Interface(
     fn=generate,

requirements.txt

@@ -15,4 +15,4 @@ jaxtyping
 typeguard
 pymeshlab
 einops
-pygltflib==1.15.6
+