Update modules/ui.py

modules/ui.py

@@ -13,11 +13,11 @@ from spacy import displacy
 from .auth import authenticate_user, register_user, get_user_role
 from .database import get_student_data, store_morphosyntax_result, store_semantic_result
 
-#Importacoines locales funciones de análisis
+#Importaciones locales funciones de análisis
 from .morpho_analysis import generate_arc_diagram, get_repeated_words_colors, highlight_repeated_words, POS_COLORS, POS_TRANSLATIONS
 from .semantic_analysis import visualize_semantic_relations, perform_semantic_analysis
 from .discourse_analysis import compare_semantic_analysis, perform_discourse_analysis
-#from .discourse_analysis import discourse_graph
+from .chatbot import initialize_chatbot, get_chatbot_response
 
 ##################################################################################################
 def login_register_page():
@@ -392,7 +392,45 @@ def display_discourse_analysis_interface(nlp_models, lang_code):
                 st.pyplot(graph2)
 
 ##################################################################################################
-def get_chatbot_response(input_text):
-    # Esta función debe ser implementada o importada de otro módulo
-    # Por ahora, retornamos un mensaje genérico
-    return "Lo siento, el chatbot no está disponible en este momento."
+def display_chatbot_interface(lang_code):
+    translations = {
+        'es': {
+            'title': "AIdeaText - Chatbot Llama 2",
+            'input_placeholder': "Escribe tu mensaje aquí...",
+            'send_button': "Enviar",
+        },
+        'en': {
+            'title': "AIdeaText - Llama 2 Chatbot",
+            'input_placeholder': "Type your message here...",
+            'send_button': "Send",
+        },
+        'fr': {
+            'title': "AIdeaText - Chatbot Llama 2",
+            'input_placeholder': "Écrivez votre message ici...",
+            'send_button': "Envoyer",
+        }
+    }
+
+    t = translations[lang_code]
+
+    st.header(t['title'])
+
+    if 'chatbot' not in st.session_state:
+        st.session_state.chatbot = initialize_chatbot()
+
+    if 'messages' not in st.session_state:
+        st.session_state.messages = []
+
+    for message in st.session_state.messages:
+        with st.chat_message(message["role"]):
+            st.markdown(message["content"])
+
+    if prompt := st.chat_input(t['input_placeholder']):
+        st.session_state.messages.append({"role": "user", "content": prompt})
+        with st.chat_message("user"):
+            st.markdown(prompt)
+
+        with st.chat_message("assistant"):
+            response = get_chatbot_response(st.session_state.chatbot, prompt)
+            st.markdown(response)
+        st.session_state.messages.append({"role": "assistant", "content": response})