Update app.py

app.py

@@ -48,19 +48,40 @@ sampler = PointCloudSampler(
 
 # Function to create point clouds
 def create_point_cloud(inp):
-    # Generate progressive samples
     samples = None
     for x in tqdm(sampler.sample_batch_progressive(batch_size=1, model_kwargs=dict(texts=[inp]))):
         samples = x
 
-    # Extract the point cloud
-    pc = sampler.output_to_point_clouds(samples)[0]
+    pc = sampler.output_to_point_clouds(samples)[0]  # Get the point cloud
+    
+    # Convert the point cloud to a Plotly figure
+    fig = go.Figure(
+        data=[
+            go.Scatter3d(
+                x=pc.coords[:, 0],  # X coordinates
+                y=pc.coords[:, 1],  # Y coordinates
+                z=pc.coords[:, 2],  # Z coordinates
+                mode='markers',
+                marker=dict(
+                    size=2,
+                    color=pc.channels[0],  # Assuming the first channel contains RGB values
+                    colorscale='Viridis',  # Adjust as needed
+                    opacity=0.8
+                )
+            )
+        ]
+    )
 
-    # Generate a Plotly figure for visualization
-    fig = plot_point_cloud(pc, grid_size=3, fixed_bounds=((-0.75, -0.75, -0.75), (0.75, 0.75, 0.75)))
-
-    # Convert Plotly figure to HTML for Gradio compatibility
-    return pc
+    fig.update_layout(
+        scene=dict(
+            xaxis_title="X",
+            yaxis_title="Y",
+            zaxis_title="Z",
+        ),
+        margin=dict(r=0, l=0, b=0, t=0)
+    )
+    
+    return fig
 
 # Create Gradio interface
 demo = gr.Interface(