Update main.py

main.py

@@ -10,8 +10,9 @@ from langchain.callbacks.streaming_stdout import StreamingStdOutCallbackHandler
 from langchain.prompts import ChatPromptTemplate, MessagesPlaceholder
 from langchain.schema import StrOutputParser
 from langchain.chains.qa_with_sources import load_qa_with_sources_chain
-from langchain.vectorstores import Pinecone
-import pinecone
+from langchain_pinecone import PineconeVectorStore
+
+from pinecone import Pinecone
 from langchain.memory import ChatMessageHistory, ConversationBufferMemory
 import pandas as pd
 import numpy as np
@@ -106,16 +107,13 @@ def to_cache(file):
 
 @cl.cache
 def retriever_to_cache():
+    os.environ['PINECONE_API_KEY'] = os.environ['PINECONE_API_KEY']
+    os.environ['PINECONE_ENVIRONMENT'] = "us-west4-gcp-free"
     index_name = os.environ['PINECONE_INDEX_NAME']
     time.sleep(5)
     embeddings = HuggingFaceEmbeddings()
     time.sleep(5)
-    pinecone.init(
-        api_key=os.environ['PINECONE_API_KEY'],
-        environment="us-west4-gcp-free"
-    )
-    time.sleep(5)
-    vectorstore = Pinecone.from_existing_index(
+    vectorstore = PineconeVectorStore(
         index_name=index_name, embedding=embeddings
     )
     time.sleep(10)
@@ -126,10 +124,6 @@ def retriever_to_cache():
 async def chat_profile():
     return [
         cl.ChatProfile(name="OF - Offre de formation",markdown_description="Requêter sur l'offre de formation - OF",icon="./public/favicon.png",),
-        cl.ChatProfile(name="Emplois - En direct de Pole Emploi",markdown_description="Emplois - En direct de Pole Emploi",icon="./public/favicon.png",),
-        cl.ChatProfile(name="K1902 - LP MDAI",markdown_description="K1902 - LP MDAI : requête sur les offres d'emploi",icon="./public/favicon.png",),
-        cl.ChatProfile(name="M1802-I1401-M1810-M1801-M1805 - Licence Maths-Info",markdown_description="M1802-I1401-M1810-M1801-M1805 - Licence Maths-Info : requête sur les offres d'emploi",icon="./public/favicon.png",),
-        cl.ChatProfile(name="K1207-G1202-G1204 - Licence STAPS",markdown_description="K1207-G1202-G1204 - Licence STAPS : requête sur les offres d'emploi",icon="./public/favicon.png",),
     ]
 @cl.on_chat_start
 async def start():
@@ -137,35 +131,14 @@ async def start():
     chatProfile = chat_profile.split(' - ')
     
     if chatProfile[0] == 'OF':
-        connexion = cl.TaskList()
-        connexion.status = "Running..."
-
-        # Create a task and put it in the running state
-        task1 = cl.Task(title="Chargement des données, en attente...", status=cl.TaskStatus.RUNNING)
-        await connexion.add_task(task1)
-        await connexion.send()
-    
-    
-        logo = [
-        cl.Image(name="Logo", size="small", display="inline", path="./public/logo_light.png")
-        ]
-    
-        await cl.Message(author="🌐🌐🌐",content="", elements=logo).send()
+        app_user = cl.user_session.get("user")
+        welcomeUser = app_user.identifier
+        welcomeUserArray = welcomeUser.split('@')
+        welcomeUserStr = welcomeUserArray[0].replace('.',' ')
+        await cl.Message(f"> Bonjour {welcomeUserStr}").send()
         await cl.Message(
             author="🌐🌐🌐",content=f"Commencez à poser vos questions sur les données \"{chat_profile}\"\n\n💡Voici des exemples de question \n\t1️⃣ Basée sur les formations : Quelles sont toutes les formations licences générales?\n\t2️⃣ Basée sur les compétences : Quelles sont les compétences de la licence Economie et gestion?\n\t3️⃣ Basée sur les métiers : Quels sont les métiers possibles de la licence Economie et gestion?\n\t4️⃣ Basée sur un souhait : Quelles formations si je veux travailler dans la vente?\n\t5️⃣ Basée sur un savoir-être : Quelles formations si j'aime travailler en équipe?\n\t6️⃣ Basée sur un état : Quelles formations si je suis créatif?\n\t7️⃣ Question multi-critère : Quelles sont les activités, les compétences et les métiers possibles de la licence Economie et gestion?"
         ).send()
-        settings = await cl.ChatSettings(
-            [
-                TextInput(id="AgentName", label="Renseigner votre code ROME", initial=""),
-            ]
-        ).send()
-        value = settings["AgentName"]
-        task1.status = cl.TaskStatus.DONE
-        await cl.sleep(0.5)
-        await connexion.remove()
-    
-        if value:
-            await cl.Message(author="🌐🌐🌐",content=settings["AgentName"]).send()
     
         from langchain_core.prompts.prompt import PromptTemplate