Update pages/tc.py

pages/tc.py

@@ -1,5 +1,6 @@
 import streamlit as st
 import openai
+import os
 openai.api_key = os.getenv("OPENAI_API_KEY")
 
 from streamlit import session_state
@@ -79,6 +80,7 @@ if uploaded_file:
     agg_clustering = AgglomerativeClustering(n_clusters=None, distance_threshold=distance_threshold, linkage='ward')
     cluster_labels = agg_clustering.fit_predict(matrix)
     df['Cluster'] = cluster_labels
+    
     # Visualize clusters with t-SNE
     tsne = TSNE(n_components=2, perplexity=15, random_state=42, init="random", learning_rate=200)
     vis_dims2 = tsne.fit_transform(matrix)
@@ -111,11 +113,10 @@ if uploaded_file:
     # Display the plot in Streamlit
     st.pyplot(fig)
     st.text_area("Number of Cluster Labels", value=len(np.unique(cluster_labels.tolist())))
-    
 
 
     # Reading a review which belong to each group.
-    rev_per_cluster = 3
+    rev_per_cluster = 1
     n_clusters = len(np.unique(cluster_labels.tolist()))
 
     for i in range(n_clusters):