Added back run_model method

demo/src/gui.py

@@ -75,12 +75,12 @@ class WebUI:
 
     def process(self, mesh_file_name):
         path = mesh_file_name.name
-        #run_model(
-        #    path,
-        #    model_path=os.path.join(self.cwd, "resources/models/"),
-        #    task=self.class_names[self.class_name],
-        #    name=self.result_names[self.class_name],
-        #)
+        run_model(
+            path,
+            model_path=os.path.join(self.cwd, "resources/models/"),
+            task=self.class_names[self.class_name],
+            name=self.result_names[self.class_name],
+        )
         LOGGER.info("Converting prediction NIfTI to OBJ...")
         nifti_to_glb("prediction.nii.gz")
 

demo/src/utils.py

@@ -46,12 +46,19 @@ def nifti_to_glb(path, output="prediction.obj"):
     resampled = resample_to_output(image, [1, 1, 1], order=1)
     data = resampled.get_fdata().astype("uint8")
 
+    # Create a material with a red diffuse color (RGB value)
+    red_material = "newmtl RedMaterial\nKd 1 0 0"  # Red diffuse color (RGB)
+
     # extract surface
     verts, faces, normals, values = marching_cubes(data, 0)
     faces += 1
 
     with open(output, "w") as thefile:
+        # Write the material definition to the OBJ file
+        thefile.write(red_material + "\n")
+        
         for item in verts:
+            #thefile.write('usemtl RedMaterial\n')
             thefile.write("v {0} {1} {2}\n".format(item[0], item[1], item[2]))
 
         for item in normals: