Update app.py

app.py

@@ -3,26 +3,24 @@ from pygltflib import GLTF2, Texture, Image
 import base64
 import io
 
-def modify_texture(new_texture):
+def modify_texture():
     # Load the GLB file
     glb = GLTF2().load("train.glb")
 
-    # Convert the new texture image to use in GLB
-    img_byte_arr = io.BytesIO()
-    new_texture.save(img_byte_arr, format='jpg')
-    encoded_img = base64.b64encode(img_byte_arr.getvalue()).decode('ascii')
+    # 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 new texture
-        glb.images[0].uri = f"data:image/png;base64,{encoded_img}"
+        # Replace the first image with the default texture
+        glb.images[0] = default_texture
     else:
-        # Add new image if none exists
-        new_image = Image(uri=f"data:image/png;base64,{encoded_img}")
-        glb.images.append(new_image)
-        # Update texture to point to the new image
-        new_texture = Texture(source=len(glb.images)-1)
-        glb.textures.append(new_texture)
+        # 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()
@@ -32,9 +30,7 @@ def modify_texture(new_texture):
 
 gr.Interface(
     fn=modify_texture,
-    inputs=[
-        gr.Image(label="Upload New Texture Image")
-    ],
+    inputs=[],
     outputs=gr.Model3D(label="Modified 3D Model"),
     title="GLB Texture Modifier"
 ).launch()