Update utils.py

utils.py

@@ -105,7 +105,7 @@ def identifier_cardinal(text):
 
     return None
 
-def link_to_subject_object(g, head_type, head_uri, relation_type, relation_word, prev_span, next_span, statement_uris, BD, RDF):
+def link_to_subject_object(g, head_type, head_uri, relation_type, relation_word, prev_span, next_span, statement_uris, EKG, RDF):
 
     relation_triplets = {}
     with open('./assets/reference_triplet_relation.csv', mode='r', encoding='utf-8') as f:
@@ -126,7 +126,7 @@ def link_to_subject_object(g, head_type, head_uri, relation_type, relation_word,
 
         # cas où le prédicat est une distance ou une orientation
         if relation_type != "Distance-Orientation":
-            stmt_uri = BD[f"Statement{len(statement_uris)}"]
+            stmt_uri = EKG[f"Statement{len(statement_uris)}"]
 
             if relation_word.lower() in statement_uris:
                 key = relation_word.lower() + str(len(statement_uris))
@@ -134,35 +134,35 @@ def link_to_subject_object(g, head_type, head_uri, relation_type, relation_word,
                 key = relation_word.lower()
             
             statement_uris[key] = stmt_uri
-            add_triplet(g, stmt_uri, RDF.subject, BD[subject_uri])
-            add_triplet(g, stmt_uri, RDF.object, BD[next_span['uri']])
-            add_triplet(g, stmt_uri, RDF.predicate, BD[relation_type.lower()])
+            add_triplet(g, stmt_uri, RDF.subject, EKG[subject_uri])
+            add_triplet(g, stmt_uri, RDF.object, EKG[next_span['uri']])
+            add_triplet(g, stmt_uri, RDF.predicate, EKG[relation_type.lower()])
         else:
             cardinal = identifier_cardinal(relation_word)
             distance = identifier_distance(relation_word)
 
             if cardinal:
-                stmt_uri = BD[f"Statement{len(statement_uris)}"]
+                stmt_uri = EKG[f"Statement{len(statement_uris)}"]
                 if relation_word.lower() in statement_uris:
                     key = relation_word.lower() + str(len(statement_uris))
                 else:
                     key = relation_word.lower()
                 statement_uris[key] = stmt_uri
-                add_triplet(g, stmt_uri, RDF.subject, BD[subject_uri])
-                add_triplet(g, stmt_uri, RDF.object, BD[next_span['uri']])
-                add_triplet(g, stmt_uri, RDF.predicate, BD['orientation'])
+                add_triplet(g, stmt_uri, RDF.subject, EKG[subject_uri])
+                add_triplet(g, stmt_uri, RDF.object, EKG[next_span['uri']])
+                add_triplet(g, stmt_uri, RDF.predicate, EKG['orientation'])
                 add_triplet(g, stmt_uri, RDF.value, Literal(cardinal))
 
             if distance:
-                stmt_uri = BD[f"Statement{len(statement_uris)}"]
+                stmt_uri = EKG[f"Statement{len(statement_uris)}"]
                 if relation_word.lower() in statement_uris:
                     key = relation_word.lower() + str(len(statement_uris))
                 else:
                     key = relation_word.lower()
                 statement_uris[key] = stmt_uri
-                add_triplet(g, stmt_uri, RDF.subject, BD[subject_uri])
-                add_triplet(g, stmt_uri, RDF.object, BD[next_span['uri']])
-                add_triplet(g, stmt_uri, RDF.predicate, BD['distance'])
+                add_triplet(g, stmt_uri, RDF.subject, EKG[subject_uri])
+                add_triplet(g, stmt_uri, RDF.object, EKG[next_span['uri']])
+                add_triplet(g, stmt_uri, RDF.predicate, EKG['distance'])
                 add_triplet(g, stmt_uri, RDF.value, Literal(distance))