Update modules/text_analysis/semantic_analysis.py

modules/text_analysis/semantic_analysis.py

@@ -286,7 +286,7 @@ def create_concept_graph(doc, key_concepts):
 
 ###############################################################################
 
-def visualize_concept_graph(G, lang_code):
+def visualize_concept_graph(G, lang_code, semantic_t):
     """
     Visualiza el grafo de conceptos con layout consistente.
     Args:
@@ -372,23 +372,21 @@ def visualize_concept_graph(G, lang_code):
             ax=ax
         )
         #################################################################
-        # Añadir leyenda de centralidad (usando traducción)
+        # Usar semantic_t para obtener las traducciones
+        plt.title(semantic_t.get('concept_network', 'Concept Network'), 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([])
-        plt.colorbar(sm, ax=ax, label=semantic_t.get('concept_centrality', 'Concept Centrality'))
-        
-        # Título traducido
-        plt.title(semantic_t.get('concept_network', 'Concept Network'), pad=20, fontsize=14)
+        cbar = plt.colorbar(sm, ax=ax)
+        cbar.set_label(semantic_t.get('concept_centrality', 'Concept Centrality'))
 
-        # Ajustar el layout para que la barra de color no se superponga
         ax.set_axis_off()
         plt.tight_layout()
-        
         return fig
-        
     except Exception as e:
         logger.error(f"Error en visualize_concept_graph: {str(e)}")
-        return plt.figure()  # Retornar figura vacía en caso de error
+        return plt.figure()
 
 ########################################################################
 def create_entity_graph(entities):