Update app.py

app.py

@@ -1,19 +1,46 @@
 import streamlit as st
 import pinecone
 from sentence_transformers import SentenceTransformer
+import logging
 
 PINECONE_KEY = st.secrets["PINECONE_KEY"]  # app.pinecone.io
+INDEX_ID = 'youtube-search'
 
 @st.experimental_singleton
 def init_pinecone():
     pinecone.init(api_key=PINECONE_KEY, environment="us-west1-gcp")
-    return pinecone.Index('youtube-search')
+    return pinecone.Index(INDEX_ID)
     
 @st.experimental_singleton
 def init_retriever():
     return SentenceTransformer("multi-qa-mpnet-base-dot-v1")
 
-index = init_pinecone()
+def make_query(query, retriever, top_k=10, include_values=True, include_metadata=True, filter=None):
+    xq = retriever.encode([query]).tolist()
+    logging.info(f"Query: {query}")
+    attempt = 0
+    while attempt < 3:
+        try:
+            xc = st.session_state.index.query(
+                xq,
+                top_k=top_k,
+                include_values=include_values,
+                include_metadata=include_metadata,
+                filter=filter
+            )
+            matches = xc['matches']
+            break
+        except:
+            # force reload
+            pinecone.init(api_key=PINECONE_KEY, environment="us-west1-gcp")
+            st.session_state.index = pinecone.Index(INDEX_ID)
+            attempt += 1
+            matches = []
+    if len(matches) == 0:
+        logging.error(f"Query failed")
+    return matches
+
+st.session_state.index = init_pinecone()
 retriever = init_retriever()
 
 def card(thumbnail: str, title: str, urls: list, contexts: list, starts: list, ends: list):
@@ -64,7 +91,10 @@ def card(thumbnail: str, title: str, urls: list, contexts: list, starts: list, e
     
 st.write("""
 # YouTube Q&A
-Ask me a question!
+
+:::info
+YouTube search built as [explained here](https://pinecone.io/learn/openai-whisper)!
+The current version has a limited video dataset, but we are working on expanding it.
 """)
 
 st.markdown("""
@@ -75,12 +105,11 @@ query = st.text_input("Search!", "")
 
 if query != "":
     print(f"query: {query}")
-    xq = retriever.encode([query]).tolist()
-    xc = index.query(xq, top_k=5, include_metadata=True)
+    matches = make_query(query, retriever, top_k=5)
     
     results = {}
     order = []
-    for context in xc['matches']:
+    for context in matches:
         video_id = context['metadata']['url'].split('/')[-1]
         if video_id not in results:
             results[video_id] = {