Update app.py

app.py

@@ -20,6 +20,20 @@ from modules.morpho_analysis import get_repeated_words_colors, highlight_repeate
 from modules.syntax_analysis import visualize_syntax
 
 @st.cache_resource
+def load_chatbot_model():
+    tokenizer = BlenderbotTokenizer.from_pretrained("facebook/blenderbot-400M-distill")
+    model = BlenderbotForConditionalGeneration.from_pretrained("facebook/blenderbot-400M-distill")
+    return tokenizer, model
+
+# Cargar el modelo del chatbot
+chatbot_tokenizer, chatbot_model = load_chatbot_model()
+
+def get_chatbot_response(input_text):
+    inputs = chatbot_tokenizer(input_text, return_tensors="pt")
+    reply_ids = chatbot_model.generate(**inputs)
+    response = chatbot_tokenizer.batch_decode(reply_ids, skip_special_tokens=True)[0]
+    return response
+
 def load_spacy_models():
     return {
         'es': spacy.load("es_core_news_lg"),
@@ -139,6 +153,14 @@ def main_app():
         st.write("Bienvenido, profesor. Aquí podrás ver el progreso de tus estudiantes.")
         # Añade aquí la lógica para mostrar el progreso de los estudiantes
 
+    # Añadir sección de chatbot
+    st.header("Chat con AIdeaText")
+    user_input = st.text_input("Escribe tu mensaje aquí:")
+    if st.button("Enviar"):
+        if user_input:
+            response = get_chatbot_response(user_input)
+            st.text_area("Respuesta del chatbot:", value=response, height=100, max_chars=None, key=None)
+
 def main():
     if 'logged_in' not in st.session_state:
         st.session_state.logged_in = False