Refactor generate3d function in inference.py to streamline mesh export process. Removed manual MTL file handling and updated export calls to enhance code clarity and maintainability.

inference.py

@@ -75,20 +75,6 @@ def generate3d(model, rgb, ccm, device):
     texture_path = base_path + ".png"
     mtl_path = base_path + ".mtl"
 
-    model.export_mesh_geometry(data_config, obj_path)  # writes .obj with UVs
-    model.export_texture_image(data_config, texture_path)  # saves PNG
-
-    # Write MTL file manually
-    with open(mtl_path, "w") as f:
-        f.write("newmtl material0\n")
-        f.write("Kd 1.000000 1.000000 1.000000\n")
-        f.write(f"map_Kd {os.path.basename(texture_path)}\n")
-
-    # Append .mtl reference to OBJ file
-    with open(obj_path, "r") as original:
-        lines = original.readlines()
-    with open(obj_path, "w") as modified:
-        modified.write(f"mtllib {os.path.basename(mtl_path)}\n")
-        modified.writelines(lines)
-
-    return obj_path
+    model.export_mesh(data_config, base_path, tri_fea_2=triplane_feature2)  # writes .obj
+
+    return obj_path