Update app.py

app.py

@@ -330,15 +330,15 @@ def visualize_results(results_df, stats_df):
     axs[1, 0].set_title('Distribution of Result Content Lengths')
     axs[1, 0].set_xticklabels(axs[1, 0].get_xticklabels(), rotation=45, ha='right')
     
-    #embeddings = np.array([embedding for embedding in results_df['embedding'] if isinstance(embedding, np.ndarray)])
-    #if len(embeddings) > 1:
-    #    tsne = TSNE(n_components=2, random_state=42)
-    #    embeddings_2d = tsne.fit_transform(embeddings)
-    #    
-    #    sns.scatterplot(x=embeddings_2d[:, 0], y=embeddings_2d[:, 1], hue=results_df['model'][:len(embeddings)], ax=axs[1, 1])
-    #    axs[1, 1].set_title('t-SNE Visualization of Result Embeddings')
-    #else:
-    #    axs[1, 1].text(0.5, 0.5, "Not enough data for t-SNE visualization", ha='center', va='center')
+    embeddings = np.array([embedding for embedding in results_df['embedding'] if isinstance(embedding, np.ndarray)])
+    if len(embeddings) > 1:
+        tsne = TSNE(n_components=2, random_state=42)
+        embeddings_2d = tsne.fit_transform(embeddings)
+        
+        sns.scatterplot(x=embeddings_2d[:, 0], y=embeddings_2d[:, 1], hue=results_df['model'][:len(embeddings)], ax=axs[1, 1])
+        axs[1, 1].set_title('t-SNE Visualization of Result Embeddings')
+    else:
+        axs[1, 1].text(0.5, 0.5, "Not enough data for t-SNE visualization", ha='center', va='center')
     
     plt.tight_layout()
     return fig