Spaces:
Running
Running
bugfix for order
Browse files- src/app.py +7 -5
src/app.py
CHANGED
@@ -30,11 +30,11 @@ query = """WITH filtered_podcasts AS (
|
|
30 |
WHERE id in ?
|
31 |
),
|
32 |
ordered_embeddings AS (
|
33 |
-
SELECT embeddings.id, embeddings.part
|
34 |
FROM embeddings
|
35 |
JOIN filtered_podcasts fp ON embeddings.id = fp.id
|
36 |
-
ORDER BY
|
37 |
-
|
38 |
)
|
39 |
SELECT
|
40 |
p.title,
|
@@ -43,6 +43,7 @@ SELECT
|
|
43 |
e.text,
|
44 |
e.part,
|
45 |
p.audio,
|
|
|
46 |
FROM
|
47 |
ordered_embeddings oe
|
48 |
JOIN
|
@@ -52,7 +53,8 @@ SELECT
|
|
52 |
JOIN
|
53 |
podcasts p
|
54 |
ON
|
55 |
-
oe.id = p.id
|
|
|
56 |
"""
|
57 |
|
58 |
st.title("terapyon cannel search")
|
@@ -75,7 +77,7 @@ if word:
|
|
75 |
result = conn.execute(query,
|
76 |
(selected_ids, word_embedding,)).df()
|
77 |
selected = st.dataframe(result,
|
78 |
-
column_order=["title", "date", "part", "start", "text", "audio"],
|
79 |
on_select="rerun",
|
80 |
selection_mode="single-row")
|
81 |
if selected:
|
|
|
30 |
WHERE id in ?
|
31 |
),
|
32 |
ordered_embeddings AS (
|
33 |
+
SELECT embeddings.id, embeddings.part, array_distance(embedding, ?::FLOAT[1024]) AS distance
|
34 |
FROM embeddings
|
35 |
JOIN filtered_podcasts fp ON embeddings.id = fp.id
|
36 |
+
ORDER BY distance
|
37 |
+
LIMIT 10
|
38 |
)
|
39 |
SELECT
|
40 |
p.title,
|
|
|
43 |
e.text,
|
44 |
e.part,
|
45 |
p.audio,
|
46 |
+
oe.distance,
|
47 |
FROM
|
48 |
ordered_embeddings oe
|
49 |
JOIN
|
|
|
53 |
JOIN
|
54 |
podcasts p
|
55 |
ON
|
56 |
+
oe.id = p.id
|
57 |
+
ORDER BY oe.distance;
|
58 |
"""
|
59 |
|
60 |
st.title("terapyon cannel search")
|
|
|
77 |
result = conn.execute(query,
|
78 |
(selected_ids, word_embedding,)).df()
|
79 |
selected = st.dataframe(result,
|
80 |
+
column_order=["title", "date", "part", "start", "distance", "text", "audio"],
|
81 |
on_select="rerun",
|
82 |
selection_mode="single-row")
|
83 |
if selected:
|