model choice conseil IA

partie_prenante_carte.py

@@ -22,6 +22,7 @@ from session import set_partie_prenante
 import os
 from streamlit_vertical_slider import vertical_slider
 from high_chart import test_chart
+from chat_with_pps import get_response
 
 load_dotenv()
 
@@ -84,6 +85,7 @@ def get_vectorstore_from_text(texts):
 
 def get_conversation_chain(vectorstore):
     llm = ChatOpenAI(model="gpt-4o",temperature=0.5, max_tokens=2048)
+    
     retriever=vectorstore.as_retriever()
 
     prompt = hub.pull("rlm/rag-prompt")
@@ -168,19 +170,8 @@ def extract_pp(docs,input_variables):
 
     Le nom de la marque de référence est le suivant : {BRAND_NAME}
 
-    TA RÉPONSE DOIT ÊTRE SOUS FORME DE LISTE DE NOMS DE MARQUES, CHAQUE NOM SUR UNE LIGNE SÉPARÉE, ET NE DOIT PAS INCLURE D'OBJETS OU DE NOMS COMMUNS.
-
-    Exemples de bonnes réponses :
-    - Nike
-    - Adidas
-    - Puma
+    TA RÉPONSE DOIT ÊTRE SOUS FORME DE LISTE DE NOMS DE MARQUES, CHAQUE NOM SUR UNE LIGNE SÉPARÉE.
 
-    Exemples de mauvaises réponses (à éviter) :
-    - Chaussures
-    - Ballon
-    - Vêtements
-
-    LISTE DE MARQUES :
     """
     #don't forget to add the input variables from the maim function
 
@@ -290,34 +281,16 @@ def complete_and_verify_url(partial_url):
     
 @st.experimental_dialog("Conseil IA",width="large")
 def show_conseil_ia():
-    st.markdown("**Prenant compte les données de l'entreprise (activité, produits, services ...), quelles sont les principales parties prenantes à animer pour une démarche RSE réussie ?**")
-    markdown_text = f"""
-        Compte tenu des informations fournies sur l'entreprise {st.session_state["Nom de la marque"]}, une boutique en ligne de nutrition sportive, voici les principales parties prenantes à animer pour une démarche RSE réussie :
-
-        1. **Les clients et consommateurs** :
-        - Comprendre leurs attentes en matière de nutrition responsable et durable.
-        - Communiquer de manière transparente sur les engagements RSE de l'entreprise (origine des ingrédients, impact environnemental, etc.).
-        - Impliquer les clients dans des initiatives RSE (collectes de dons, campagnes de sensibilisation, etc.).
-
-        2. **Les fournisseurs** :
-        - Évaluer et sélectionner les fournisseurs selon des critères RSE (pratiques de production, conditions de travail, etc.).
-        - Accompagner les fournisseurs dans l'amélioration de leurs propres démarches RSE.
-        - Co-construire des solutions innovantes et durables pour la filière.
-
-        3. **Les employés** :
-        - Sensibiliser et former les employés aux enjeux RSE liés aux produits et services.
-        - Encourager leurs initiatives et leur engagement personnel dans la démarche RSE.
-        - Valoriser les bonnes pratiques et le rôle des employés dans la réussite de la démarche.
-        """
-    st.markdown(markdown_text)
-    if st.button("Saisissez une autre URL"):
-        st.rerun()
-
+    prompt = "Prenant compte les données de l'entreprise (activité, produits, services ...), quelles sont les principales parties prenantes à animer pour une démarche RSE réussie ?"
+    st.markdown(f"**{prompt}**")
+    response = st.write_stream(get_response(prompt, "",st.session_state["latest_doc"][0].page_content))
+    st.warning("Quittez et saisissez une autre URL")
 
 def display_pp():
 
     load_dotenv()
     fire_crawl_api_key = os.getenv("FIRECRAWL_API_KEY")
+    st.write(fire_crawl_api_key)
     #check if brand name and description are already set
     if "Nom de la marque" not in st.session_state:
         st.session_state["Nom de la marque"] = ""