AIdeaText commited on
Commit
c03f906
·
verified ·
1 Parent(s): afddf7c

Update modules/text_analysis/semantic_analysis.py

Browse files
modules/text_analysis/semantic_analysis.py CHANGED
@@ -293,6 +293,31 @@ def visualize_concept_graph(G, lang_code):
293
  matplotlib.figure.Figure - Figura del grafo
294
  """
295
  try:
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
296
  # Crear nueva figura con mayor tamaño y definir los ejes explícitamente
297
  fig, ax = plt.subplots(figsize=(15, 10))
298
 
@@ -375,9 +400,9 @@ def visualize_concept_graph(G, lang_code):
375
  norm=plt.Normalize(vmin=0, vmax=1)
376
  )
377
  sm.set_array([])
378
- plt.colorbar(sm, ax=ax, label='Centralidad del concepto')
379
 
380
- plt.title("Red de conceptos relacionados", pad=20, fontsize=14)
381
  ax.set_axis_off()
382
 
383
  # Ajustar el layout para que la barra de color no se superponga
 
293
  matplotlib.figure.Figure - Figura del grafo
294
  """
295
  try:
296
+
297
+ try:
298
+ # 1. Diccionario de traducciones hardcodeado (¡solo para las 2 etiquetas del gráfico!)
299
+ GRAPH_LABELS = {
300
+ 'es': {
301
+ 'concept_network': 'Relaciones entre conceptos clave',
302
+ 'concept_centrality': 'Centralidad de conceptos clave'
303
+ },
304
+ 'en': {
305
+ 'concept_network': 'Relationships between key concepts',
306
+ 'concept_centrality': 'Key Concept Centrality'
307
+ },
308
+ 'fr': {
309
+ 'concept_network': 'Relations entre concepts clés',
310
+ 'concept_centrality': 'Concept clé de centralité'
311
+ },
312
+ 'pt': {
313
+ 'concept_network': 'Relações entre conceitos-chave',
314
+ 'concept_centrality': 'Centralidade do conceito-chave'
315
+ }
316
+ }
317
+
318
+ # 2. Obtener las etiquetas según el idioma (con valores por defecto en español)
319
+ labels = GRAPH_LABELS.get(lang_code, GRAPH_LABELS['es'])
320
+
321
  # Crear nueva figura con mayor tamaño y definir los ejes explícitamente
322
  fig, ax = plt.subplots(figsize=(15, 10))
323
 
 
400
  norm=plt.Normalize(vmin=0, vmax=1)
401
  )
402
  sm.set_array([])
403
+ plt.colorbar(sm, ax=ax, label=labels['concept_centrality'])
404
 
405
+ plt.title("concept_network", pad=20, fontsize=14)
406
  ax.set_axis_off()
407
 
408
  # Ajustar el layout para que la barra de color no se superponga