Update modules/chatbot.py

modules/chatbot.py

@@ -36,15 +36,28 @@ class ClaudeAPIChat:
             st.error(f"Error al llamar a la API de Claude: {str(e)}")
             return "Lo siento, hubo un error al procesar tu solicitud."
 
-def initialize_chatbot():
-    return ClaudeAPIChat()
+###################################################################################
+    def initialize_chatbot():
+        return ClaudeAPIChat()
 
-def get_chatbot_response(chatbot, prompt, lang_code):
-    if 'api_calls' not in st.session_state:
-        st.session_state.api_calls = 0
-    
-    if st.session_state.api_calls >= 50:  # Límite de 50 llamadas por sesión
-        return "Lo siento, has alcanzado el límite de consultas para esta sesión."
+#########################################################################################
+    def get_chatbot_response(chatbot, prompt, lang_code):
+        if 'api_calls' not in st.session_state:
+            st.session_state.api_calls = 0
+        
+        if st.session_state.api_calls >= 50:  # Límite de 50 llamadas por sesión
+            yield "Lo siento, has alcanzado el límite de consultas para esta sesión."
+            return
     
-    st.session_state.api_calls += 1
-    return chatbot.generate_response(prompt, lang_code)
+        try:
+            st.session_state.api_calls += 1
+            response = chatbot.generate_response(prompt, lang_code)
+            
+            # Dividir la respuesta en palabras
+            words = response.split()
+            
+            # Devolver las palabras una por una
+            for word in words:
+                yield word + " "
+        except Exception as e:
+            yield f"Error: {str(e)}"