upgrades multi url

partie_prenante_carte.py

@@ -21,6 +21,7 @@ import altair as alt
 from session import set_partie_prenante
 import os
 from streamlit_vertical_slider import vertical_slider
+from pp_viz import display_viz
 
 load_dotenv()
 
@@ -90,6 +91,25 @@ def text_to_list(text):
         item[1] = re.sub(r'\D', '', item[1])
     return items
 
+def display_list_urls():
+    for index, item in enumerate(st.session_state["urls"]):
+        emp = st.empty()  # Create an empty placeholder
+        col1, col2 = emp.columns([7, 3])  # Divide the space into two columns
+
+        # Button to delete the entry, placed in the second column
+        if col2.button("❌", key=f"but{index}"):
+            del st.session_state.urls[index]
+            del st.session_state["parties_prenantes"][index]
+            st.experimental_rerun()  # Rerun the app to update the display
+
+        if len(st.session_state.urls) > index:
+            # Instead of using markdown, use an expander in the first column
+            with col1.expander(f"URL {index}: {item}"):
+                pp = st.session_state["parties_prenantes"][index]
+                st.write(pd.DataFrame(pp, columns=["Partie prenante"]))
+        else:
+            emp.empty()  # Clear the placeholder if the index exceeds the list
+
 def extract_pp(urls,input_variables):
     template_extraction_PP = '''
     Objectif : identifiez et proposez tout les noms de marques qui serviront comme partie prenante de la marque suivante pour développer un marketing de coopération (co-op marketing)
@@ -147,41 +167,69 @@ def disp_vertical_slider(partie_prenante):
             st.write(tst)
 
 
+
 def display_pp():
+
     load_dotenv()
-    st.header("INDIQUEZ VOS PAGES WEB ET/OU DOCUMENTS D’ENTREPRISE POUR AUDITER LE CONTENU RSE")
-    loaded = False
+    #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"] = ""
+    if "Description de la marque" not in st.session_state:
+        st.session_state["Description de la marque"] = ""
+
+    #check if urls and partie prenante are already set
+    if "urls" not in st.session_state:
+        st.session_state["urls"] = []
+    if "parties_prenantes" not in st.session_state:
+        st.session_state['parties_prenantes'] = []
+
+    st.header("Parties prenantes de la marque")
+
+    #set brand name and description
+    brand_name = st.text_input("Nom de la marque", st.session_state["Nom de la marque"])
+    st.session_state["Nom de la marque"] = brand_name
+    brand_description = st.text_input("Description de la marque", st.session_state["Description de la marque"])
+    st.session_state["Description de la marque"] = brand_description
+
     option = st.radio("Source", ("A partir de votre site web", "A partir de vos documents entreprise"))
-    
+    #if the user chooses to extract from website
     if option == "A partir de votre site web":
-        url1 = st.text_input("URL 1")
-        brand_name = st.text_input("Nom de la marque")
-        brand_description = st.text_area("Description de la marque")
-        if st.button("Process") and loaded == False:
-            loaded = True
-            with st.spinner("Processing..."):
-                input_variables = {"BRAND_NAME": brand_name, "BRAND_DESCRIPTION": brand_description}
-                partie_prenante = extract_pp([url1], input_variables)
-                partie_prenante = sorted(partie_prenante)
-                set_partie_prenante(partie_prenante)
-                st.write(pd.DataFrame(partie_prenante, columns=["Partie prenante"]))
-                
-                # alphabet = [ pp[0] for pp in partie_prenante]
-                # pouvoir = [ 50 for _ in range(len(partie_prenante))]
-                # df = pd.DataFrame({'partie_prenante': partie_prenante, 'pouvoir': pouvoir, 'code couleur': partie_prenante})
-                # st.write(df)
-
-                # c = (
-                # alt.Chart(df)
-                # .mark_circle(size=300)
-                # .encode(x="partie_prenante", y=alt.Y("pouvoir",scale=alt.Scale(domain=[0,100])), color="code couleur")
-                # )
-                # st.subheader("Vertical Slider")
-                # age = st.slider("How old are you?", 0, 130, 25)
-                # st.write("I'm ", age, "years old")
-
-                # disp_vertical_slider(partie_prenante)
-                # st.altair_chart(c, use_container_width=True)
+
+        
+        url = st.text_input("Ajouter une URL")
+
+        #if the user clicks on the button
+        if st.button("ajouter"):
+            if url in st.session_state["urls"]:
+                st.error("URL déjà ajoutée")
+            else:
+                with st.spinner("Processing..."):
+
+                    #handle the extraction
+                    input_variables = {"BRAND_NAME": brand_name, "BRAND_DESCRIPTION": brand_description}
+                    partie_prenante = extract_pp([url], input_variables)
+                    partie_prenante = sorted(partie_prenante)
+                    st.session_state['parties_prenantes'].append(partie_prenante)
+                    st.session_state["urls"].append(url)
+                    # alphabet = [ pp[0] for pp in partie_prenante]
+                    # pouvoir = [ 50 for _ in range(len(partie_prenante))]
+                    # df = pd.DataFrame({'partie_prenante': partie_prenante, 'pouvoir': pouvoir, 'code couleur': partie_prenante})
+                    # st.write(df)
+
+                    # c = (
+                    # alt.Chart(df)
+                    # .mark_circle(size=300)
+                    # .encode(x="partie_prenante", y=alt.Y("pouvoir",scale=alt.Scale(domain=[0,100])), color="code couleur")
+                    # )
+                    # st.subheader("Vertical Slider")
+                    # age = st.slider("How old are you?", 0, 130, 25)
+                    # st.write("I'm ", age, "years old")
+
+                    # disp_vertical_slider(partie_prenante)
+                    # st.altair_chart(c, use_container_width=True)
+    display_list_urls()
+    display_viz()
+
                 
 
 

pp_viz.py

@@ -4,21 +4,21 @@ import numpy as np
 import re
 
 import altair as alt
-from session import get_partie_prenante
+from session import get_parties_prenantes
 import os
 from streamlit_vertical_slider import vertical_slider
+from st_draggable_list import DraggableList
 
 def display_viz():
 
-    st.header("")
-    st.title("Visualisation des parties prenantes")
 
-    partie_prenante = get_partie_prenante()
+    parties_prenantes = get_parties_prenantes()
 
-    if partie_prenante is None:
-        st.write("Veuillez entrer l'url dans l'onglet partie prenante avant de visualiser les parties prenantes")
+    if parties_prenantes is None or len(parties_prenantes) == 0:
+        st.write("aucune partie prenante n'a été définie")
     else:
-        alphabet = [ pp[0] for pp in partie_prenante]
+        partie_prenante_non_filtre = [item for sublist in parties_prenantes for item in sublist]
+        partie_prenante = sorted(list(set(partie_prenante_non_filtre)))
         pouvoir = [ 50 for _ in range(len(partie_prenante))]
         df = pd.DataFrame({'partie_prenante': partie_prenante, 'pouvoir': pouvoir, 'code couleur': partie_prenante})
         c = (
@@ -30,17 +30,23 @@ def display_viz():
         number_of_sliders = len(partie_prenante)
         st.write("Modifiez le pouvoir des parties prenantes en utilisant les sliders ci-dessous")
         bar = st.columns(number_of_sliders)
+
+
         for i in range(number_of_sliders):
             with bar[i]:
+                temp = 50
                 df["pouvoir"][i] = vertical_slider(
                     label=partie_prenante[i],
                     height=100,
                     key=partie_prenante[i],
-                    default_value=50,
+                    default_value=temp,
                     thumb_color= "orange", #Optional - Defaults to Streamlit Red
                     step=1,
                     min_value=0,
                     max_value=100,
                     value_always_visible=False,
                 )
-        st.altair_chart(c, use_container_width=True)
+        st.altair_chart(c, use_container_width=True)
+        data = [{'id':partie_prenante[i], 'name':partie_prenante[i],'pouvoir':int(df["pouvoir"][i])} for i in range(len(partie_prenante))]
+        slist = DraggableList(data)
+        st.write(slist)

requirements.txt

@@ -29,3 +29,5 @@ altair[all]
 streamlit-vertical-slider
 streamlit_toggle
 langchain_experimental
+streamlit_draggable_list
+

session.py

@@ -14,10 +14,18 @@ def set_rag(vectorstore, chain):
     st.session_state['chain'] = chain
 
 def set_partie_prenante(partie_prenante):
+    if "partie_prenante" not in st.session_state:
+        st.session_state['partie_prenante'] = []
     st.session_state['partie_prenante'] = partie_prenante
 
 def get_partie_prenante():
-    if 'partie_prenante' in st.session_state:
+    if 'parties_prenante' in st.session_state:
         return st.session_state['partie_prenante']
+    else:
+        return None
+
+def get_parties_prenantes():
+    if 'parties_prenantes' in st.session_state:
+        return st.session_state['parties_prenantes']
     else:
         return None