Update app.py

app.py

@@ -40,7 +40,7 @@ def create_similarity_based_visualization(texts):
         for j in range(n):
             similarity_matrix[i][j] = wl.similarity(texts[i], texts[j])
     
-    # Convert similarities to distances (1 - similarity)
+    # Convert similarities to distances
     distance_matrix = 1 - similarity_matrix
     
     if n == 2:
@@ -50,12 +50,15 @@ def create_similarity_based_visualization(texts):
         # Place points based on similarity
         similarity = similarity_matrix[0][1]
         fig.add_trace(go.Scatter(
-            x=[0, 1-similarity],  # Distance proportional to similarity
+            x=[0, 1-similarity],
             y=[0, 0],
             mode='markers+text',
             text=texts,
             textposition='top center',
-            marker=dict(size=10, color=['blue', 'red'])
+            marker=dict(
+                size=10,
+                color=['blue', 'red']
+            )
         ))
         
         fig.update_layout(
@@ -73,18 +76,14 @@ def create_similarity_based_visualization(texts):
         mds = MDS(n_components=3, dissimilarity='precomputed', random_state=42)
         coords = mds.fit_transform(distance_matrix)
         
-        # Create DataFrame for plotting
-        df = pd.DataFrame(
-            coords,
-            columns=['X', 'Y', 'Z']
-        )
-        df['text'] = texts
-        
         # Create 3D scatter plot
-        fig = go.Figure(data=[go.Scatter3d(
-            x=df['X'],
-            y=df['Y'],
-            z=df['Z'],
+        fig = go.Figure()
+        
+        # Add points
+        fig.add_trace(go.Scatter3d(
+            x=coords[:, 0],
+            y=coords[:, 1],
+            z=coords[:, 2],
             mode='markers+text',
             text=texts,
             textposition='top center',
@@ -93,22 +92,28 @@ def create_similarity_based_visualization(texts):
                 color=list(range(len(texts))),
                 colorscale='Viridis',
                 opacity=0.8
-            )
-        )])
+            ),
+            name='Texts'
+        ))
         
-        # Add lines between points to show connections
+        # Add lines between points with valid opacity values
         for i in range(n):
             for j in range(i+1, n):
+                # Calculate opacity based on similarity (ensure it's between 0.1 and 1)
+                opacity = max(0.1, min(1.0, similarity_matrix[i,j]))
+                
                 fig.add_trace(go.Scatter3d(
                     x=[coords[i,0], coords[j,0]],
                     y=[coords[i,1], coords[j,1]],
                     z=[coords[i,2], coords[j,2]],
                     mode='lines',
                     line=dict(
-                        color=f'rgba(100,100,100,{similarity_matrix[i,j]:.2f})',
+                        color='gray',
                         width=2
                     ),
-                    showlegend=False
+                    opacity=opacity,
+                    showlegend=False,
+                    hoverinfo='skip'
                 ))
         
         fig.update_layout(
@@ -116,9 +121,21 @@ def create_similarity_based_visualization(texts):
             scene=dict(
                 xaxis_title="Dimension 1",
                 yaxis_title="Dimension 2",
-                zaxis_title="Dimension 3"
+                zaxis_title="Dimension 3",
+                camera=dict(
+                    up=dict(x=0, y=0, z=1),
+                    center=dict(x=0, y=0, z=0),
+                    eye=dict(x=1.5, y=1.5, z=1.5)
+                )
             ),
-            height=700
+            height=700,
+            showlegend=True,
+            legend=dict(
+                yanchor="top",
+                y=0.99,
+                xanchor="left",
+                x=0.01
+            )
         )
     
     return fig