Spaces:
Build error
Build error
| from app.services.download_media import download_whatsapp_media | |
| from app.services.message import process_message_with_llm | |
| from app.services.cache import MessageCache | |
| import logging | |
| message_cache = MessageCache() | |
| logger = logging.getLogger(__name__) | |
| async def handle_message(message, user_chats, message_cache, access_token): | |
| try: | |
| # Process the message content | |
| content = message.get("text", {}).get("body") | |
| message_id = message.get("id") | |
| image = message.get("image") | |
| document = message.get("document") | |
| video = message.get("video") | |
| sender_id = message.get("from") | |
| logger.info(f"Content: {content}") | |
| logger.info(f"sender_id: {sender_id}") | |
| # Download media if exists | |
| if image: | |
| file_path = f"{image.get('id')}.jpg" | |
| logger.info(f"file_path: {file_path}") | |
| image_file_path = await download_whatsapp_media(image.get("id"), access_token=access_token, file_path=file_path) | |
| logger.info(f"image file_path: {image_file_path}") | |
| else: | |
| image_file_path = None | |
| if document: | |
| filename = message.get("document", {}).get("filename") | |
| logger.info(f"file_path: {filename}") | |
| document_file_path = await download_whatsapp_media(document.get("id"), access_token=access_token, file_path=filename) | |
| logger.info(f"document file_path: {document_file_path}") | |
| else: | |
| document_file_path = None | |
| if video: | |
| video_type = message.get("video", {}).get("mime_type") | |
| video_filename= f"{video.get('id')}.{video_type.split('/')[-1]}" | |
| logger.info(f"video_type: {video_type}") | |
| video_file_path = await download_whatsapp_media(video.get("id"), access_token=access_token, file_path=video_filename) | |
| else: | |
| video_file_path = None | |
| # Check for duplicates | |
| if message_cache.exists(message_id): | |
| logger.info(f"Duplicate message detected and skipped: {message_id}") | |
| return {"status": "duplicate", "message_id": message_id} | |
| # Initialize chat history | |
| if sender_id not in user_chats: | |
| user_chats[sender_id] = [] | |
| # Append content to user chat | |
| if user_chats[sender_id] == []: | |
| user_chats[sender_id].append({"role": "user", "parts": "This is the chat history so far"}) | |
| logger.info(f"user_chats: {user_chats}") | |
| # Process the message | |
| result = await process_message_with_llm(sender_id, content, user_chats[sender_id], image_file_path = image_file_path, doc_path = document_file_path) | |
| if content != None: | |
| user_chats[sender_id].append({"role": "user", "parts": content}) | |
| user_chats[sender_id].append({"role": "model", "parts": result}) | |
| message_cache.add(message_id) | |
| return {"status": "success", "message_id": message_id, "result": result} | |
| except Exception as e: | |
| return {"status": "error", "message_id": message.get("id"), "error": str(e)} | |