Update modules/ui/ui.py

modules/ui/ui.py

@@ -986,6 +986,9 @@ def display_discourse_analysis_interface(nlp_models, lang_code):
         st.info(t['initial_message'])  # Asegúrate de que 'initial_message' esté en tus traducciones
 
 #################################################
+import plotly.graph_objects as go
+import random
+
 def display_discourse_results(result, lang_code, t):
     if result is None:
         st.warning(t.get('no_results', "No hay resultados disponibles."))
@@ -995,10 +998,6 @@ def display_discourse_results(result, lang_code, t):
 
     with col1:
         with st.expander(t.get('file_uploader1', "Documento 1"), expanded=True):
-            if 'graph1' in result:
-                st.pyplot(result['graph1'])
-            else:
-                st.warning(t.get('graph_not_available', "El gráfico no está disponible."))
             st.subheader(t.get('key_concepts', "Conceptos Clave"))
             if 'key_concepts1' in result:
                 df1 = pd.DataFrame(result['key_concepts1'], columns=['Concepto', 'Frecuencia'])
@@ -1006,13 +1005,14 @@ def display_discourse_results(result, lang_code, t):
                 st.table(df1)
             else:
                 st.warning(t.get('concepts_not_available', "Los conceptos clave no están disponibles."))
+            
+            if 'graph1' in result:
+                st.pyplot(result['graph1'])
+            else:
+                st.warning(t.get('graph_not_available', "El gráfico no está disponible."))
 
     with col2:
         with st.expander(t.get('file_uploader2', "Documento 2"), expanded=True):
-            if 'graph2' in result:
-                st.pyplot(result['graph2'])
-            else:
-                st.warning(t.get('graph_not_available', "El gráfico no está disponible."))
             st.subheader(t.get('key_concepts', "Conceptos Clave"))
             if 'key_concepts2' in result:
                 df2 = pd.DataFrame(result['key_concepts2'], columns=['Concepto', 'Frecuencia'])
@@ -1020,6 +1020,11 @@ def display_discourse_results(result, lang_code, t):
                 st.table(df2)
             else:
                 st.warning(t.get('concepts_not_available', "Los conceptos clave no están disponibles."))
+            
+            if 'graph2' in result:
+                st.pyplot(result['graph2'])
+            else:
+                st.warning(t.get('graph_not_available', "El gráfico no está disponible."))
 
     # Relación de conceptos entre ambos documentos (Diagrama de Sankey)
     st.subheader(t.get('comparison', "Relación de conceptos entre ambos documentos"))