Update main.py

main.py

@@ -72,8 +72,8 @@ def searchByRome(rome):
     if rome.find(',') != -1:
         romeArray = rome.split(',')
         for i in range(0,len(romeArray)):
-            while i < 6:
-                codeRome = romeArray[i].strip()
+            codeRome = romeArray[i].strip()
+            if i <= 5 and len(codeRome) == 5:
                 all_docs = index.query(
                     top_k=1,
                     vector= [0] * 768, # embedding dimension
@@ -256,7 +256,6 @@ async def setup_agent(settings):
         anthropic = Anthropic(
             api_key=os.environ['ANTHROPIC_API_KEY'],
         )
-        print("ROME :" + settings['romeInput'] + "ROME")
         if settings['romeInput']:
             romeList = settings['romeInput'] + " - "
             cl.user_session.set("romeFree", romeList)