Update src/streamlit_app.py

src/streamlit_app.py

@@ -37,4 +37,21 @@ st.altair_chart(alt.Chart(df, height=700, width=700)
         y=alt.Y("y", axis=None),
         color=alt.Color("idx", legend=None, scale=alt.Scale()),
         size=alt.Size("rand", legend=None, scale=alt.Scale(range=[1, 150])),
-    ))
+    ))
+
+from embedding_atlas.streamlit import embedding_atlas
+from embedding_atlas.projection import compute_text_projection
+
+df = pd.DataFrame({'projection_x': [1], 'projection_y': [1], 'description': ['x']})
+
+# Compute text embedding and projection of the embedding
+compute_text_projection(df, text="description",
+    x="projection_x", y="projection_y", neighbors="neighbors"
+)
+
+# Create an Embedding Atlas component for a given data frame
+value = embedding_atlas(
+    df, text="description",
+    x="projection_x", y="projection_y", neighbors="neighbors",
+    show_table=True
+)