fix: avoid while loop

app.py

@@ -1,5 +1,3 @@
-import time
-
 import duckdb
 import gradio as gr
 import polars as pl
@@ -63,23 +61,21 @@ def vectorize_dataset(split: str, column: str):
 
 
 def run_query(query: str, column: str):
-    global df
-    while f"{column}_embeddings" not in df.columns:
-        sleeper = 5
-        gr.Info(
-            f"Waiting for vectorization to complete... ({sleeper}s)", duration=sleeper
-        )
-        time.sleep(sleeper)
-    vector = model.encode(query)
-    df_results = duckdb.sql(
-        query=f"""
-        SELECT *
-        FROM df
-        ORDER BY array_cosine_distance({column}_embeddings, {vector.tolist()}::FLOAT[256])
-        LIMIT 5
-        """
-    ).to_df()
-    return gr.Dataframe(df_results, visible=True)
+    try:
+        global df
+
+        vector = model.encode(query)
+        df_results = duckdb.sql(
+            query=f"""
+            SELECT *
+            FROM df
+            ORDER BY array_cosine_distance({column}_embeddings, {vector.tolist()}::FLOAT[256])
+            LIMIT 5
+            """
+        ).to_df()
+        return gr.Dataframe(df_results, visible=True)
+    except Exception as e:
+        raise gr.Error(f"Error running query: {e}")
 
 
 def hide_components():