Update app.py

app.py

@@ -88,4 +88,49 @@ def card(thumbnail: str, title: str, urls: list, contexts: list, starts: list, e
             {html_text}
     <br><br>
     """
-    return st.markdown(html, unsafe_allow_html=True)
+    return st.markdown(html, unsafe_allow_html=True)
+
+query = st.text_input("Search!", "")
+
+if query != "":
+    channels = [channel_map[name] for name in channel_options]
+    print(f"query: {query}")
+    matches = make_query(
+        query, retriever, top_k=5,
+        filter={
+            'channel_id': {'$in': channels}
+        }
+    )
+    
+    results = {}
+    order = []
+    for context in matches:
+        video_id = context['metadata']['url'].split('/')[-1]
+        if video_id not in results:
+            results[video_id] = {
+                'title': context['metadata']['title'],
+                'urls': [f"{context['metadata']['url']}?t={int(context['metadata']['start'])}"],
+                'contexts': [context['metadata']['text']],
+                'starts': [int(context['metadata']['start'])],
+                'ends': [int(context['metadata']['end'])]
+            }
+            order.append(video_id)
+        else:
+            results[video_id]['urls'].append(
+                f"{context['metadata']['url']}?t={int(context['metadata']['start'])}"
+            )
+            results[video_id]['contexts'].append(
+                context['metadata']['text']
+            )
+            results[video_id]['starts'].append(int(context['metadata']['start']))
+            results[video_id]['ends'].append(int(context['metadata']['end']))
+    # now display cards
+    for video_id in order:
+        card(
+            thumbnail=f"https://img.youtube.com/vi/{video_id}/maxresdefault.jpg",
+            title=results[video_id]['title'],
+            urls=results[video_id]['urls'],
+            contexts=results[video_id]['contexts'],
+            starts=results[video_id]['starts'],
+            ends=results[video_id]['ends']
+        )