Update visualizer_app.py

visualizer_app.py

@@ -2010,28 +2010,39 @@ def combined_chart_visualization(
 ):
     # Usage with highlight_closest=True
     if chart_dataframe is not None and query.value:
-        # Get the query embedding
-        query_emb = embedding.embed_documents([query.value])
-        set_query_state(query_emb)
-
-        # Get appropriate coordinates for the query
-        query_coords = get_query_coordinates(
-            reference_embeddings=chart_dataframe,
-            query_embedding=get_query_state()
-        )
-
-        # Add the query to the chart with closest points highlighted
-        result = add_query_to_embedding_chart(
-            existing_chart=emb_plot,
-            query_coords=query_coords,
-            query_text=query.value,
-        )
-
-        chart_with_query = result
+        with mo.status.spinner(title="Embedding Query...", remove_on_exit=True) as _spinner:
+                query_emb = embedding.embed_documents([query.value])
+                set_query_state(query_emb)
+        
+                _spinner.update("Preparing Query Coordinates") # --- --- ---
+                time.sleep(1.0)
+        
+                # Get appropriate coordinates for the query
+                query_coords = get_query_coordinates(
+                    reference_embeddings=chart_dataframe,
+                    query_embedding=get_query_state()
+                )
 
-        # Create the visualization
-        combined_viz = mo.ui.plotly(chart_with_query)
-        set_chart_state(combined_viz)
+                _spinner.update("Adding Query to Chart") # --- --- ---
+                time.sleep(1.0)
+        
+                # Add the query to the chart with closest points highlighted
+                result = add_query_to_embedding_chart(
+                    existing_chart=emb_plot,
+                    query_coords=query_coords,
+                    query_text=query.value,
+                )
+        
+                chart_with_query = result
+        
+                _spinner.update("Preparing Visualization") # --- --- ---
+                time.sleep(1.0)
+        
+                # Create the visualization
+                combined_viz = mo.ui.plotly(chart_with_query)
+                set_chart_state(combined_viz)
+        
+                _spinner.update("Done") # --- --- ---
     else:
         combined_viz = None
     return