model choice conseil IA

.streamlit/.env

@@ -1,3 +1,4 @@
 API_TOKEN_PERPLEXITYAI = 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
 OPENAI_API_KEY = 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
 FIRECRAWL_API_KEY = fc-381ecdb1175147aab5d2b48023961491
+MISTRAL_API_KEY = Hcm1IhfT38thMSaBIMt0t6L1SmrbgbXj

app.py

@@ -20,7 +20,7 @@ from collaborons import display_company_selection_for_materiality,display_materi
 from documentations import display_documentation
 from RAG_PDF_WEB import rag_pdf_web
 from prompt import get_prompts_list,prompt_execution,execute_prompt
-from chat_with_pps import display_chat,display_container_chat
+from chat_with_pps import display_chat
 from high_chart import test_chart
 from chat_te import display_chat_te
 

chat_te.py

@@ -62,6 +62,7 @@ def get_response(chain,user_query, chat_history):
 
     return chain.stream(question)
 
+
 @st.experimental_dialog("Cast your vote")
 def vote(item):
     st.write(f"Why is {item} your favorite?")

chat_with_pps.py

@@ -4,7 +4,9 @@ from langchain_community.chat_models import ChatOpenAI
 from dotenv import load_dotenv
 from langchain_core.output_parsers import StrOutputParser
 from langchain_core.prompts import ChatPromptTemplate
+from langchain_mistralai.chat_models import ChatMistralAI
 from download_chart import construct_plot
+from prompt import get_prompts_list
 
 
 load_dotenv()
@@ -21,16 +23,16 @@ def format_context(partie_prenante_grouped,marque):
     - Gérer étroitement: le pouvoir est entre 50 et 100 et l'influence est entre 50 et 100
     - Suivre de près: le pouvoir est entre 0 et 50 et l'influence est entre 0 et 50
     - Tenir informé: le pouvoir est entre 0 et 50 et l'influence est entre 50 et 100
+    '''
 
-'''
     context += segmentation
     return context
 
 
-def get_response(user_query, chat_history, context):
+def get_response(user_query, chat_history, context,llm=None):
 
     template = """
-    Fournir des réponses précises et contextuelles en agissant comme un expert en affaires, en utilisant le contexte des parties prenantes et leur pouvoir en pourcentage et leur influence en pourcentage pour expliquer les implications pour la marque. Le modèle doit connecter les informations du contexte et de l'historique de la conversation pour donner une réponse éclairée à la dernière question posée.
+    Fournir des réponses, en francais, précises et contextuelles en agissant comme un expert en affaires, en utilisant le contexte des parties prenantes et leur pouvoir en pourcentage et leur influence en pourcentage pour expliquer les implications pour la marque. Le modèle doit connecter les informations du contexte et de l'historique de la conversation pour donner une réponse éclairée à la dernière question posée.
 
     Contexte: {context}
 
@@ -41,7 +43,13 @@ def get_response(user_query, chat_history, context):
 
     prompt = ChatPromptTemplate.from_template(template)
 
-    llm = ChatOpenAI(model="gpt-4o")
+    #llm = ChatOpenAI(model="gpt-4o")
+    if not llm:
+        llm = ChatOpenAI(model="gpt-4o")
+    elif llm == "GPT-4o":
+        llm = ChatOpenAI(model="gpt-4o")
+    elif llm == "Mistral (FR)":
+        llm = ChatMistralAI(model_name="mistral-large-latest")
         
     chain = prompt | llm | StrOutputParser()
     
@@ -58,32 +66,77 @@ def display_chart():
     plot = construct_plot()
     st.plotly_chart(plot)
 
-@st.experimental_dialog("Cast your vote")
-def vote(item):
-    st.write(f"Why is {item} your favorite?")
-    reason = st.text_input("Because...")
-    if st.button("Submit"):
+@st.experimental_dialog("Choisissez un prompt",width="large")
+def show_prompts():
+    if get_prompts_list() == 1:
+        st.rerun()
+    if st.button("Fermer"):
+        st.rerun()
+
+@st.experimental_dialog("Choisissez votre IA",width="small")
+def choose_model(index):
+    
+    model = st.radio("Choisissez votre IA", ["(US) ChatGpt 4.o","(FR) Mistral AI - Large (open source)"],index=index)
+    if model == "(FR) Mistral AI - Large (open source)":
+        st.session_state.model = "Mistral (FR)"
+    if model == "(US) ChatGpt 4.o":
+        st.session_state.model = "GPT-4o"
+    if st.button("Valider"):
         st.rerun()
 
 def display_chat():
     # app config
     st.title("Chatbot")
 
+    models_name = {
+        "Mistral (FR)":1,
+        "GPT-4o":0
+    }
+
     # session state
     if "chat_history" not in st.session_state:
         st.session_state.chat_history = [
             AIMessage(content="Salut, voici votre cartographie des parties prenantes. Que puis-je faire pour vous?"),
         ]
-    if "user_query" not in st.session_state:
-        st.session_state.user_query = "test"
     
- 
+    if "model" not in st.session_state:
+        st.session_state.model = "GPT-4o"
+    
+    #sticky bar at the top
+    header = st.container()
+    col1,col2,col3 = header.columns([1,2,4])
+
+    if col1.button("Prompts"):
+        show_prompts()
+    if col2.button(f"Modèle: {st.session_state.model}"):
+        index = models_name[st.session_state.model]
+        choose_model(index)
+    header.write("""<div class='fixed-header'/>""", unsafe_allow_html=True)
+
+    # Custom CSS for the sticky header
+    st.markdown(
+        """
+    <style>
+        div[data-testid="stVerticalBlock"] div:has(div.fixed-header) {
+            position: sticky;
+            top: 2.875rem;
+            background-color: white;
+            z-index: 999;
+        }
+        .fixed-header {
+            border-bottom: 0px solid black;
+        }
+    </style>
+        """,
+        unsafe_allow_html=True
+    )
+
     # conversation
     for message in st.session_state.chat_history:
         if isinstance(message, AIMessage):
             with st.chat_message("AI"):
                 st.write(message.content)
-                if "cartographie des parties prenantes" in message.content:
+                if "cartographie" in message.content:
                     display_chart()
         elif isinstance(message, HumanMessage):
             with st.chat_message("Moi"):
@@ -92,11 +145,15 @@ def display_chat():
     #check if the last message is from the user , that means execute button has been clicked
     last_message = st.session_state.chat_history[-1]
     if isinstance(last_message, HumanMessage):
-        response = st.write_stream(get_response(last_message.content, st.session_state.chat_history,format_context(st.session_state['pp_grouped'],st.session_state['Nom de la marque'])))
-        st.session_state.chat_history.append(AIMessage(content=response))
+        with st.chat_message("AI"):
+            response = st.write_stream(get_response(last_message.content, st.session_state.chat_history,format_context(st.session_state['pp_grouped'],st.session_state['Nom de la marque']),st.session_state.model))
+            st.session_state.chat_history.append(AIMessage(content=response))
+
+
 
     if "pp_grouped" not in st.session_state or st.session_state['pp_grouped'] is None or len(st.session_state['pp_grouped']) == 0:
-        return None
+        st.session_state['pp_grouped'] = []
+
     # user input
     user_query = st.chat_input("Par ici...")
     if user_query is not None and user_query != "":
@@ -104,18 +161,9 @@ def display_chat():
         with st.chat_message("Moi"):
             st.markdown(user_query)
         with st.chat_message("AI"):
-            response = st.write_stream(get_response(user_query, st.session_state.chat_history,format_context(st.session_state['pp_grouped'],st.session_state['Nom de la marque'])))
+            st.markdown(f"**{st.session_state.model}**")
+            response = st.write_stream(get_response(user_query, st.session_state.chat_history,format_context(st.session_state['pp_grouped'],st.session_state['Nom de la marque']),st.session_state.model))
             if "cartographie des parties prenantes" in message.content:
                     display_chart()
 
         st.session_state.chat_history.append(AIMessage(content=response))
-    
-
-def display_container_chat():
-    st.markdown(":point_left: Cliquez pour discuter avec le Chatbot", unsafe_allow_html=True)
-    col1, col2 = st.columns([1, 10])
-    with col1:
-        if st.button("display_prompts"):
-            vote("D")
-    with col2:
-        display_chat()

high_chart.py

@@ -190,21 +190,18 @@ def test_chart():
    chart = hct.streamlit_highcharts(cd2,640) #640 is the chart height
    # if chart:
    #    st.session_state['pp_grouped'] = chart
-   if st.session_state['save']:
-      st.session_state['save'] = False
-      st.session_state['pp_grouped'] = chart.copy()
    emp = st.empty()
 
    
 
    col0, col1, col2, col3 = st.columns([1,1,1,1])
 
-   if col1.button("Sauvegarder"):
+   if col1.button("Sauvegarder",key="save"):
       st.session_state['pp_grouped'] = chart.copy()
       emp.success("Saved")
       st.experimental_rerun()
 
-   if col2.button("Télécharger"):
+   if col2.button("Télécharger",key="download"):
       if not chart:
          st.error("Un truc ne va pas. Veuillez réessayer")
       else:

partie_prenante_carte.py

@@ -2,6 +2,7 @@ import streamlit as st
 import pandas as pd
 import numpy as np
 import re
+import random
 
 import streamlit as st
 from dotenv import load_dotenv
@@ -124,8 +125,6 @@ def display_list_urls():
 
         # Button to delete the entry, placed in the second column
         if col2.button("❌", key=f"but{index}"):
-
-            st.session_state["save"] = True
             temp  = st.session_state['parties_prenantes'][index]
             delete_pp(temp)
             del st.session_state.urls[index]
@@ -164,13 +163,25 @@ def display_list_pps():
 
                 
 def extract_pp(docs,input_variables):
-    template_extraction_PP = '''
-        Objectif : identifiez tout les parties prenantes de la marque suivante:
+    template_extraction_PP = """
+    Objectif : Identifiez toutes les parties prenantes de la marque suivante :
+
+    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.
 
-        Le nom de la marque de référence est le suivant : {BRAND_NAME}
+    Exemples de bonnes réponses :
+    - Nike
+    - Adidas
+    - Puma
 
-        TA REPONSE DOIT ETRE SOUS FORME DE LISTE DE NOMS DE MARQUES SANS INCLURE LE NOM DE LA MARQUE DE REFERENCE SANS NUMEROTATION ET SEPARES PAR DES RETOURS A LA LIGNE
-        '''
+    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
 
     if docs == None:
@@ -191,6 +202,8 @@ def extract_pp(docs,input_variables):
     # version plus poussée a considérer
     # each item in the list is a list with the name of the brand and the similarity percentage
     # partie_prenante = text_to_list(response.content)
+    if "ne sais pas" in response.content:
+        return "444"
 
     #version simple
     partie_prenante = response.content.replace("- ","").split('\n')
@@ -233,22 +246,28 @@ def add_pp(new_pp, default_value=50):
 
 def add_pp_input_text():
     new_pp = st.text_input("Ajouter une partie prenante")
-    if st.button("Ajouter"):
-        st.session_state["save"] = True
+    if st.button("Ajouter",key="add_single_pp"):
         format_pp_add_viz(new_pp)
 
-import re
-import random
 
 def complete_and_verify_url(partial_url):
     # Regex pattern for validating a URL
     regex = re.compile(
         r'^(?:http|ftp)s?://'  # http:// or https://
-        r'(?:(?:[A-Z0-9](?:[A-Z0-9-]{0,61}[A-Z0-9])?\.)+[A-Z]{2,6}\.?|'  # domain
+        r'(?:(?:[A-Z0-9](?:[A-Z0-9-]{0,61}[A-Z0-9])?\.)+[A-Z]{2,8}\.?|'  # domain
         r'localhost|'  # localhost...
         r'\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3})'  # ...or ip
         r'(?::\d+)?'  # optional port
         r'(?:/?|[/?]\S+)$', re.IGNORECASE)
+    
+    regex = re.compile(
+        r'^(?:http|ftp)s?://'  # http:// or https://
+        r'(?:(?:[A-Z0-9](?:[A-Z0-9-]{0,61}[A-Z0-9])?\.)+[A-Z]{2,8}\.?|'  # domain name
+        r'localhost|'  # or localhost
+        r'\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3})'  # or IPv4 address
+        r'(?::\d+)?'  # optional port
+        r'(?:[/?#][^\s]*)?$',  # optional path, query, or fragment
+        re.IGNORECASE)
 
     # Complete the URL if it doesn't have http:// or https://
     if not partial_url.startswith(('http://', 'https://', 'www.')):
@@ -268,6 +287,31 @@ def complete_and_verify_url(partial_url):
         return (True, complete_url)
     else:
         return (False, complete_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()
 
 
 def display_pp():
@@ -285,9 +329,11 @@ def display_pp():
         st.session_state['parties_prenantes'] = []
     if "pp_grouped" not in st.session_state: #servira pour le plot et la cartographie des parties prenantes, regroupe sans doublons
         st.session_state['pp_grouped'] = []
+    if "latest_doc" not in st.session_state:
+        st.session_state['latest_doc'] = ""
+    if "not_pp" not in st.session_state:
+        st.session_state["not_pp"] = ""
     
-    if "save" not in st.session_state:
-        st.session_state["save"] = False
 
     st.header("Parties prenantes de la marque")
     #set brand name and description
@@ -300,11 +346,12 @@ def display_pp():
     if option == "A partir de votre site web":
 
         url = st.text_input("Ajouter une URL")
-
-        scraping_option = st.radio("Mode", ("Analyse rapide", "Analyse profonde"),horizontal=True)
+        
+        captions = ["L’IA prend en compte uniquement les textes contenus dans les pages web analysées","L’IA prend en compte les textes, les images et les liens URL contenus dans les pages web analysées"]
+        scraping_option = st.radio("Mode", ("Analyse rapide", "Analyse profonde"),horizontal=True,captions = captions)
         #if the user clicks on the button
-        if st.button("ajouter"):
-            st.session_state["save"] = True
+        if st.button("ajouter",key="add_pp"):
+            st.session_state["not_pp"] = ""
             #complete and verify the url
             is_valid,url = complete_and_verify_url(url)
             if not is_valid:
@@ -326,14 +373,11 @@ def display_pp():
                         docs = get_docs_from_website([url])
                 
                 if docs is None:
-                    st.error("Erreur lors de la collecte des données")
+                    st.error("Erreur lors de la collecte des données, URL unvalide")
+                    st.session_state["latest_doc"] = ""
                 else:
                 # Création de l'expander
-                    with st.expander("Cliquez ici pour éditer et voir le document"):
-                        cleaned_text = re.sub(r'\n\n+', '\n\n', docs[0].page_content.strip())
-                        text_value = st.text_area("Modifier le texte ci-dessous:", value=cleaned_text, height=300)
-                        if st.button('Sauvegarder'):
-                            st.success("Texte sauvegardé avec succès!")
+                    st.session_state["latest_doc"] = docs
                     
                     with st.spinner("Processing..."):
 
@@ -341,14 +385,17 @@ def display_pp():
                         input_variables = {"BRAND_NAME": brand_name, "BRAND_DESCRIPTION": ""}
                         partie_prenante = extract_pp(docs, input_variables)
 
-                        if "444" in partie_prenante: #444 is the code for no brand found , chosen
-                            st.error("Aucune partie prenante trouvée")
-                        elif "445" in partie_prenante: #445 is the code for no website found with the given url
-                            st.error("Aucun site web trouvé avec l'url donnée")
-                        else:
-                            partie_prenante = sorted(partie_prenante)
-                            st.session_state["urls"].append(url)
-                            add_pp(partie_prenante)
+                    if "444" in partie_prenante: #444 is the code for no brand found , chosen
+                        st.session_state["not_pp"] = "444"
+
+                    elif "445" in partie_prenante: #445 is the code for no website found with the given url
+                        st.error("Aucun site web trouvé avec l'url donnée")
+                        st.session_state["not_pp"] = ""
+                    else:
+                        st.session_state["not_pp"] = ""
+                        partie_prenante = sorted(partie_prenante)
+                        st.session_state["urls"].append(url)
+                        add_pp(partie_prenante)
                     
                     # alphabet = [ pp[0] for pp in partie_prenante]
                     # pouvoir = [ 50 for _ in range(len(partie_prenante))]
@@ -366,6 +413,21 @@ def display_pp():
 
                     # disp_vertical_slider(partie_prenante)
                     # st.altair_chart(c, use_container_width=True)
+
+    if st.session_state["not_pp"] == "444":
+        st.warning("Aucune parties prenantes n'est identifiable sur l'URL fournie. Fournissez une autre URL ou bien cliquez sur le boutton ci-dessous pour un Conseils IA")
+                        
+        if st.button("Conseil IA"):
+            show_conseil_ia()
+    #display docs
+    if st.session_state["latest_doc"] != "":
+        with st.expander("Cliquez ici pour éditer et voir le document"):
+            docs = st.session_state["latest_doc"]
+            cleaned_text = re.sub(r'\n\n+', '\n\n', docs[0].page_content.strip())
+            text_value = st.text_area("Modifier le texte ci-dessous:", value=cleaned_text, height=300)
+            if st.button('Sauvegarder',key="save_doc_fake"):
+                st.success("Texte sauvegardé avec succès!")
+    
     display_list_urls()
     with st.expander("Liste des parties prenantes"):
         add_pp_input_text()

prompt.py

@@ -41,10 +41,11 @@ def get_prompts_list():
                     button_phold = col4.empty()  # create a placeholder
                     but1, but2 = button_phold.columns(2)
                     
-                    do_action = but1.button('Voir plus', key=i)
+                    do_action = but1.button('Voir plus', key=f"v{i}")
                     execute = but2.button('Executer', key=f"e{i}")
                     if execute:
                         st.session_state.chat_history.append(HumanMessage(content=prompts[i]['text']))
+                        return 1
                     if do_action:
                         prompt_html = prompts[i]['text'].replace('\n', '<br>')
                         prompt_metadata = extract_metadata(prompts[i])