Update main.py

main.py

@@ -49,7 +49,7 @@ async def VectorDatabase(categorie):
 async def Retriever(categorie):
     vectorstore = await VectorDatabase(categorie)
     if categorie == "videosTC":
-        retriever = vectorstore.as_retriever(search_type="similarity_score_threshold", search_kwargs={"score_threshold": .7, "k": 200,"filter": {"title": {"$eq": "videos-table-rondeia"}}})
+        retriever = vectorstore.as_retriever(search_type="similarity_score_threshold", search_kwargs={"score_threshold": .7, "k": 250,"filter": {"title": {"$eq": "videos-table-rondeia"}, "time": {"$gte": 1320}}})
     return retriever
 
 @cl.step(type="embedding")
@@ -63,17 +63,18 @@ async def Search(input, categorie):
     count = 0
     countOffres = 0
     if categorie == "videosTC":
-        search = vectorstore.similarity_search(input,k=50, filter={"title": {"$eq": "videos-confinement-timeline"}})
+        search = vectorstore.similarity_search(input,k=50, filter={"title": {"$eq": "videos-table-rondeia"}, "time": {"$gte": 1320}})
         for i in range(0,len(search)):
-            if count <= 17:
+            if count <= 23:
                 count = count + 1
                 timeSeq = search[i].metadata["time"]
                 timeSeqRound = round(timeSeq)
                 time = timedelta(seconds=timeSeqRound)
                 sources_text = sources_text + '<div class="gridvid"><a target="_blank" title="' + search[i].metadata['titre'] + ' : ...' + search[i].page_content + '" href="' + search[i].metadata['video'] + '#start=' + str(timeSeq) + '"><img src="' + search[i].metadata['image'] + '" width="100%" alt="' + search[i].metadata['titre'] + ' : ...' + search[i].page_content + '"/><p>🕓 ' + str(time) + ' : ...' + search[i].page_content + ' : ' + search[i].metadata['titre'] + '</p></a></div>'
+                sources_audio = sources_text + '<div class="gridvid"><a target="_blank" title="' + search[i].metadata['titre'] + ' : ...' + search[i].page_content + '" href="' + search[i].metadata['Audio'] + '#start=' + str(timeSeq) + '"><img src="' + search[i].metadata['thumb'] + '" width="100%" alt="' + search[i].metadata['titre'] + ' : ...' + search[i].page_content + '"/><p>🕓 ' + str(time) + ' : ...' + search[i].page_content + ' : ' + search[i].metadata['titre'] + '</p></a></div>'
                 verbatim_text = verbatim_text + "<p style='font-size:0.8rem'>" + str(count) + ". " + search[i].metadata['titre'] + "</p><p style='font-size:0.8rem'>🕓 "+ str(time) + " : " + search[i].page_content + "</p>"
             
-    results = [sources_text, verbatim_text, sources_offres]
+    results = [sources_text, verbatim_text, sources_audio]
     return results
         
 @cl.set_chat_profiles
@@ -170,18 +171,18 @@ async def on_message(message: cl.Message):
     search = await Search(message.content, "videosTC")
     
     sources = [
-        cl.Text(name="Sources", content=search[0], display="inline")
+        cl.Text(name="Sources vidéo", content=search[0], display="inline")
     ]
     await cl.Message(
-        content="Sources : ",
+        content="Vidéos : ",
         elements=sources,
     ).send()
     if search[2]:
         sourcesOffres = [
-            cl.Text(name="Exemples d'offres d'emploi", content=search[2], display="inline")
+            cl.Text(name="Sources audio", content=search[2], display="inline")
         ]
         await cl.Message(
-            content="Offres d'emploi : ",
+            content="Audio : ",
             elements=sourcesOffres,
         ).send()
     verbatim = [