Update app.py

app.py

@@ -17,14 +17,15 @@ def get_bert_embeddings(word):
 
 def plot_interactive_bert_embeddings():
     embeddings, sentences = get_all_embeddings()
-    if len(sentences) >= 3:
-        pca = PCA(n_components=3)
+    if len(sentences) > 0:
+        # Even if there's less than 3, PCA can still run with min(n_samples, n_features)
+        pca = PCA(n_components=min(3, len(sentences)))
         reduced_embeddings = pca.fit_transform(np.array(embeddings))
         fig = go.Figure(data=[
             go.Scatter3d(
                 x=[emb[0]],
-                y=[emb[1]],
-                z=[emb[2]],
+                y=[emb[1] if len(emb) > 1 else 0],  # Ensure there are enough dimensions
+                z=[emb[2] if len(emb) > 2 else 0],
                 mode='markers+text',
                 text=sent,
                 name=sent
@@ -42,7 +43,8 @@ def plot_interactive_bert_embeddings():
         ))
         st.plotly_chart(fig, use_container_width=True)
     else:
-        st.error("Please add at least three different words/phrases to visualize.")
+        st.error("No data available for visualization.")
+
 
 
 def main():