Spaces:
Build error
Build error
| from typing import Dict, List, Optional, Any | |
| from app.services.cache import MessageCache | |
| from app.services.chat_manager import ChatManager | |
| from app.handlers.media_handler import MediaHandler | |
| from app.services.message_parser import MessageParser | |
| from app.services.download_media import download_whatsapp_media | |
| from app.services.message import process_message_with_llm | |
| from app.models.message_types import Message, MediaType, MediaContent | |
| from app.memory import AgentMemory | |
| import logging | |
| logger = logging.getLogger(__name__) | |
| class MessageHandler: | |
| def __init__( | |
| self, | |
| message_cache: MessageCache, | |
| chat_manager: ChatManager, | |
| media_handler: MediaHandler, | |
| logger: logging.Logger | |
| ): | |
| self.message_cache = message_cache | |
| self.chat_manager = chat_manager | |
| self.media_handler = media_handler | |
| self.logger = logger | |
| async def handle(self, raw_message: dict, whatsapp_token: str, whatsapp_url:str,gemini_api:str, rag_system:Any = None, agentMemory:Any=None, memory:Any=None) -> dict: | |
| try: | |
| # Parse message | |
| message = MessageParser.parse(raw_message) | |
| self.logger.info(f"Parsed Message: {message}") | |
| if self.message_cache.exists(message.id): | |
| self.logger.info(f"Duplicate message detected and skipped: {message.id}") | |
| return {"status": "duplicate", "message_id": message.id} | |
| # Download media | |
| media_paths = await self._process_media(message, whatsapp_token) | |
| # Simple class to store chat temporarily | |
| # self.chat_manager.initialize_chat(message.sender_id) | |
| user = await agentMemory.add_user(message.sender_id, message.sender_id) | |
| await memory.add_message(message.sender_id, "user", message.content) | |
| history = await memory.get_history(message.sender_id, last_n = 2) | |
| print(f"chat_history: {history }") | |
| # Process message with LLM | |
| result = await process_message_with_llm( | |
| message.sender_id, | |
| message.content, | |
| # self.chat_manager.get_chat_history(message.sender_id), | |
| history, | |
| rag_system = rag_system, | |
| agentMemory=agentMemory, | |
| memory = memory, | |
| whatsapp_token=whatsapp_token, | |
| whatsapp_url=whatsapp_url, | |
| **media_paths | |
| ) | |
| self.logger.info(f"Result: {result}") | |
| # Append message to chat to keep track of conversation | |
| # self.chat_manager.append_message(message.sender_id, "user", message.content) | |
| # self.chat_manager.append_message(message.sender_id, "model", result) | |
| await memory.add_message(message.sender_id, "model", result) | |
| self.message_cache.add(message.id) | |
| return {"status": "success", "message_id": message.id, "result": result} | |
| except Exception as e: | |
| self.logger.error(f"Error processing message: {str(e)}") | |
| return {"status": "error", "message_id": raw_message.get("id"), "error": str(e)} | |
| async def _process_media(self, message: Message, whatsapp_token: str) -> Dict[str, Optional[str]]: | |
| media_paths = { | |
| "image_file_path": None, | |
| "doc_path": None, | |
| "video_file_path": None | |
| } | |
| if not message.media: | |
| return media_paths | |
| for media_type, content in message.media.items(): | |
| self.logger.info(f"Processing {media_type.value}: {content.file_path}") | |
| file_path = await self.media_handler.download( | |
| content.id, | |
| whatsapp_token, | |
| content.file_path | |
| ) | |
| self.logger.info(f"{media_type.value} file_path: {file_path}") | |
| if media_type == MediaType.IMAGE: | |
| media_paths["image_file_path"] = file_path | |
| elif media_type == MediaType.DOCUMENT: | |
| media_paths["doc_path"] = file_path | |
| elif media_type == MediaType.VIDEO: | |
| media_paths["video_file_path"] = file_path | |
| return media_paths | |