V1.0 RSECategorizer.py

RSECategorizer.py

@@ -0,0 +1,32 @@
+# RSECategorizer.py
+
+from transformers import pipeline
+import pandas as pd
+
+# Charger le pipeline de classification avec un modèle léger
+classifier = pipeline("zero-shot-classification", model="typeform/distilbert-base-uncased-mnli")
+
+def classify_rse_actions(descriptions):
+    categories = [
+        "La gouvernance de la structure",
+        "Les droits humains",
+        "Les conditions et relations de travail",
+        "La responsabilité environnementale",
+        "La loyauté des pratiques",
+        "Les questions relatives au consommateur et à la protection du consommateur",
+        "Les communautés et le développement local"
+    ]
+    
+    classified_data = []
+    for description in descriptions:
+        # Classification de chaque description
+        result = classifier(description, categories)
+        # Récupération de la catégorie avec la probabilité la plus élevée
+        top_category = result['labels'][0]
+        classified_data.append(top_category)
+    
+    return classified_data
+
+# Exemple d'utilisation (à des fins de test, à commenter ou supprimer pour l'intégration finale)
+# descriptions = ["Promotion de l'énergie renouvelable", "Amélioration des conditions de travail"]
+# print(classify_rse_actions(descriptions))

app.py

@@ -3,6 +3,7 @@ import pandas as pd
 import requests
 import folium
 from streamlit_folium import folium_static
+from transformers import pipeline
 
 # Fonction pour récupérer les données de l'API
 def get_data():
@@ -33,33 +34,58 @@ def display_organisations_engagees():
         df = df[["Nom", "Commune", "Section NAF", "Effectif", "Action RSE"]]
         st.dataframe(df, width=None, height=None)
 
-# Nouvelle fonction pour l'onglet "GeoRSE Insights"
+# Fonction pour l'onglet "GeoRSE Insights"
 def display_geo_rse_insights():
+    # La même fonction que celle définie précédemment pour afficher la carte
+
+# Classification des actions RSE basée sur les descriptions
+def classify_rse_actions(descriptions):
+    classifier = pipeline("zero-shot-classification", model="typeform/distilbert-base-uncased-mnli")
+    categories = [
+        "La gouvernance de la structure",
+        "Les droits humains",
+        "Les conditions et relations de travail",
+        "La responsabilité environnementale",
+        "La loyauté des pratiques",
+        "Les questions relatives au consommateur et à la protection du consommateur",
+        "Les communautés et le développement local"
+    ]
+    
+    classified_data = []
+    for description in descriptions:
+        result = classifier(description, categories)
+        top_category = result['labels'][0]
+        classified_data.append(top_category)
+    
+    return classified_data
+
+# Nouvelle fonction pour l'onglet de classification RSE
+def display_rse_categorizer():
+    st.header("Classification des Actions RSE")
+    st.write("Classification automatique des actions RSE des entreprises.")
+    
     data, _ = get_data()
     if data:
-        m = folium.Map(location=[44.84474, -0.60711], zoom_start=11)
-        for item in data:
-            point_geo = item.get('point_geo', [])
-            if point_geo:
-                lat, lon = point_geo
-                lat, lon = float(lat), float(lon)
-                if lat and lon:
-                    folium.Marker(
-                        [lat, lon],
-                        popup=f"<b>{item.get('nom_courant_denomination', 'Sans nom')}</b><br>Action RSE: {item.get('action_rse', 'Non spécifié')}",
-                        icon=folium.Icon(color="green", icon="leaf"),
-                    ).add_to(m)
-        folium_static(m)
+        descriptions = [item['action_rse'] for item in data if 'action_rse' in item]
+        categories = classify_rse_actions(descriptions)
+        
+        # Affichage des résultats
+        for i, category in enumerate(categories):
+            st.write(f"Action RSE: {descriptions[i]}")
+            st.write(f"Catégorie prédite: {category}")
+            st.write("---")
 
 # Main function orchestrating the app UI
 def main():
     st.sidebar.title("Navigation")
-    app_mode = st.sidebar.radio("Choisissez l'onglet", ["Organisations engagées", "GeoRSE Insights"])
+    app_mode = st.sidebar.radio("Choisissez l'onglet", ["Organisations engagées", "GeoRSE Insights", "Classification RSE"])
 
     if app_mode == "Organisations engagées":
         display_organisations_engagees()
     elif app_mode == "GeoRSE Insights":
         display_geo_rse_insights()
+    elif app_mode == "Classification RSE":
+        display_rse_categorizer()
 
 if __name__ == "__main__":
     main()