Update modules/ui.py

modules/ui.py

@@ -334,6 +334,55 @@ def display_semantic_analysis_interface(nlp_models, lang_code):
             with st.expander(t['semantic_relations'], expanded=True):
                 st.pyplot(relations_graph)
 
+##################################################################################################
+
+def display_discourse_analysis_interface(nlp_models, lang_code):
+    translations = {
+        'es': {
+            'title': "AIdeaText - Análisis del discurso",
+            'file_uploader1': "Cargar archivo de texto 1 (Patrón)",
+            'file_uploader2': "Cargar archivo de texto 2 (Comparación)",
+            'analyze_button': "Analizar textos",
+            'comparison': "Comparación de Relaciones Semánticas",
+        },
+        'en': {
+            'title': "AIdeaText - Discourse Analysis",
+            'file_uploader1': "Upload text file 1 (Pattern)",
+            'file_uploader2': "Upload text file 2 (Comparison)",
+            'analyze_button': "Analyze texts",
+            'comparison': "Comparison of Semantic Relations",
+        },
+        'fr': {
+            'title': "AIdeaText - Analyse du discours",
+            'file_uploader1': "Télécharger le fichier texte 1 (Modèle)",
+            'file_uploader2': "Télécharger le fichier texte 2 (Comparaison)",
+            'analyze_button': "Analyser les textes",
+            'comparison': "Comparaison des Relations Sémantiques",
+        }
+    }
+
+    t = translations[lang_code]
+
+    st.header(t['title'])
+
+    col1, col2 = st.columns(2)
+
+    with col1:
+        uploaded_file1 = st.file_uploader(t['file_uploader1'], type=['txt'])
+
+    with col2:
+        uploaded_file2 = st.file_uploader(t['file_uploader2'], type=['txt'])
+
+    if uploaded_file1 is not None and uploaded_file2 is not None:
+        text_content1 = uploaded_file1.getvalue().decode('utf-8')
+        text_content2 = uploaded_file2.getvalue().decode('utf-8')
+
+        if st.button(t['analyze_button']):
+            comparison_graph = perform_discourse_analysis(text_content1, text_content2, nlp_models[lang_code], lang_code)
+
+            with st.expander(t['comparison'], expanded=True):
+                st.pyplot(comparison_graph)
+
 ##################################################################################################
 def get_chatbot_response(input_text):
     # Esta función debe ser implementada o importada de otro módulo