Update main.py

main.py

@@ -69,7 +69,7 @@ def Retriever(categorie):
     if categorie != "year":
         retriever = vectorstore.as_retriever(search_type="similarity_score_threshold", search_kwargs={"score_threshold": .7, "k": 150,"filter": {'categorie': {'$eq': categorie}}})
     else:
-        retriever = vectorstore.as_retriever(search_type="similarity_score_threshold", search_kwargs={"score_threshold": .7, "k": 150,"filter": {'year': {'$gte': 2019}}})
+        retriever = vectorstore.as_retriever(search_type="similarity_score_threshold", search_kwargs={"score_threshold": .7, "k": 25,"filter": {'year': {'$gte': 2019}}})
     #search = vectorstore.similarity_search(query,k=50, filter={"categorie": {"$eq": "bibliographie-OPP-DGDIN"}, 'Source': {'$eq': 'Source : Persée'}})    
     return retriever
 
@@ -95,7 +95,7 @@ def Search(input, categorie):
         for i in range(0,len(search)):
             if count <= 15:
                 count = count + 1
-                sources_text = sources_text + str(count) + ". " + search[i].metadata['title'] + ' (JDLP : ' + search[i].metadata['year'] + '), ' + search[i].metadata['author'] + ', ' + search[i].metadata['file'] + "\n"
+                sources_text = sources_text + str(count) + ". " + search[i].metadata['title'] + ' (JDLP : ' + str(search[i].metadata['year']) + '), ' + search[i].metadata['author'] + ', ' + search[i].metadata['file'] + "\n"
                 verbatim_text = verbatim_text + "<p>" + str(count) + ". JDLP : " + search[i].metadata['jdlp'] + "</p><p>" + search[i].metadata['text'] + "</p>"
                     
     results = [sources_text, verbatim_text]
@@ -106,7 +106,7 @@ async def on_chat_start():
     await cl.Message(f"> REVIEWSTREAM").send()
     #await cl.Message(f"Nous avons le plaisir de vous accueillir dans l'application de recherche et d'analyse des publications.").send()
     res = await cl.AskActionMessage(
-        content="<div style='width:100%;text-align:center'>Sélectionnez source documentaire</div>",
+        content="<div style='width:100%;text-align:center'>Sélectionnez une source documentaire</div>",
         actions=[
             cl.Action(name="bibliographie-OPP-DGDIN", value="Pédagogie durable", label="🔥 Pédagogie durable : exemple : «quels sont les modèles d'apprentissage dans les universités?»"),
             cl.Action(name="bibliographie-OPP-DGDIN", value="Lieux d'apprentissage", label="🔥 Lieux d'apprentissage : exemple : «donne des exemples de lieu d'apprentissage dans les universités?»"),