Update app.py

app.py

@@ -14,9 +14,15 @@ ner = pipeline("token-classification", model="GEODE/camembert-base-edda-span-cla
 placename_classifier = pipeline("text-classification", model="GEODE/bert-base-multilingual-cased-classification-ner", truncation=True)
 relation_classifier = pipeline("text-classification", model="GEODE/bert-base-multilingual-cased-classification-relation", truncation=True)
 generator = pipeline("text2text-generation", model="GEODE/mt5-small-coords-norm", truncation=True)
-
-
-
+'''
+entry_type_classifier = pipeline("text-classification", model="no-name-research/multilingual-bert-entry-type-classifier", truncation=True)
+place_type_classifier = pipeline("text-classification", model="no-name-research/multilingual-bert-place-type-classifier", truncation=True)
+cardinality_classifier = pipeline("text-classification", model="no-name-research/multilingual-bert-cardinality-classifier", truncation=True)
+ner = pipeline("token-classification", model="no-name-research/camembert-token-classification", aggregation_strategy="simple")
+placename_classifier = pipeline("text-classification", model="no-name-research/multilingual-bert-place-ner-classifier", truncation=True)
+relation_classifier = pipeline("text-classification", model="no-name-research/multilingual-bert-spatial-relations-classifier", truncation=True)
+generator = pipeline("text2text-generation", model="anonymous-research/mt5-coordinates", truncation=True)
+'''
 
 def build_visualization(graph, height="1000px", width="100%"):
     """Convert the rdflib graph into a pyvis network graph for visualization"""
@@ -70,13 +76,13 @@ def run_pipeline(text):
     placenames = []
     relations = []
 
-    BD = Namespace("http://exemple.bd/")
+    EKG = Namespace("http://encyclokg.geo/")
     SKOS = Namespace("http://www.w3.org/2004/02/skos/core#")
     g = Graph()
     g.bind("rdf", RDF)
     g.bind("rdfs", RDFS)
     g.bind("xsd", XSD)
-    g.bind("bd", BD)
+    g.bind("ekg", EKG)
     g.bind("skos", SKOS)
    
     entity_uris = {}
@@ -105,31 +111,31 @@ def run_pipeline(text):
                     span['uri'] = uri_entity
                     span['label'] = label[0]['label']
                     entity_uris[word] = uri_entity
-                    add_triplet(g, BD[uri_entity], RDF.type, BD[label[0]['label']])
-                    #TODO: normaliser les noms par exemple ceux qui commencent par l'
+                    add_triplet(g, EKG[uri_entity], RDF.type, EKG[label[0]['label']])
+    
                     if word.startswith("l'"):
                         word = word[2:]
-                    add_triplet(g, BD[uri_entity], SKOS['prefLabel'], Literal(word))
+                    add_triplet(g, EKG[uri_entity], SKOS['prefLabel'], Literal(word))
                 if span['entity_group'] == 'Latlong':
                     latlong = True
                     
             
             entity_uris[head] = uri_article
-            add_triplet(g, BD[uri_article], RDF.type, BD[place_type[0]['label']])
+            add_triplet(g, EKG[uri_article], RDF.type, EKG[place_type[0]['label']])
             
             if aliases:
                 if len(aliases) > 0:
-                    add_triplet(g, BD[uri_article], SKOS['prefLabel'], Literal(aliases[0]))
+                    add_triplet(g, EKG[uri_article], SKOS['prefLabel'], Literal(aliases[0]))
                 for alias in aliases[1:]:
-                    add_triplet(g, BD[uri_article], SKOS['altLabel'], Literal(alias))
+                    add_triplet(g, EKG[uri_article], SKOS['altLabel'], Literal(alias))
             else:
-                add_triplet(g, BD[uri_article], SKOS['prefLabel'], Literal(head))
+                add_triplet(g, EKG[uri_article], SKOS['prefLabel'], Literal(head))
             
             if latlong:
                 predicted_coordinates_from_pipeline = generator(text, max_length=128)
                 coords = dms_to_dd(predicted_coordinates_from_pipeline[0]['generated_text'])
-                add_triplet(g, BD[uri_article], BD['latitude'], Literal(coords[0], datatype=XSD.float))
-                add_triplet(g, BD[uri_article], BD['longitude'], Literal(coords[1], datatype=XSD.float))
+                add_triplet(g, EKG[uri_article], EKG['latitude'], Literal(coords[0], datatype=XSD.float))
+                add_triplet(g, EKG[uri_article], EKG['longitude'], Literal(coords[1], datatype=XSD.float))
 
             # filter out non NP_Spatial or Relation spans
             filtered_spans = [span for span in spans if span['entity_group'] in ['NP_Spatial', 'Relation']]
@@ -142,16 +148,15 @@ def run_pipeline(text):
                     relations.append({'relation': word, 'label': label[0]['label']})
 
                     if idx > 0 and idx < len(filtered_spans) - 1:   
-                        link_to_subject_object(g, place_type[0]['label'], uri_article, label[0]['label'],word, filtered_spans[idx-1], filtered_spans[idx+1], statement_uris, BD, RDF)
+                        link_to_subject_object(g, place_type[0]['label'], uri_article, label[0]['label'],word, filtered_spans[idx-1], filtered_spans[idx+1], statement_uris, EKG, RDF)
 
-            #TODO: add triplet like ville de France: NC_Spatial + de + NP_Spatial (type Pays ou Region)
             span_object = pattern_starting_article(text, spans)
             if span_object is not None:
-                stmt_uri = BD[f"Statement{len(statement_uris)}"]
+                stmt_uri = EKG[f"Statement{len(statement_uris)}"]
                 statement_uris["de"] = stmt_uri
-                add_triplet(g, stmt_uri, RDF.subject, BD[uri_article])
-                add_triplet(g, stmt_uri, RDF.object, BD[entity_uris[span_object['word']]])
-                add_triplet(g, stmt_uri, RDF.predicate, BD["inclusion"])
+                add_triplet(g, stmt_uri, RDF.subject, EKG[uri_article])
+                add_triplet(g, stmt_uri, RDF.object, EKG[entity_uris[span_object['word']]])
+                add_triplet(g, stmt_uri, RDF.predicate, EKG["inclusion"])
         
             
             img_path, html_path = create_graph_viz(g)
@@ -166,9 +171,9 @@ def run_pipeline(text):
 
 
 examples = [
+    "Jean de Luz, S. (Géog.) Lucius Vicus ; le nom basque est Loitzun, petite ville de France en Gascogne, la deuxieme du pays de Labour, & la derniere du côté de l'Espagne, avec un port. Elle est sur une petite riviere, que Piganiol de la Force nomme la Ninette, & M. de Lisle le Nivelet, à 4 lieues N. E. de Fontarabie, 4 S. O. de Bayonne, 174 S. O. de Paris. Long. 15. 59. 28. lat. 43. 23. 15. (D. J.)",
     "* AACH ou ACH, s. f. petite ville d'Allemagne dans le cercle de Souabe, près de la source de l'Aach. Long. 26. 57. lat. 47. 55.",
-    "* ARCALU (Principauté d') petit état des Tartares-Monguls, sur la riviere d'Hoamko, où commence la grande muraille de la Chine, sous le 122e degré de longitude & le 42e de latitude septentrionale.",
-    "* ARÉQUIPE, ou ARIQUIPA, (Géog.) ville de l'Amérique méridion. dans le Pérou, sur une riviere, dans un terrein fertile. Long. 308. lat. mérid. 16. 40.",
+    "OCHRIDA, lac d', (Géog.) lac de la Turquie en Europe, entre l'Albanie au couchant, & le Coménolitari  au levant. Ce lac n'a qu'une demi-lieue de large sur dix lieues de long, & une seule ville du même nom, autrement dite Giustandil. Les anciens ont connu ce lac sous le nom de lacus Lycuicus. ",
     "* AUTAN-KELURAN, (Géog.) ville du Turquestan. Long. 110d. & lat. 46. 45. selon Uluhbeg ; & long. 116. & lat. 45. selon Nassiredden.",
     "Boston ; c'est le nom qu'on a donné à la ville capitale de la nouvelle Angleterre, dans l'Amérique septentrionale ; elle est grande & a un très-bon port. Lat. 42 degrés, 20 minutes ; long. 306 degrés, 50 & quelques minutes.",
 ]
@@ -210,7 +215,6 @@ with gr.Blocks() as demo:
 
             out_places = gr.Textbox(label="Placenames and their type (City/River/Mountain/Other...)", lines=4)
             out_relations = gr.Textbox(label="Spatial relations and their type (Adjacency/Orientation/Distance/Other...)", lines=4)
-            #out_map = gr.Map(label="Coordinates on Map")
             run_btn.click(fn=run_pipeline, inputs=inp, outputs=[img_output, html_output, out_type_entry, out_place_type, out_cardinality, out_places, out_relations])