Update main.py

main.py

@@ -72,14 +72,16 @@ def searchByRome(rome):
     if rome.find(',') != -1:
         romeArray = rome.split(',')
         for i in range(0,len(romeArray)):
-            all_docs = index.query(
-                top_k=1,
-                vector= [0] * 768, # embedding dimension
-                namespace='',
-                filter={"categorie": {"$eq": "rome"}, "rome":{"$eq": romeArray[i]}},
-                include_metadata=True
-            )
-            libelle = libelle + " " + all_docs['matches'][0]['metadata']['libelle_rome']
+            while i < 6:
+                codeRome = romeArray[i].strip()
+                all_docs = index.query(
+                    top_k=1,
+                    vector= [0] * 768, # embedding dimension
+                    namespace='',
+                    filter={"categorie": {"$eq": "rome"}, "rome":{"$eq": codeRome}},
+                    include_metadata=True
+                )
+                libelle = libelle + " " + all_docs['matches'][0]['metadata']['libelle_rome']
     else:
         all_docs = index.query(
             top_k=1,
@@ -171,7 +173,7 @@ async def start():
         ]
     
         await cl.Message(author="🌐🌐🌐",content="", elements=logo).send()
-        valuesRome = np.array(["", "M1302 - DU entrepreneuriat", "F1106,F1201,F1103,H1206 - Licence Génie Urbain", "M1502,M1503 - LP Gestion Opérationnelles des Ressources Humaines", "K1902 - LP Management et Droits des Affaires Immobilières"])
+        valuesRome = np.array(["", "M1302 - DU entrepreneuriat", "F1106,F1201,F1103,H1206,K1802 - Licence Génie Urbain", "M1502,M1503 - LP Gestion Opérationnelles des Ressources Humaines", "K1902 - LP Management et Droits des Affaires Immobilières"])
         cl.user_session.set("arraySettingsRome", valuesRome)
         settings = await cl.ChatSettings(
             [
@@ -181,6 +183,7 @@ async def start():
                     values=valuesRome,
                     initial_index=0,
                 ),
+                TextInput(id="romeInput", label="ou saisir une liste de codes ROME jusqu'à 5 codes", placeholder="ou saisir une liste de codes ROME jusqu'à 5 codes", tooltip="saisir une liste de codes Rome séparés par des virgules, jusqu'à 5 codes ROME"),
                 Select(
                     id="type",
                     label="Type de fiche",
@@ -253,7 +256,11 @@ async def setup_agent(settings):
         anthropic = Anthropic(
             api_key=os.environ['ANTHROPIC_API_KEY'],
         )
-        romeList = settings['rome']
+        if settings['romeInput']:
+            romeList = settings['romeInput'] + " - "
+            cl.user_session.set("romeFree", romeList)
+        else:
+            romeList = settings['rome']
         romeListArray = romeList.split(' - ')
                 
         #session = cl.user_session.get("id")
@@ -369,9 +376,10 @@ async def setup_agent(settings):
                         Select(
                         id="rome",
                         label="Codes Rome",
-                        values=["", "M1302 - DU entrepreneuriat", "F1106,F1201,F1103,H1206 - Licence Génie Urbain", "M1502,M1503 - LP Gestion Opérationnelles des Ressources Humaines", "K1902 - LP Management et Droits des Affaires Immobilières"],
+                        values=["", "M1302 - DU entrepreneuriat", "F1106,F1201,F1103,H1206,K1802 - Licence Génie Urbain", "M1502,M1503 - LP Gestion Opérationnelles des Ressources Humaines", "K1902 - LP Management et Droits des Affaires Immobilières"],
                         initial_index=indexOfRome,
                         ),
+                        TextInput(id="romeInput", label="ou saisir une liste de codes ROME jusqu'à 5 codes", initial=cl.user_session.get('romeFree'), placeholder="ou saisir une liste de codes ROME jusqu'à 5 codes", tooltip="saisir une liste de codes Rome séparés par des virgules, jusqu'à 5 codes ROME"),
                         Select(
                         id="type",
                         label="Type de fiche",
@@ -632,6 +640,10 @@ async def setup_agent(settings):
                 await cl.Message(
                     author="🌐🌐🌐",content="⛔ Vous n'avez pas encore créé de Fiche synoptique!"
                 ).send()
+        else:
+            await cl.Message(
+                author="🌐🌐🌐",content=f"👍 Changement de contexte : {settings['rome']}\n⚠️ Pas de type de fiche : {settings['type']}\n⛔ Vous ne pouvez pas élaborer de fiche!"
+            ).send()
     os.environ['ANTHROPIC_API_KEY'] = os.environ['ANTHROPIC_API_KEY']
     contextChat = cl.user_session.get("contextChatBot")
     if not contextChat: