Update app.py

app.py

@@ -1,39 +1,43 @@
 import gradio as gr
 import trimesh
+import numpy as np
 from PIL import Image
-import base64
 import io
 
-def modify_texture():
+def visualize_texture():
     # Load the GLB file
-    scene = trimesh.load_mesh("train.glb")
-    # Open the texture image
-    im = Image.open("defect.jpg")
-    
-    # Create a material with the texture image
+    mesh = trimesh.load('train.glb', file_type='glb', process=False)
+
+    # Load the texture image
+    im = Image.open('defect.jpg')
+    im = im.convert('RGB')  # Ensure the image is in RGB format
+
+    # Create random UV coordinates for the mesh
+    uv = np.random.rand(len(mesh.vertices), 2)
+
+    # Create material and apply texture
     material = trimesh.visual.texture.SimpleMaterial(image=im)
-    
-    # Get the uv coordinates from the mesh
-    uv = scene.visual.uv
-    
-    # Create TextureVisuals with the uv coordinates and the texture image
     color_visuals = trimesh.visual.TextureVisuals(uv=uv, image=im, material=material)
-    
-    # Apply the visuals to the mesh
-    mesh.visual = color_visuals
-    
-    # Export the mesh to a .glb file
-    mesh.export(file_obj='modified_train.glb')
-    
-    # Convert the GLB file to base64 for Gradio Model3D output
-    with open('modified_train.glb', 'rb') as f:
-        glb_base64 = base64.b64encode(f.read()).decode()
-    
-    return glb_base64
-
-gr.Interface(
-    fn=modify_texture,
-    inputs=[],
-    outputs=gr.Model3D(label="Download Modified GLB File"),
-    title="GLB Texture Modifier"
-).launch()
+
+    # Apply the texture to the original mesh
+    textured_mesh = trimesh.Trimesh(vertices=mesh.vertices, faces=mesh.faces, visual=color_visuals, validate=True, process=False)
+
+    # Save the original and textured meshes to GLB format in memory
+    original_glb = io.BytesIO()
+    mesh.export(original_glb, file_type='glb')
+    original_glb.seek(0)
+
+    modified_glb = io.BytesIO()
+    textured_mesh.export(modified_glb, file_type='glb')
+    modified_glb.seek(0)
+
+    return original_glb, modified_glb
+
+with gr.Blocks() as app:
+    gr.Markdown("### 3D Model Texture Application")
+    original_model = gr.Model3D(label="Original Model")
+    modified_model = gr.Model3D(label="Textured Model")
+    button = gr.Button("Visualize Texture")
+    button.click(visualize_texture, outputs=[original_model, modified_model])
+
+app.launch()