Update modules/ui/ui.py

modules/ui/ui.py

@@ -649,35 +649,35 @@ def display_morphosyntax_analysis_interface(nlp_models, lang_code):
         # Mostrar el DataFrame
         st.dataframe(morph_df[columns_to_display])
             
-            # Mostrar diagramas de arco (código existente)
-            with st.expander(t['arc_diagram'], expanded=True):
-                sentences = list(doc.sents)
-                arc_diagrams = []
-                for i, sent in enumerate(sentences):
-                    st.subheader(f"{t['sentence']} {i+1}")
-                    html = displacy.render(sent, style="dep", options={"distance": 100})
-                    html = html.replace('height="375"', 'height="200"')
-                    html = re.sub(r'<svg[^>]*>', lambda m: m.group(0).replace('height="450"', 'height="300"'), html)
-                    html = re.sub(r'<g [^>]*transform="translate\((\d+),(\d+)\)"', lambda m: f'<g transform="translate({m.group(1)},50)"', html)
-                    st.write(html, unsafe_allow_html=True)
-                    arc_diagrams.append(html)
+        # Mostrar diagramas de arco (código existente)
+        with st.expander(t['arc_diagram'], expanded=True):
+            sentences = list(doc.sents)
+            arc_diagrams = []
+            for i, sent in enumerate(sentences):
+                st.subheader(f"{t['sentence']} {i+1}")
+                html = displacy.render(sent, style="dep", options={"distance": 100})
+                html = html.replace('height="375"', 'height="200"')
+                html = re.sub(r'<svg[^>]*>', lambda m: m.group(0).replace('height="450"', 'height="300"'), html)
+                html = re.sub(r'<g [^>]*transform="translate\((\d+),(\d+)\)"', lambda m: f'<g transform="translate({m.group(1)},50)"', html)
+                st.write(html, unsafe_allow_html=True)
+                arc_diagrams.append(html)
             
-            # Guardar resultados
-            if store_morphosyntax_result(
-                st.session_state.username,
-                current_input,
-                word_colors,
-                advanced_analysis['arc_diagram'],
-                advanced_analysis['pos_analysis'],
-                advanced_analysis['morphological_analysis'],
-                advanced_analysis['sentence_structure']
+        # Guardar resultados
+        if store_morphosyntax_result(
+            st.session_state.username,
+            current_input,
+            word_colors,
+            advanced_analysis['arc_diagram'],
+            advanced_analysis['pos_analysis'],
+            advanced_analysis['morphological_analysis'],
+            advanced_analysis['sentence_structure']
                 
-            ):
-                st.success(t['success_message'])
-            else:
-                st.error(t['error_message'])
+        ):
+            st.success(t['success_message'])
         else:
-            st.warning(t['warning_message'])
+            st.error(t['error_message'])
+    else:
+        st.warning(t['warning_message'])
 
 ###############################################################################################################
 def display_semantic_analysis_interface(nlp_models, lang_code):