Update main.py

main.py

@@ -103,8 +103,10 @@ def Search(input, categorie):
     results = []
     test = []
     sources_text = ""
+    sources_offres = ""
     verbatim_text = ""
     count = 0
+    countOffres = 0
     if categorie == "bibliographie-OPP-DGDIN":
         search = vectorstore.similarity_search(input,k=50, filter={"categorie": {"$eq": categorie}})
         for i in range(0,len(search)):
@@ -124,13 +126,18 @@ def Search(input, categorie):
                 verbatim_text = verbatim_text + "<p style='font-size:0.8rem'>" + str(count) + ". JDLP : " + search[i].metadata['jdlp'] + "</p><p style='font-size:0.8rem'>" + search[i].page_content + "</p>"
     elif categorie == "skills":
         search = vectorstore.similarity_search(input,k=50, filter={"file": {"$eq": 'competences-master-CFA.csv'}})
+        searchOffres = vectorstore.similarity_search(input,k=50, filter={"file": {"$eq": 'marche-emploi-CFA.csv'}})
         for i in range(0,len(search)):
             if count <= 15:
                 count = count + 1
                 sources_text = sources_text + str(count) + ". " + search[i].metadata['diplôme'] + ' (année : ' + search[i].metadata['année'] + '), ' + search[i].metadata['domaine'] + ', https://www.francecompetences.fr/recherche/rncp/' + str(search[i].metadata['rncp'])[4:] + "/\n"
                 verbatim_text = verbatim_text + "<p style='font-size:0.8rem'>" + str(count) + ". " + search[i].metadata['diplôme'] + "</p><p style='font-size:0.8rem'>" + search[i].page_content + "</p>"
-                    
-    results = [sources_text, verbatim_text]
+        for i in range(0,len(searchOffres)):
+            if countOffres <= 15:
+                countOffres = countOffres + 1
+                sources_offres = sources_offres + str(countOffres) + ". " + searchOffres[i].metadata['Poste'] + ' (type de contrat : ' + search[i].metadata['Contrat'] + ')\n"
+            
+    results = [sources_text, verbatim_text, sources_offres]
     return results
     
 @cl.on_chat_start
@@ -143,8 +150,8 @@ async def on_chat_start():
             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?»"),
             cl.Action(name="year", value="Journée de La Pédagogie", label="🔥 Journée de La Pédagogie : exemple : «Quelles sont les bonnes pratiques des plateformes de e-learning?»"),
-            cl.Action(name="skills", value="Compétences du CFA Descartes", label="🔥 Les compétences des masters du CFA Descartes : exemple : «Quelles sont les compétences en marketing dispensées au CFA?»"),
-            cl.Action(name="OF", value="Formations Gustave Eiffel", label="🔥 Les formations de l'université Gustave Eiffel : exemple : «Quels sont les formations et les métiers possibles si on est créatifs?»"),
+            cl.Action(name="skills", value="Compétences du CFA Descartes", label="🔥 Les compétences des masters du CFA Descartes : exemple : «Quels sont les Master qui dispensent des compétences en marketing dispensées au CFA?»"),
+            cl.Action(name="OF", value="Formations Gustave Eiffel", label="🔥 Les formations de l'université Gustave Eiffel : exemple : «Quels sont les formations et les métiers possibles si on est créatif?»"),
             cl.Action(name="videos", value="Vidéos paroles de confiné.es", label="🔥 Les vidéos paroles de confiné.es : exemple : «Quelle est la méthodologie employée avec les plateformes d'enseignement à distance?»"),
             cl.Action(name="offreST", value="Offres d'emploi France Travail", label="🔥 Les offres d'emploi de France Travail : exemple : «Quels sont les types de contrat proposés par les recruteurs?»"),
             
@@ -229,10 +236,6 @@ async def on_message(message: cl.Message):
     
     #search = vectorstore.similarity_search(message.content,k=50, filter={"categorie": {"$eq": "bibliographie-OPP-DGDIN"}})
     search = Search(message.content, cl.user_session.get("selectRequest"))
-    
-    sources = [
-        cl.Text(name="Sources", content=search[0], display="inline")
-    ]
 
     #os.environ["GOOGLE_CSE_ID"] = os.getenv('GOOGLE_CSE_ID')
     #os.environ["GOOGLE_API_KEY"] = os.getenv('GOOGLE_API_KEY')
@@ -252,14 +255,24 @@ async def on_message(message: cl.Message):
     #else:
     #    print('None')
     
+    sources = [
+        cl.Text(name="Sources", content=search[0], display="inline")
+    ]
     await cl.Message(
         content="Sources : ",
         elements=sources,
     ).send()
+    if search[2]:
+        sourcesOffres = [
+            cl.Text(name="Exemples d'offres d'emploi", content=search[2], display="inline")
+        ]
+        await cl.Message(
+            content="Offres d'emploi : ",
+            elements=sourcesOffres,
+        ).send()
     verbatim = [
         cl.Text(name="Verbatim", content=search[1], display="side")
     ]
-
     await cl.Message(
         content="📚 Liste des Verbatim ",
         elements=verbatim,