Update modules/text_analysis/semantic_analysis.py

modules/text_analysis/semantic_analysis.py

@@ -56,6 +56,13 @@ POS_TRANSLATIONS = {
         'NOUN': 'Nom', 'NUM': 'Nombre', 'PART': 'Particule', 'PRON': 'Pronom',
         'PROPN': 'Nom Propre', 'SCONJ': 'Conjonction de Subordination', 'SYM': 'Symbole',
         'VERB': 'Verbe', 'X': 'Autre',
+    },
+    'pt': {
+        'ADJ': 'Adjetivo', 'ADP': 'Preposição', 'ADV': 'Advérbio', 'AUX': 'Auxiliar',
+        'CCONJ': 'Conjunção Coordenativa', 'DET': 'Determinante', 'INTJ': 'Interjeição',
+        'NOUN': 'Substantivo', 'NUM': 'Número', 'PART': 'Partícula', 'PRON': 'Pronome',
+        'PROPN': 'Nome Próprio', 'SCONJ': 'Conjunção Subordinativa', 'SYM': 'Símbolo',
+        'VERB': 'Verbo', 'X': 'Outro',
     }
 }
 
@@ -81,6 +88,13 @@ ENTITY_LABELS = {
         "Inventions": "lightgreen",
         "Dates": "lightyellow",
         "Concepts": "lightpink"
+    },
+    'pt': {
+        "Pessoas": "lightblue",          # Personas/People
+        "Lugares": "lightcoral",          # Lugares/Places
+        "Invenções": "lightgreen",        # Inventos/Inventions
+        "Datas": "lightyellow",           # Fechas/Dates
+        "Conceitos": "lightpink"          # Conceptos/Concepts
     }
 }
 
@@ -373,13 +387,13 @@ def visualize_concept_graph(G, lang_code, semantic_t):
         )
         #################################################################
         # Usar semantic_t para obtener las traducciones
-        plt.title(semantic_t.get('concept_network', 'Relationships between key concepts'), pad=20, fontsize=14)
+        plt.title(semantic_t.get('concept_network', 'Relaciones entre los conceptos clave'), pad=20, fontsize=14)
         
         # Leyenda de centralidad (traducida)
         sm = plt.cm.ScalarMappable(cmap=plt.cm.viridis, norm=plt.Normalize(vmin=0, vmax=1))
         sm.set_array([])
         cbar = plt.colorbar(sm, ax=ax)
-        cbar.set_label(semantic_t.get('concept_centrality', 'Centrality of key concepts'))
+        cbar.set_label(semantic_t.get('concept_centrality', 'Centralidad de los conceptos clave"'))
 
         ax.set_axis_off()
         plt.tight_layout()