Update app.py

app.py

@@ -95,22 +95,32 @@ st.set_page_config(
     layout="wide",
     page_icon="random"
 )
-
+#####################################################################################################
 @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
+    try:
+        from transformers import BlenderbotTokenizer, BlenderbotForConditionalGeneration
+        tokenizer = BlenderbotTokenizer.from_pretrained("facebook/blenderbot-400M-distill")
+        model = BlenderbotForConditionalGeneration.from_pretrained("facebook/blenderbot-400M-distill")
+        return tokenizer, model
+    except Exception as e:
+        logger.error(f"Error al cargar el modelo del chatbot: {str(e)}")
+        return None, None
 
 # Load the chatbot model
 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
-
+    if chatbot_tokenizer is None or chatbot_model is None:
+        return "Lo siento, el chatbot no está disponible en este momento."
+    try:
+        inputs = chatbot_tokenizer(input_text, return_tensors="pt")
+        reply_ids = chatbot_model.generate(**inputs)
+        return chatbot_tokenizer.batch_decode(reply_ids, skip_special_tokens=True)[0]
+    except Exception as e:
+        logger.error(f"Error al generar respuesta del chatbot: {str(e)}")
+        return "Lo siento, hubo un error al procesar tu mensaje."
+########################################################################################################
 def load_spacy_models():
     return {
         'es': spacy.load("es_core_news_lg"),