Update src/core/inference_engine.py

src/core/inference_engine.py

@@ -1,7 +1,6 @@
 """
-MODULE: INFERENCE ENGINE (R-BOX REASONER)
-=========================================
-Mise à jour : Règle 'HighValue' rendue plus robuste aux variations de noms de colonnes.
+MODULE: INFERENCE ENGINE
+========================
 """
 from rdflib import Graph, Namespace
 
@@ -12,49 +11,32 @@ class InferenceEngine:
         self.rdf_store = rdf_store
 
     def run_inference(self):
-        print("🧠 [R-BOX] Démarrage du moteur d'inférence...")
+        print("🧠 [R-BOX] Inférence...")
         initial_count = len(self.rdf_store.g)
         
-        # 1. Réciprocité (Relations Inverses)
-        rule_inverse = """
+        # 1. Réciprocité
+        self._apply_rule("""
         CONSTRUCT { ?pret vortex:a_emprunteur ?client . }
         WHERE { ?client vortex:possede_pret ?pret . }
-        """
-        self._apply_rule(rule_inverse, "Inverse: possede_pret -> a_emprunteur")
+        """, "Inverse: possede_pret")
 
-        # 2. Règle 'HighValue' (CORRIGÉE & ASSOUPLIE)
-        # On cherche toute propriété contenant "montant" ou "capital" ayant une valeur > 1M
-        rule_risk = """
-        CONSTRUCT {
-            ?pret vortex:statut_risque "HighValue" .
-        }
+        # 2. Règle HighValue (ROBUSTE)
+        # Cherche n'importe quelle propriété numérique > 1M
+        self._apply_rule("""
+        CONSTRUCT { ?pret vortex:statut_risque "HighValue" . }
         WHERE {
-            ?pret a ?type .
+            ?pret a vortex:Pret .
             ?pret ?p ?valeur .
-            FILTER(
-                (CONTAINS(LCASE(STR(?p)), "montant") || CONTAINS(LCASE(STR(?p)), "capital")) 
-                && 
-                xsd:decimal(?valeur) > 1000000
-            )
+            FILTER(isNumeric(?valeur) && ?valeur > 1000000)
         }
-        """
-        self._apply_rule(rule_risk, "Logique: Prêt Valeur Élevée (>1M)")
+        """, "Risque: HighValue > 1M")
 
-        final_count = len(self.rdf_store.g)
-        print(f"✅ [R-BOX] Inférence terminée. {final_count - initial_count} nouveaux faits.")
+        print(f"✅ [R-BOX] +{len(self.rdf_store.g) - initial_count} faits déduits.")
 
-    def _apply_rule(self, sparql_construct, rule_name):
+    def _apply_rule(self, sparql, name):
         try:
-            full_query = f"""
-            PREFIX vortex: <http://vortex.ai/ontology#>
-            PREFIX xsd: <http://www.w3.org/2001/XMLSchema#>
-            {sparql_construct}
-            """
-            inferred_triples = self.rdf_store.g.query(full_query)
-            count = 0
-            for triple in inferred_triples:
-                self.rdf_store.g.add(triple)
-                count += 1
-            if count > 0: print(f"   🔹 Règle '{rule_name}' : +{count} triplets.")
+            full = f"PREFIX vortex: <http://vortex.ai/ontology#>\nPREFIX xsd: <http://www.w3.org/2001/XMLSchema#>\n{sparql}"
+            res = self.rdf_store.g.query(full)
+            for t in res: self.rdf_store.g.add(t)
         except Exception as e:
-            print(f"   ❌ Erreur Règle '{rule_name}' : {e}")
+            print(f"❌ Erreur Règle '{name}' : {e}")