| | |
| | import os |
| | import anthropic |
| | import streamlit as st |
| |
|
| | class ClaudeAPIChat: |
| | def __init__(self): |
| | api_key = os.environ.get("ANTHROPIC_API_KEY") |
| | if not api_key: |
| | raise ValueError("No se encontr贸 la clave API de Anthropic. Aseg煤rate de configurarla en las variables de entorno.") |
| | self.client = anthropic.Anthropic(api_key=api_key) |
| | self.conversation_history = [] |
| |
|
| | def generate_response(self, prompt, lang_code): |
| | self.conversation_history.append(f"Human: {prompt}") |
| | full_message = "\n".join(self.conversation_history) |
| | try: |
| | response = self.client.completions.create( |
| | model="claude-2", |
| | prompt=f"{full_message}\n\nAssistant:", |
| | max_tokens_to_sample=300, |
| | temperature=0.7, |
| | stop_sequences=["Human:"] |
| | ) |
| | claude_response = response.completion.strip() |
| | self.conversation_history.append(f"Assistant: {claude_response}") |
| | if len(self.conversation_history) > 10: |
| | self.conversation_history = self.conversation_history[-10:] |
| | return claude_response |
| | except anthropic.APIError as e: |
| | 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 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: |
| | yield "Lo siento, has alcanzado el l铆mite de consultas para esta sesi贸n." |
| | return |
| |
|
| | try: |
| | st.session_state.api_calls += 1 |
| | response = chatbot.generate_response(prompt, lang_code) |
| | |
| | |
| | words = response.split() |
| | |
| | |
| | for word in words: |
| | yield word + " " |
| | except Exception as e: |
| | yield f"Error: {str(e)}" |
