use Open3D as headless

app.py

@@ -8,7 +8,6 @@ import os
 from types import SimpleNamespace
 
 DEBUG = False # True: skip gradio and execute in main function
-DEBUG_LOG = True
 
 def process_point_cloud(file_obj):
     # Save uploaded file temporarily
@@ -28,17 +27,17 @@ def process_point_cloud(file_obj):
     # Estimate normals for visualization
     downsampled.estimate_normals()
 
-    # Visualize to image
-    vis = o3d.visualization.Visualizer()
-    vis.create_window(visible=False)
-    vis.add_geometry(downsampled)
-    vis.poll_events()
-    vis.update_renderer()
-    img = vis.capture_screen_float_buffer(True)
-    vis.destroy_window()
+    points = np.asarray(downsampled.points)
+    fig = plt.figure(figsize=(6, 6))
+    ax = fig.add_subplot(111, projection='3d')
+    ax.scatter(points[:, 0], points[:, 1], points[:, 2], s=0.5, c='b')
+    ax.axis('off')
+    plt.tight_layout()
+    temp_img_path = os.path.join(tempfile.gettempdir(), "rendered.png")
+    plt.savefig(temp_img_path, dpi=150)
+    plt.close()
 
-    # Convert image to numpy and return
-    img_np = (255 * np.asarray(img)).astype(np.uint8)
+    img_np = plt.imread(temp_img_path)
     return img_np
 
 iface = gr.Interface(