Update app.py

app.py

@@ -263,6 +263,11 @@ def dendrogram_plot(embeddings, labels):
     ax.set_title("Hierarchical Dendrogram of Document Clusters", fontsize=14)
     ax.set_xlabel("Distance", fontsize=12)
     ax.set_ylabel("Documents", fontsize=12)
+    unique_labels = set(labels)
+    legend_labels = [
+        f"Cluster {label}" if label != -1 else "Noise" for label in unique_labels
+    ]
+    ax.legend(legend_labels, loc="upper right", title="Clusters", fontsize=10)
     plt.tight_layout()
     return fig
 
@@ -297,6 +302,10 @@ def tsne_plot(embeddings, labels):
     ax.set_title("t-SNE Visualization of Document Clusters", fontsize=14)
     ax.set_xlabel("t-SNE Dimension 1", fontsize=12)
     ax.set_ylabel("t-SNE Dimension 2", fontsize=12)
+    unique_labels = set(labels)
+    for label in unique_labels:
+        ax.scatter([], [], label=f"Cluster {label}" if label != -1 else "Noise", s=50)
+    ax.legend(loc="upper right", title="Clusters", fontsize=10)
     cbar = plt.colorbar(scatter, ax=ax)
     cbar.set_label("Cluster Labels", fontsize=12)
     return fig