Update app.py

app.py

@@ -45,7 +45,7 @@ def make_query(query, retriever, top_k=10, include_values=True, include_metadata
 st.session_state.index = init_pinecone()
 retriever = init_retriever()
 
-def card(thumbnail: str, title: str, urls: list, contexts: list, starts: list, ends: list):
+def card(thumbnail: str, title: str, urls: list, contexts: list, starts: list, ends: list, publication: str):
     meta = [(e, s, u, c) for e, s, u, c in zip(ends, starts, urls, contexts)]
     meta.sort(reverse=False)
     text_content = []
@@ -84,6 +84,9 @@ def card(thumbnail: str, title: str, urls: list, contexts: list, starts: list, e
             <div  class="col-md-8 col-sm-8">
                 <h2>{title}</h2>
             </div>
+            <div  class="col-md-8 col-sm-8">
+                <h2>{publication}</h2>
+            </div>
         <div>
             {html_text}
     <br><br>
@@ -100,7 +103,7 @@ publication_map = {
 }
     
 st.write("""
-# YouTube Q&A
+# Example
 """)
 
 st.info("""
@@ -141,15 +144,18 @@ if query != "":
         if video_id not in results:
             results[video_id] = {
                 'title': context['metadata']['title'],
-                'urls': [f"{context['metadata']['url']}?t={int(context['metadata']['start_second'])}"],
+                'thumbnail': context['metadata']['thumbnail'],
+                'urls': [f"{context['metadata']['url']}"],
                 'contexts': [context['metadata']['text']],
                 'starts': [int(context['metadata']['start_second'])],
-                'ends': [int(context['metadata']['end_second'])]
+                'ends': [int(context['metadata']['end_second'])],
+                'publication': context['metadata']['publication'],
+                'category': context['metadata']['category']
             }
             order.append(video_id)
         else:
             results[video_id]['urls'].append(
-                f"{context['metadata']['url']}?t={int(context['metadata']['start_second'])}"
+                f"{context['metadata']['url']}"
             )
             results[video_id]['contexts'].append(
                 context['metadata']['text']
@@ -159,10 +165,11 @@ if query != "":
     # now display cards
     for video_id in order:
         card(
-            thumbnail=f"https://img.youtube.com/vi/{video_id}/maxresdefault.jpg",
+            thumbnail=results[video_id]['thumbnail'],
             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']
+            ends=results[video_id]['ends'],
+            publication=results[video_id]['publication']
         )