Create app.py

app.py

@@ -0,0 +1,36 @@
+import gradio as gr
+from pygltflib import GLTF2, Texture, Image
+import base64
+import io
+
+def modify_texture():
+    # Load the GLB file
+    glb = GLTF2().load("train.glb")
+
+    # Load the default texture image
+    with open("defect.jpg", "rb") as f:
+        default_texture = Image(uri=f"data:image/jpeg;base64,{base64.b64encode(f.read()).decode()}")
+
+    # Assuming there's at least one image in the original GLB
+    if glb.images:
+        # Replace the first image with the default texture
+        glb.images[0] = default_texture
+    else:
+        # Add default image if none exists
+        glb.images.append(default_texture)
+        # Update texture to point to the default image
+        default_texture = Texture(source=len(glb.images)-1)
+        glb.textures.append(default_texture)
+
+    # Save the modified GLB to a temporary file and return it
+    output_file = io.BytesIO()
+    glb.save(output_file)
+    output_file.seek(0)
+    return output_file
+
+gr.Interface(
+    fn=modify_texture,
+    inputs=[],
+    outputs=gr.Model3D(label="Modified 3D Model"),
+    title="GLB Texture Modifier"
+).launch()