updated to save temp mesh file

app.py

@@ -201,51 +201,12 @@ def proc_all(mesh_path, mode_path):
       alldicts.append(single)
   return alldicts
 
-def create_scene(mesh_path, dicts, plane_idx):
-    mesh = trimesh.load(mesh_path)
-    verts = np.array(mesh.vertices)
-    colors = []
-    for i in range(len(verts)):
-        if i in dicts[plane_idx]['left']:
-            color = [8, 136, 255]
-        elif i in dicts[plane_idx]['right']:
-            color = [204, 20, 245]
-        else:
-            color = [225, 225, 225]
-        colors.append(color)
-
-    mesh.visual.vertex_colors = colors
-    newplane = dicts[plane_idx]['plane']
-    newplane.visual.vertex_colors = np.tile([100, 100, 100, 100], (4, 1))
-    scene = trimesh.Scene([mesh, newplane])
+def create_scene(allscenes, plane_idx):
+    scene = allscenes[plane_idx] 
     temp_file = f"/tmp/scene_{plane_idx}.obj"
     scene.export(temp_file)
     return temp_file
 
-# def load_mesh_max(mesh_file_name):
-#     plane_path = "planes_processed/" + mesh_file_name.split('/')[-1].split('.')[0][:-7] + "_planes.pickle"
-#     with open(plane_path, 'rb') as f:
-#         dicts = pickle.load(f)
-#     #dicts = proc_all(mesh_file_name)  # Assuming proc_all is defined and returns a list of dictionaries
-#     allmesh = []
-#     for i in range(min(5, len(dicts))):  # Limiting to maximum 5 outputs
-#         temp_file = create_scene(mesh_file_name, dicts, -(i+1))
-#         allmesh.append(temp_file)
-#     return allmesh + [None] * (5 - len(allmesh))  # Fill the rest with None if less than 5
-
-# def load_mesh_min(mesh_file_name):
-#     plane_path = "planes_processed/" + mesh_file_name.split('/')[-1].split('.')[0][:-7] + "_planes.pickle"
-#     with open(plane_path, 'rb') as f:
-#         dicts = pickle.load(f)
-#     #dicts = proc_all(mesh_file_name)  # Assuming proc_all is defined and returns a list of dictionaries
-#     allmesh = []
-#     for i in range(min(5, len(dicts))):  # Limiting to maximum 5 outputs
-#         temp_file = create_scene(mesh_file_name, dicts, i)
-#         allmesh.append(temp_file)
-#     return allmesh + [None] * (5 - len(allmesh))  # Fill the rest with None if less than 5
-
-
-
 def load_mesh_max(scene_path):
     dict_path = "planes_processed/" + scene_path.split('/')[-1].split('.')[0][:-7] + "_dicts.pickle"
     with open(dict_path, 'rb') as f:
@@ -254,8 +215,13 @@ def load_mesh_max(scene_path):
         scenes = pickle.load(f)
     alllengths = [dic['len'] for dic in dicts]
     sort_idcs = np.argsort(alllengths)
-    sorted_planes = np.array(scenes)[np.flip(sort_idcs)]
-    return sorted_planes[:5]
+    sorted_planes = np.array(scenes)[sort_idcs]
+    allmesh = []
+    for i in range(min(5, len(dicts))):  # Limiting to maximum 5 outputs
+        temp_file = create_scene(sorted_planes, -(i+1))
+        allmesh.append(temp_file)
+    return allmesh + [None] * (5 - len(allmesh))  # Fill the rest with None if less than 5
+
 
 def load_mesh_min(scene_path):
     dict_path = "planes_processed/" + scene_path.split('/')[-1].split('.')[0][:-7] + "_dicts.pickle"
@@ -266,13 +232,19 @@ def load_mesh_min(scene_path):
     alllengths = [dic['len'] for dic in dicts]
     sort_idcs = np.argsort(alllengths)
     sorted_planes = np.array(scenes)[sort_idcs]
-    return sorted_planes[:5]
+    allmesh = []
+    for i in range(min(5, len(dicts))):  # Limiting to maximum 5 outputs
+        temp_file = create_scene(sorted_planes, i)
+        allmesh.append(temp_file)
+    return allmesh + [None] * (5 - len(allmesh))  # Fill the rest with None if less than 5
+
+
 
 
 def reset_outputs():
     # Returns a list of None, one for each 3D model output to reset them
     return [None] * 10  # Adjust to the total number of Model3D components you have
-
+    
 examples = glob.glob("planes_processed/*_planes.pickle")
 
 with gr.Blocks() as demo:
@@ -303,4 +275,4 @@ with gr.Blocks() as demo:
 
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(debug=True)