Update app.py

app.py

@@ -54,7 +54,19 @@ def create_point_cloud(inp):
 
     pc = sampler.output_to_point_clouds(samples)[0]  # Get the point cloud
     
-    # Convert the point cloud to a Plotly figure
+    # Check if auxiliary channels (e.g., RGB) are available
+    if 'R' in pc.channels and 'G' in pc.channels and 'B' in pc.channels:
+        # Combine R, G, B channels into a single color value
+        colors = (
+            pc.channels['R'] / 255.0,  # Normalize to [0, 1]
+            pc.channels['G'] / 255.0,
+            pc.channels['B'] / 255.0
+        )
+    else:
+        # Fall back to a single color if no RGB data is available
+        colors = 'blue'  # Choose a default color
+
+    # Create a Plotly 3D scatter plot
     fig = go.Figure(
         data=[
             go.Scatter3d(
@@ -64,8 +76,8 @@ def create_point_cloud(inp):
                 mode='markers',
                 marker=dict(
                     size=2,
-                    color=pc.channels[0],  # Assuming the first channel contains RGB values
-                    colorscale='Viridis',  # Adjust as needed
+                    color=colors,
+                    colorscale='Viridis' if isinstance(colors, tuple) else None,  # Use Viridis if RGB
                     opacity=0.8
                 )
             )