terapyon commited on
Commit
8584df7
·
1 Parent(s): 199eb26

bugfix for order

Browse files
Files changed (1) hide show
  1. 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 array_distance(embedding, ?::FLOAT[1024])
37
- LIMIT 10
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: