Spaces:
Build error
Build error
File size: 3,098 Bytes
07fbc67 f169c98 07fbc67 f169c98 07fbc67 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 |
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)}
|