Update app.py

app.py

@@ -88,6 +88,13 @@ def card(thumbnail: str, title: str, urls: list, contexts: list, starts: list, e
     """
     return st.markdown(html, unsafe_allow_html=True)
 
+channel_map = {
+    'James Briggs': 'UCv83tO5cePwHMt1952IVVHw',
+    'Daniel Bourke': 'UCr8O8l5cCX85Oem1d18EezQ',
+    'Yannic Kilcher': 'UCZHmQk67mSJgfCCTn7xBfew',
+    'AI Coffee Break with Letitia': 'UCobqgqE4i5Kf7wrxRxhToQA',
+    'sentdex': 'UCfzlCWGWYyIQ0aLC5w48gBQ'
+}
     
 st.write("""
 # YouTube Q&A
@@ -104,9 +111,22 @@ st.markdown("""
 
 query = st.text_input("Search!", "")
 
+with st.expander("Advanced Options"):
+    channel_options = st.multiselect(
+        'Channels to Search',
+        ['James Briggs', 'Daniel Bourke', 'Yannic Kilcher', 'AI Coffee Break with Letitia', 'sentdex'],
+        ['James Briggs', 'Daniel Bourke', 'Yannic Kilcher', 'AI Coffee Break with Letitia', 'sentdex']
+    )
+
 if query != "":
+    channels = [channel_map[name] for name in channel_options]
     print(f"query: {query}")
-    matches = make_query(query, retriever, top_k=5)
+    matches = make_query(
+        query, retriever, top_k=5,
+        filter={
+            'channel_id': {'$in': channels}
+        }
+    )
     
     results = {}
     order = []