Update src/modules/ontology_graph.py

src/modules/ontology_graph.py

@@ -0,0 +1,181 @@
+import streamlit as st
+import pandas as pd
+import networkx as nx
+from pyvis.network import Network
+import tempfile
+import streamlit.components.v1 as components
+import hashlib
+
+# --- GÉNÉRATEUR DE COULEURS GOTHAM AUTOMATIQUE ---
+def get_auto_gotham_color(entity_type):
+    """
+    Génère une couleur unique et déterministe pour chaque type d'entité.
+    Ne nécessite aucune définition manuelle.
+    Assure un look 'Gotham' (Teintes froides, saturées, lisibles sur fond noir).
+    """
+    # 1. On hache le nom du type (ex: "Client", "Pret") pour avoir un nombre unique
+    hash_object = hashlib.md5(entity_type.encode())
+    hash_hex = hash_object.hexdigest()
+    
+    # 2. On utilise le hash pour choisir une teinte (Hue)
+    # Gotham Palette Logic : On favorise les Bleus, Cyans, Violets, Ors, Verts d'eau
+    # On évite les marrons ou roses bonbons trop vifs.
+    
+    # Liste de palettes "Tech/Safe"
+    gotham_palettes = [
+        "#25C1F7", # Cyan
+        "#D4AF37", # Gold
+        "#54BD4B", # Green Tech
+        "#9B59B6", # Purple
+        "#E74C3C", # Red Alert
+        "#34495E", # Steel Blue
+        "#1ABC9C", # Turquoise
+        "#F39C12", # Orange
+        "#7F8C8D", # Grey
+        "#2980B9"  # Deep Blue
+    ]
+    
+    # On prend une couleur dans la liste modulo la taille de la liste
+    # Cela garantit que tous les "Clients" auront toujours la même couleur, idem pour "Prêts"
+    idx = int(hash_hex, 16) % len(gotham_palettes)
+    return gotham_palettes[idx]
+
+def show_ontology_graph(client, sheet_name):
+    st.header("🕸️ ONTOLOGIE : LE SYSTÈME NERVEUX")
+    
+    # 1. Chargement des Données
+    try:
+        sh = client.open(sheet_name)
+        # On utilise get_all_records pour avoir des dicts
+        df_clients = pd.DataFrame(sh.worksheet("Clients_KYC").get_all_records())
+        df_prets = pd.DataFrame(sh.worksheet("Prets_Master").get_all_records())
+    except Exception as e:
+        st.error(f"Erreur de lecture des données : {e}")
+        return
+
+    # 2. Construction du Graphe (NetworkX)
+    G = nx.Graph()
+
+    # --- NOEUD CENTRAL : HQ ---
+    hq_color = get_auto_gotham_color("HQ")
+    # shape="box" pour faire une carte
+    G.add_node("VORTEX_HQ", label="🏢 VORTEX HQ\n(Siège Social)", title="Centre Névralgique", color=hq_color, shape="box", font={'color': 'white'})
+
+    # --- COUCHE 1 : LES CLIENTS (Cartes) ---
+    if not df_clients.empty:
+        client_color = get_auto_gotham_color("Client") # Automatique
+        
+        for _, row in df_clients.iterrows():
+            c_id = str(row['ID_Client'])
+            c_nom = str(row['Nom_Complet'])
+            c_ville = str(row.get('Ville', 'Inconnu'))
+            c_rev = str(row.get('Revenus_Mensuels', '0'))
+            
+            # FORMATTAGE TYPE "CARTE" AVEC SAUTS DE LIGNE
+            # Pyvis supporte le multi-ligne dans les labels
+            card_label = f"👤 {c_nom}\n🆔 {c_id}\n📍 {c_ville}"
+            
+            # Métadonnées complètes au survol
+            hover_html = f"""
+            <b>CLIENT CARD</b><br>
+            ----------------<br>
+            Nom: {c_nom}<br>
+            ID: {c_id}<br>
+            Revenus: {c_rev} XOF
+            """
+            
+            G.add_node(c_id, label=card_label, title=hover_html, color=client_color, shape="box", font={'color': 'white'})
+            
+            # Lien HQ
+            G.add_edge("VORTEX_HQ", c_id, color="#404854", weight=2)
+
+    # --- COUCHE 2 : LES PRÊTS (Cartes) ---
+    if not df_prets.empty:
+        # Ici on peut varier la couleur selon le STATUT automatiquement
+        # ex: "Pret_ACTIF", "Pret_CLOTURE"
+        
+        for _, row in df_prets.iterrows():
+            p_id = str(row['ID_Pret'])
+            c_ref = str(row['ID_Client'])
+            montant = str(row['Montant_Capital'])
+            statut = str(row['Statut'])
+            duree = str(row['Duree_Semaines'])
+            
+            # Couleur auto basée sur le statut (ex: Pret_ACTIF aura une couleur, Pret_CLOSE une autre)
+            loan_color = get_auto_gotham_color(f"Pret_{statut}")
+            
+            # FORMATTAGE CARTE FINANCIÈRE
+            card_label = f"📄 CONTRAT {statut}\n💰 {montant} XOF\n⏱️ {duree} Semaines"
+            
+            hover_html = f"""
+            <b>LOAN CARD</b><br>
+            ----------------<br>
+            Ref: {p_id}<br>
+            Montant: {montant}<br>
+            Status: {statut}
+            """
+            
+            G.add_node(p_id, label=card_label, title=hover_html, color=loan_color, shape="box", font={'color': 'white'})
+            
+            # Lien Relationnel Client <-> Prêt
+            # (Si le client existe dans le graphe)
+            if c_ref in list(G.nodes):
+                G.add_edge(c_ref, p_id, color="#404854", weight=1, label="Possède")
+
+    # 3. Visualisation Interactive (PyVis)
+    nt = Network(height="750px", width="100%", bgcolor="#10161A", font_color="white", notebook=False)
+    nt.from_nx(G)
+    
+    # OPTIONS PHYSIQUES POUR "EFFET CARTE"
+    # On force les noeuds à s'écarter davantage car ce sont des rectangles
+    nt.options = {
+        "nodes": {
+            "borderWidth": 2,
+            "borderWidthSelected": 4,
+            "font": {"size": 16, "face": "Space Grotesk", "align": "center"},
+            "shadow": {"enabled": True, "color": "rgba(0,0,0,0.5)", "size": 10}
+        },
+        "edges": {
+            "color": {"color": "#404854", "highlight": "#25C1F7"},
+            "smooth": {"type": "continuous"},
+            "width": 1
+        },
+        "physics": {
+            "forceAtlas2Based": {
+                "gravitationalConstant": -100, # Répulsion forte pour ne pas chevaucher les cartes
+                "centralGravity": 0.01,
+                "springLength": 200, # Liens longs
+                "springConstant": 0.05
+            },
+            "solver": "forceAtlas2Based",
+            "stabilization": {"enabled": True, "iterations": 100}
+        },
+        "interaction": {
+            "hover": True,
+            "tooltipDelay": 200
+        }
+    }
+
+    # Rendu
+    try:
+        path = tempfile.gettempdir() + "/ontology.html"
+        nt.save_graph(path)
+        with open(path, 'r', encoding='utf-8') as f:
+            source_code = f.read()
+        
+        # Injection CSS pour forcer le style Gotham sur les tooltips
+        custom_css = """
+        <style>
+        div.vis-tooltip {
+            background-color: #182026 !important;
+            color: #F5F8FA !important;
+            border: 1px solid #25C1F7 !important;
+            font-family: 'Space Grotesk', sans-serif !important;
+            box-shadow: 0px 0px 15px rgba(37, 193, 247, 0.3) !important;
+        }
+        </style>
+        """
+        components.html(custom_css + source_code, height=800)
+        
+    except Exception as e:
+        st.error(f"Erreur de rendu graphique : {e}")