Update modules/text_analysis/discourse_analysis.py

modules/text_analysis/discourse_analysis.py

@@ -108,62 +108,81 @@ def fig_to_bytes(fig, dpi=100):
         logger.error(f"Error en fig_to_bytes: {str(e)}")
         return None
         
-#################
+################################################################################################
 def compare_semantic_analysis(text1, text2, nlp, lang):
     """
     Realiza el análisis semántico comparativo entre dos textos
     """
     try:
+        # Diccionario de traducciones para los títulos de los gráficos COMPARATIVOS
+        COMPARE_GRAPH_TITLES = {
+            'es': {
+                'doc1_network': 'Relaciones entre conceptos clave del documento 1',
+                'doc1_centrality': 'Centralidad de los conceptos clave del documento 1',
+                'doc2_network': 'Relaciones entre conceptos clave del documento 2',
+                'doc2_centrality': 'Centralidad de los conceptos clave del documento 2'
+            },
+            'en': {
+                'doc1_network': 'Key concept relationships in document 1',
+                'doc1_centrality': 'Key concept centrality in document 1',
+                'doc2_network': 'Key concept relationships in document 2',
+                'doc2_centrality': 'Key concept centrality in document 2'
+            },
+            'fr': {
+                'doc1_network': 'Relations entre concepts clés du document 1',
+                'doc1_centrality': 'Centralité des concepts clés du document 1',
+                'doc2_network': 'Relations entre concepts clés du document 2',
+                'doc2_centrality': 'Centralité des concepts clés du document 2'
+            },
+            'pt': {
+                'doc1_network': 'Relações entre conceitos-chave do documento 1',
+                'doc1_centrality': 'Centralidade dos conceitos-chave do documento 1',
+                'doc2_network': 'Relações entre conceitos-chave do documento 2',
+                'doc2_centrality': 'Centralidade dos conceitos-chave do documento 2'
+            }
+        }
+
+        # Obtener traducciones (inglés por defecto)
+        titles = COMPARE_GRAPH_TITLES.get(lang, COMPARE_GRAPH_TITLES['en'])
+        
         logger.info(f"Iniciando análisis comparativo para idioma: {lang}")
         
-        # Obtener stopwords
+        # Resto del código permanece exactamente igual...
         stopwords = get_custom_stopwords(lang)
         logger.info(f"Obtenidas {len(stopwords)} stopwords para el idioma {lang}")
         
-        # Procesar los textos
         doc1 = nlp(text1)
         doc2 = nlp(text2)
         
-        # Identificar conceptos clave
-        logger.info("Identificando conceptos clave del primer texto...")
         key_concepts1 = identify_key_concepts(doc1, stopwords=stopwords, min_freq=2, min_length=3)
-        
-        logger.info("Identificando conceptos clave del segundo texto...")
         key_concepts2 = identify_key_concepts(doc2, stopwords=stopwords, min_freq=2, min_length=3)
 
         if not key_concepts1 or not key_concepts2:
             raise ValueError("No se pudieron identificar conceptos clave en uno o ambos textos")
 
-        # Crear grafos
-        logger.info("Creando grafos de conceptos...")
         G1 = create_concept_graph(doc1, key_concepts1)
         G2 = create_concept_graph(doc2, key_concepts2)
-
-        # Visualizar grafos
-        logger.info("Visualizando grafos...")
         
-        # Primer grafo
+        # Primer grafo con título traducido
         plt.figure(figsize=(12, 8))
         fig1 = visualize_concept_graph(G1, lang)
-        plt.title("Análisis del primer texto", pad=20)
+        plt.title(titles['doc1_network'], pad=20)
         plt.tight_layout()
         
-        # Segundo grafo
+        # Segundo grafo con título traducido
         plt.figure(figsize=(12, 8))
         fig2 = visualize_concept_graph(G2, lang)
-        plt.title("Análisis del segundo texto", pad=20)
+        plt.title(titles['doc2_network'], pad=20)
         plt.tight_layout()
 
-        logger.info("Análisis comparativo completado exitosamente")
         return fig1, fig2, key_concepts1, key_concepts2
 
     except Exception as e:
         logger.error(f"Error en compare_semantic_analysis: {str(e)}")
-        plt.close('all')  # Limpiar recursos en caso de error
+        plt.close('all')
         raise
     finally:
-        plt.close('all')  # Asegurar limpieza en todos los casos
-
+        plt.close('all')
 
 ############################################
 def create_concept_table(key_concepts):