Update app.py

app.py

@@ -4,7 +4,7 @@ from typing import List
 from pydantic import BaseModel
 from fastapi import FastAPI, HTTPException
 import rdflib
-from rdflib import RDF, RDFS, OWL
+from rdflib import RDF, RDFS, OWL, URIRef
 from sentence_transformers import SentenceTransformer
 import faiss
 import json
@@ -93,6 +93,11 @@ def extract_ontology(rdf_file: str, output_file: str):
     for entity in g.subjects(RDF.type, OWL.NamedIndividual):
         label = g.value(entity, RDFS.label, default=str(entity))
         description = g.value(entity, RDFS.comment, default="No description.")
+        # Se l'etichetta è un URI, estrai il fragment
+        if isinstance(label, URIRef):
+            label = label.split('#')[-1].replace('_', ' ')
+        else:
+            label = str(label)
         # Estrai le proprietà dell'entità
         entity_properties = {}
         for predicate, obj in g.predicate_objects(entity):
@@ -300,7 +305,7 @@ Ecco alcune informazioni rilevanti recuperate dalla base di conoscenza:
 
 ### Regole Stringenti ###
 1) Se l'utente chiede informazioni su questa ontologia, genera SEMPRE una query SPARQL in UNA SOLA RIGA, con prefix:
-   PREFIX base: <http://www.semanticweb.org/lucreziamosca/ontologies/progettoMuseo#>
+   PREFIX base: <http://www.semanticweb.org/lucreziamosca/ontologies/progettomuseo#>
 2) La query SPARQL deve essere precisa e cercare esattamente le entità specificate dall'utente. Ad esempio, se l'utente chiede "Chi ha creato l'opera 'Amore e Psiche'?", la query dovrebbe cercare l'opera esattamente con quel nome.
 3) Se la query produce 0 risultati o fallisce, ritenta con un secondo tentativo.
 4) Se la domanda è generica (tipo 'Ciao, come stai?'), rispondi breve.
@@ -309,7 +314,7 @@ Ecco alcune informazioni rilevanti recuperate dalla base di conoscenza:
 7) Non multiline. Esempio: PREFIX base: <...> SELECT ?x WHERE {{ ... }}.
 Esempio:
 Utente: Chi ha creato l'opera 'Amore e Psiche'?
-Risposta: PREFIX base: <http://www.semanticweb.org/lucreziamosca/ontologies/progettoMuseo#> SELECT ?creatore WHERE {{ ?opera base:hasName "Amore e Psiche" . ?opera base:creatoDa ?creatore . }}
+Risposta: PREFIX base: <http://www.semanticweb.org/lucreziamosca/ontologies/progettomuseo#> SELECT ?creatore WHERE {{ ?opera base:hasName "Amore e Psiche" . ?opera base:creatoDa ?creatore . }}
 FINE REGOLE
 
 ### Conversazione ###
@@ -378,7 +383,16 @@ def load_entity_labels(documents_file: str):
     try:
         with open(documents_file, "r", encoding="utf-8") as f:
             document = json.load(f)
-        entity_labels = [entity['label'].lower() for entity in document['entities']]
+        # Estrai etichette vere e proprie
+        entity_labels = []
+        for entity in document.get('entities', []):
+            label = entity.get('label', '')
+            if label.startswith("http://") or label.startswith("https://"):
+                # Estrai il fragment dell'URI
+                label = label.split('#')[-1].replace('_', ' ')
+            else:
+                label = label.replace('_', ' ')
+            entity_labels.append(label.lower())
         logger.info(f"Elenco delle etichette delle entità caricato: {entity_labels}")
     except Exception as e:
         logger.error(f"Errore nel caricamento delle etichette delle entità: {e}")