Update modules/chatbot/chat_process.py

modules/chatbot/chat_process.py

@@ -6,6 +6,8 @@ from typing import Dict, Generator
 
 logger = logging.getLogger(__name__)
 
+
+####################################################
 class ChatProcessor:
     def __init__(self):
         """
@@ -17,7 +19,48 @@ class ChatProcessor:
             
         self.client = anthropic.Anthropic(api_key=api_key)
         self.conversation_history = []
+        
+####################################################
+def process_chat_input(self, message: str, lang_code: str) -> Generator[str, None, None]:
+    """Procesa el mensaje y genera una respuesta"""
+    try:
+        # Agregar mensaje a la historia
+        self.conversation_history.append({"role": "user", "content": message})
+        
+        # Generar respuesta usando la API de Claude
+        response = self.client.messages.create(
+            model="claude-3.5-sonnet-20241022",
+            messages=self.conversation_history,
+            temperature=0.7,
+        )
+        
+        # Procesar la respuesta
+        claude_response = response.content[0].text
+        self.conversation_history.append({"role": "assistant", "content": claude_response})
+        
+        # Mantener un historial limitado 
+        if len(self.conversation_history) > 10:
+            self.conversation_history = self.conversation_history[-10:]
+            
+        # Dividir la respuesta en palabras para streaming
+        words = claude_response.split()
+        for word in words:
+            yield word + " "
+            
+    except Exception as e:
+        logger.error(f"Error en process_chat_input: {str(e)}")
+        yield f"Error: {str(e)}"
+
+
+
+
+
+
+
+
 
+################################################################
+'''    
     def process_chat_input(self, message: str, lang_code: str) -> Generator[str, None, None]:
         """
         Procesa el mensaje y genera una respuesta
@@ -52,7 +95,9 @@ class ChatProcessor:
         except Exception as e:
             logger.error(f"Error en process_chat_input: {str(e)}")
             yield f"Error: {str(e)}"
+'''
 
+##########################################    
     def get_conversation_history(self) -> list:
         """
         Retorna el historial de la conversación