Spaces:
Runtime error
Runtime error
| import streamlit as st | |
| import pinecone | |
| from sentence_transformers import SentenceTransformer | |
| PINECONE_KEY = st.secrets["PINECONE_KEY"] # app.pinecone.io | |
| def init_pinecone(): | |
| pinecone.init(api_key=PINECONE_KEY, environment="us-west1-gcp") | |
| return pinecone.Index('youtube-search') | |
| def init_retriever(): | |
| return SentenceTransformer("multi-qa-mpnet-base-dot-v1") | |
| index = init_pinecone() | |
| retriever = init_retriever() | |
| def card(thubmnail, title, url, context): | |
| return st.markdown(f""" | |
| <div class="container-fluid"> | |
| <div class="row align-items-start"> | |
| <div class="col-md-4 col-sm-4"> | |
| <div class="position-relative"> | |
| <a href={url}><img src={thubmnail} class="img-fluid" style="width: 192px; height: 106px"></a> | |
| </div> | |
| </div> | |
| <div class="col-md-8 col-sm-8"> | |
| <a href={url}>{title}</a> | |
| <br> | |
| <span style="color: #808080;"> | |
| <small>{context[:200].capitalize()+"...."}</small> | |
| </span> | |
| </div> | |
| </div> | |
| </div> | |
| """, unsafe_allow_html=True) | |
| st.write(""" | |
| # YouTube Q&A | |
| Ask me a question! | |
| """) | |
| st.markdown(""" | |
| <link rel="stylesheet" href="https://cdn.jsdelivr.net/npm/bootstrap@4.0.0/dist/css/bootstrap.min.css" integrity="sha384-Gn5384xqQ1aoWXA+058RXPxPg6fy4IWvTNh0E263XmFcJlSAwiGgFAW/dAiS6JXm" crossorigin="anonymous"> | |
| """, unsafe_allow_html=True) | |
| query = st.text_input("Search!", "") | |
| if query != "": | |
| xq = retriever.encode([query]).tolist() | |
| xc = index.query(xq, top_k=5, include_metadata=True) | |
| for context in xc['matches']: | |
| card( | |
| f"https://img.youtube.com/vi/{context['metadata']['url'].split('/')[-1]}/maxresdefault.jpg", | |
| context['metadata']['title'], | |
| f"{context['metadata']['url']}?t={int(context['metadata']['start'])}", | |
| context['metadata']['text'] | |
| ) |