Update view

views.py

@@ -42,6 +42,7 @@ def plot(df: pd.DataFrame, embeddings: np.ndarray, vectors_2d, reducer, correcto
     inferred_embedding = None
     # Add a card container to the right of the content with Streamlit columns
     col1, col2 = st.columns([0.6, 0.4])  # Adjusting ratio to allocate space for the card container
+    inversion_output_text = None
 
     with col1:
         # Main content stays here (scatterplot, form, etc.)
@@ -58,20 +59,17 @@ def plot(df: pd.DataFrame, embeddings: np.ndarray, vectors_2d, reducer, correcto
             vec = np.array([x, y]).astype("float32")
 
 
-            submit_button = st.form_submit_button("Submit")
+            submit_button = st.form_submit_button("Synthesize")
 
-            if selected_points or submit_button:
+            if submit_button:
                 inferred_embedding = reducer.inverse_transform(np.array([[x, y]]) if not isinstance(reducer, UMAP) else np.array([[x, y]]))
                 inferred_embedding = inferred_embedding.astype("float32")
 
-                output = vec2text.invert_embeddings(
+                inversion_output_text, = vec2text.invert_embeddings(
                     embeddings=torch.tensor(inferred_embedding).cuda(),
                     corrector=corrector,
                     num_steps=20,
                 )
-
-                st.text(str(output))
-                st.text(str(inferred_embedding))
             else:
                 st.text("Click on a point in the scatterplot to see its coordinates.")
 
@@ -84,6 +82,10 @@ def plot(df: pd.DataFrame, embeddings: np.ndarray, vectors_2d, reducer, correcto
             f"### Selected text:\n```console\n{selected_sentence}\n```"
         )
 
+        st.markdown(
+            f"### Synthesized text:\n```console\n{inversion_output_text}\n```"
+        )
+
         if inferred_embedding is not None and (closest_sentence_index != -1):
             couple = selected_sentence_embedding.squeeze(), inferred_embedding.squeeze()
             st.markdown(f"### Inferred embedding distance:")