Added YT logic

app.py

@@ -213,11 +213,23 @@ if QUESTION:
             d['url']= r['metadata']['url']
             
         meta_score= get_meta_score(d['url'],question_embedding)
+        
         score=.5* r['score'] + .5*meta_score
         d['score']=score
         #st.write(d['url'], score, r['score'], meta_score)
         resplist.append(d)
 
+    #check youtube
+    respmeta= query_from_pinecone(index, 'meta', question_embedding.tolist(), 5)
+    for r in respmeta:
+        if 'youtube' in r['id'] and r['score']>=.8:
+            d['id']=r['id']
+            d['content']=r['metadata']['data']
+            d['title']=r['metadata']['title']
+            d['url']= r['metadata']['url']   
+            d['score']=r['score']
+            resplist.append(d)
+
     if len(resplist)>0:
         sorted_indices = sorted(range(len(resplist)), key=lambda i: resplist[i]['score'], reverse=True)