Update app.py

app.py

@@ -1,38 +1,92 @@
 import gradio as gr
 import trimesh
 from instant_texture import Converter
+import os
+import uuid
+from PIL import Image
 
+# Inicializa o conversor
 converter = Converter()
 
-
 def extract_texture(mesh_path: str):
-    mesh = trimesh.load(mesh_path)
-    mesh = next(iter(mesh.geometry.values()))
-    texture = mesh.visual.material.baseColorTexture
-    return texture
+    """Tenta extrair a textura base do arquivo GLB gerado."""
+    try:
+        # Carrega a cena GLB
+        scene = trimesh.load(mesh_path)
+        
+        # GLBs são carregados como 'Scenes'. Precisamos pegar a primeira malha (geometry) da cena.
+        # Se a cena estiver vazia, isso vai gerar um erro que o 'except' pega.
+        if not scene.geometry:
+             print("Erro: Nenhuma geometria encontrada no GLB gerado.")
+             return None
+             
+        mesh = next(iter(scene.geometry.values()))
+
+        # Verifica se a malha tem material e se o material tem a textura de cor base
+        if hasattr(mesh.visual, 'material') and hasattr(mesh.visual.material, 'baseColorTexture'):
+            texture = mesh.visual.material.baseColorTexture
+            # Verifica se a textura é uma imagem PIL válida antes de retornar
+            if isinstance(texture, Image.Image):
+                return texture
+        
+        print("Aviso: O modelo GLB foi gerado, mas nenhuma textura baseColorTexture foi encontrada.")
+        return None
 
+    except Exception as e:
+        print(f"Erro ao extrair textura: {e}")
+        return None
 
 def convert(input_mesh_path: str):
+    # Validação de entrada
+    if input_mesh_path is None:
+        raise gr.Error("Por favor, envie um arquivo .obj.")
+        
     if not input_mesh_path.endswith(".obj"):
-        raise ValueError("Only .obj files are supported")
-    output_path = "/tmp/output.glb"
-    converter.convert(input_mesh_path, output_path)
-    texture = extract_texture(output_path)
-    return output_path, texture
+        raise gr.Error("Apenas arquivos .obj são suportados.")
+
+    # --- MUDANÇA CRÍTICA 1: Arquivos Temporários Únicos ---
+    # Em um Space, múltiplos usuários podem usar o app ao mesmo tempo.
+    # Usar sempre "/tmp/output.glb" faria um usuário sobrescrever o arquivo do outro.
+    # Usamos UUID para criar um nome de arquivo único para cada execução.
+    unique_filename = str(uuid.uuid4())
+    output_path = os.path.join("/tmp", f"output_{unique_filename}.glb")
+
+    try:
+        # Executa a conversão (assumindo que esta biblioteca funciona corretamente)
+        converter.convert(input_mesh_path, output_path)
+
+        # Verifica se o arquivo de saída foi realmente criado
+        if not os.path.exists(output_path):
+             raise gr.Error("Falha na conversão: O arquivo de saída não foi gerado.")
+
+        # Extrai a textura do novo arquivo GLB
+        texture = extract_texture(output_path)
+        
+        if texture is None:
+             # Se não conseguiu extrair a textura, retorna o modelo, mas avisa no log
+             print("Retornando modelo sem imagem de textura separada.")
+             return output_path, None
 
+        return output_path, texture
+        
+    except Exception as e:
+        # Captura erros da biblioteca instant_texture e mostra pro usuário
+        raise gr.Error(f"Erro durante o processo de conversão: {str(e)}")
 
+# Configuração da Interface Gradio
 demo = gr.Interface(
-    convert,
+    fn=convert,
     title="Instant Texture",
     description="Convert a vertex-colored mesh (.obj) to a uv-mapped, textured mesh (.glb).",
     inputs=gr.Model3D(label="Vertex-colored mesh (.obj)"),
     outputs=[
         gr.Model3D(label="Output uv-mapped, textured mesh (.glb)"),
-        gr.Image(label="Output texture"),
+        gr.Image(label="Output texture", type="pil"), # Força o tipo PIL para garantir compatibilidade
     ],
-    examples=["examples/chair.obj"],
-    cache_examples=True,
+    # Verifique se o caminho do exemplo está correto no seu repositório
+    # examples=[["examples/chair.obj"]], 
+    cache_examples=False, # Desativado temporariamente para evitar erros se o exemplo não existir
 )
 
 if __name__ == "__main__":
-    demo.queue().launch()
+    demo.queue().launch()