Upload 20 files

app/services/agent_service.py

@@ -5,4 +5,4 @@ def verify_agent_by_agent_id_and_broker_name(agent_id, broker_name):
         return True
     elif agent_id == "king_agent" and broker_name == "king_broker":
         return True
-    return False
+    return False

app/services/realtor_service.py

@@ -1,8 +1,10 @@
 from app.db import crud
+import requests, os, json, uuid
+from app.utils.log_utils import setup_logger
+from sqlalchemy.orm.attributes import flag_modified
+from app.task_scheduler.task import schedule_n8n_workflow
 from app.services.agent_service import verify_agent_by_agent_id_and_broker_name
 from app.services.properties_service import get_property_by_address_or_mls_number
-from app.utils.log_utils import setup_logger
-import requests, os, json, uuid
 from dotenv import load_dotenv
 load_dotenv()
 
@@ -13,7 +15,7 @@ n8n_webhook_url = os.getenv("N8N_WEBHOOK_URL")
 
 logger = setup_logger("realtor_service")
 
-def create_payload_for_whatsapp_message(from_number, to_number, body):
+def create_payload_for_whatsapp_message(from_number, body):
     payload = [
         {
             "SmsMessageSid": "SM0b07afa5f1ede39427957b4dc127cab3",
@@ -24,7 +26,7 @@ def create_payload_for_whatsapp_message(from_number, to_number, body):
             "WaId": from_number,
             "SmsStatus": "received",
             "Body": f"paparaphrase this: {body}",
-            "To": f"whatsapp:{to_number}",
+            "To": f"whatsapp:{ai_agnet_number}",
             "NumSegments": "1",
             "ReferralNumMedia": "0",
             "MessageSid": "SM0b07afa5f1ede39427957b4dc127cab3",
@@ -78,6 +80,7 @@ class RealtorService:
                     if booking:
                         booking.buyer_selected_date = buyer_selected_date
                         booking.buyer_selected_time = buyer_selected_time
+                        booking.status = "pending"
                         db.commit()
                         db.refresh(booking)
                         return f"A booking already exists with this MLS number. The rescheduling request has been successfully processed. Your booking ID is {booking.id} for future reference."
@@ -86,6 +89,7 @@ class RealtorService:
                                                         address=booking_address, mls_number=mls_number, buyer_selected_date=buyer_selected_date,
                                                         buyer_selected_time=buyer_selected_time, listing_agent_phone_number=listing_agent_phone_number,
                                                         listing_agent_session_id=new_session_id)
+                        return f"Do not say like its booked successfully say like: Your showing request is awaiting confirmation from the listing agent, {listing_user.full_name}. I will notify you once the listing agent confirms the showing appointment. Your booking ID is #{booking.id} for future reference."
                 else:
                     logger.info(f"The user already exists but no active session found : {buyer_agent_phone_number}")
                     listing_user.active_session_id = new_session_id
@@ -125,18 +129,18 @@ class RealtorService:
 
         listing_agent_body = f"Hello {listing_agent_full_name}, this is Alice assistant of {buyer_user.full_name}. We would like to schedule a showing for {booking_address} (MLS # {mls_number}) at {buyer_selected_time} on {buyer_selected_date}. The booking ID for this request is #{booking.id} for future reference. Can you make that work?."
         logger.info(f"Sending message to listing agent from: {ai_agnet_number}, body: {listing_agent_body}, to: {listing_agent_phone_number}")
-        response = create_payload_for_whatsapp_message(from_number=listing_agent_phone_number, to_number=ai_agnet_number, body=listing_agent_body)
+        response = create_payload_for_whatsapp_message(from_number=listing_agent_phone_number, body=listing_agent_body)
         if response == "Message sent successfully":
-            return f"Do not say like its booked successfully say like: Your showing has been pending with the listing agent {listing_user.full_name}. I will notify you once the listing agent confirms the showing appointment. Your booking ID is #{booking.id} for future reference."
+            return f"Do not say like its booked successfully say like: Your showing request is awaiting confirmation from the listing agent, {listing_user.full_name}. I will notify you once the listing agent confirms the showing appointment. Your booking ID is #{booking.id} for future reference."
         return "Error in sending message to listing agent"
 
 
     def listing_realtor_confirmation(self, db, session_id, listing_selected_date, listing_selected_time, confirmation):
+        is_listing_user_session_queued = False
         logger.warning(f"Listing realtore - listing_selected_date: {listing_selected_date}, listing_selected_time: {listing_selected_time}, session_id: {session_id}, confirmation: {confirmation}.")
         try:
             booking = crud.get_booking_by_session_id(db, session_id=session_id)
             listing_user = crud.get_user_by_phone_number(db, phone_number=booking.listing_agent_phone_number)
-            is_listing_user_session_queued = False
         except Exception as e:
             logger.error(f"Error in creating user: {e}")
             return f"booking not found with this session id, Error: {str(e)}"
@@ -154,22 +158,28 @@ class RealtorService:
             listing_selected_time = f"{listing_selected_time} AM"
         if confirmation == "confirmed":
             logger.warning(f"booking Schudule is confirmed by listing agent.")
-            booking.status = "confirmed"
             if listing_user.queued_session_ids:
-                logger.info(f"listing user has queued sessions Found.")
+                logger.info(f"listing user has queued sessions Found. active session id: {listing_user.active_session_id}, queued sessions: {listing_user.queued_session_ids}")
+                listing_user.completed_session_ids = listing_user.completed_session_ids + [listing_user.active_session_id]
                 listing_user.active_session_id = listing_user.queued_session_ids.pop(0)
+                flag_modified(listing_user, "queued_session_ids")
                 is_listing_user_session_queued = True
-            listing_agent_body = f"Your showing at {booking.address} (MLS # {booking.mls_number}) is confirmed on {listing_selected_date} at {listing_selected_time} . I'll send you a reminder the day before to ensure you're prepared."
-            buyer_agent_body = f"Your showing at {booking.address} (MLS # {booking.mls_number}) is confirmed by {listing_user.full_name} on {listing_selected_date} at {listing_selected_time} . I'll send you a reminder the day before to ensure you're prepared."
+            booking.status = "confirmed"
+            listing_agent_body = f"Your showing at {booking.address} (MLS # {booking.mls_number}) is confirmed on {listing_selected_date} at {listing_selected_time}. I'll send you a reminder the day before to ensure you're prepared. The booking ID for this request is #{booking.id} for future reference."
+            buyer_agent_body = f"Your showing at {booking.address} (MLS # {booking.mls_number}) is confirmed by {listing_user.full_name} on {listing_selected_date} at {listing_selected_time} . I'll send you a reminder the day before to ensure you're prepared. The booking ID for this request is #{booking.id} for future reference."
+
         elif confirmation == "cancelled":
             logger.warning(f"booking Schudule is cancelled by listing agent.")
             if listing_user.queued_session_ids:
-                logger.info(f"listing user has queued sessions Found.")
+                logger.info(f"listing user has queued sessions Found. active session id: {listing_user.active_session_id}, queued sessions: {listing_user.queued_session_ids}")
+                listing_user.completed_session_ids = listing_user.completed_session_ids + [listing_user.active_session_id]
                 listing_user.active_session_id = listing_user.queued_session_ids.pop(0)
+                flag_modified(listing_user, "queued_session_ids")
                 is_listing_user_session_queued = True
             booking.status = "cancelled"
-            listing_agent_body = f"Thank you for letting us know. We will inform the buyer that the showing has been cancelled."
-            buyer_agent_body = f"Unfortunately, the listing agent has cancelled the showing. We apologize for any inconvenience caused."
+            listing_agent_body = f"Thank you for letting us know. I'll inform the buyer that the showing has been canceled. The booking ID for this request is #{booking.id} for future reference."
+            buyer_agent_body = f"Unfortunately, the listing agent, {listing_user.full_name} has canceled the showing for the property at {booking.address} (MLS # {booking.mls_number}). We apologize for any inconvenience. Your booking ID is #{booking.id} for future reference."
+
         elif confirmation == "rescheduled":
             logger.warning(f"booking Schudule is rescheduled by listing agent. because of listing_selected_date: {listing_selected_date}, listing_selected_time: {listing_selected_time} and buyer_selected_date: {booking.buyer_selected_date}, buyer_selected_time: {booking.buyer_selected_time}")
             booking.status = "rescheduled"
@@ -179,8 +189,18 @@ class RealtorService:
             logger.warning(f"Invalid confirmation status")
             return "Invalid confirmation status"
 
-        logger.info(f"Sending message to listing agent from: {ai_agnet_number}, body: {buyer_agent_body}, to: {booking.buyer_agent_phone_number}")
-        response = create_payload_for_whatsapp_message(from_number=booking.buyer_agent_phone_number, to_number=ai_agnet_number, body=buyer_agent_body)
+        if is_listing_user_session_queued:
+            countdown_second=60
+            queued_booking = crud.get_booking_by_session_id(db, listing_user.active_session_id)
+            logger.info(f"Queued sessions id: {queued_booking.listing_agent_session_id}")
+            buyer_user = queued_booking.buyer_agent
+            listing_agent_body = f"Hello {listing_user.full_name}, this is Alice assistant of {buyer_user.full_name}. We would like to schedule a showing for {queued_booking.address} (MLS # {queued_booking.mls_number}) at {queued_booking.buyer_selected_time} on {queued_booking.buyer_selected_date}. The booking ID for this request is #{queued_booking.id} for future reference. Can you make that work?."
+            logger.info(f"Sending message to listing agent, body: {listing_agent_body}, to: {listing_user.phone_number}")
+            result = schedule_n8n_workflow.apply_async(args = [listing_user.phone_number, listing_agent_body], countdown=countdown_second)
+            logger.info(f"Task scheduled to run in {countdown_second} seconds.")
+
+        logger.info(f"Sending message to listing agent, body: {buyer_agent_body}, to: {booking.buyer_agent_phone_number}")
+        response = create_payload_for_whatsapp_message(from_number=booking.buyer_agent_phone_number, body=buyer_agent_body)
         if response != "Message sent successfully":
             logger.error(f"Error in sending message to listing agent")
             return "Error in sending message to buyer agent"
@@ -188,6 +208,7 @@ class RealtorService:
         return listing_agent_body
 
     def buyer_realtor_confirmation(self, db, buyer_agent_phone_number, booking_id, mls_number, buyer_selected_date, buyer_selected_time, confirmation):
+        is_listing_user_session_queued = False
         logger.warning(f"Buyer realtore - booking_id: {booking_id}, mls_number: {mls_number}, buyer_selected_date: {buyer_selected_date}, buyer_selected_time: {buyer_selected_time}, confirmation: {confirmation}.")
         try:
             if booking_id:
@@ -196,6 +217,7 @@ class RealtorService:
                 booking = crud.get_booking_by_mls_number_and_buyer_agent_phone_number(db, mls_number=mls_number, buyer_agent_phone_number=buyer_agent_phone_number)
             else:
                 return "Invalid booking id or mls number"
+            listing_user = crud.get_user_by_phone_number(db, phone_number=booking.listing_agent_phone_number)
         except Exception as e:
             logger.error(f"Error in creating user: {e}")
             return f"booking not found with this booking id, Error: {str(e)}"
@@ -212,25 +234,50 @@ class RealtorService:
             buyer_selected_time = f"{buyer_selected_time} AM"
         if confirmation == "confirmed":
             logger.warning(f"booking Schudule is confirmed by buyer agent.")
+            if listing_user.queued_session_ids:
+                logger.info(f"listing user has queued sessions Found. active session id: {listing_user.active_session_id}, queued sessions: {listing_user.queued_session_ids}")
+                listing_user.completed_session_ids = listing_user.completed_session_ids + [listing_user.active_session_id]
+                listing_user.active_session_id = listing_user.queued_session_ids.pop(0)
+                flag_modified(listing_user, "queued_session_ids")
+                is_listing_user_session_queued = True
             booking.status = "confirmed"
             buyer_agent_body = f"Your showing at {booking.address} (MLS # {booking.mls_number}) is confirmed on {buyer_selected_date} at {buyer_selected_time}. Your booking ID is {booking.id} for future reference. I'll send you a reminder the day before to ensure you're prepared."
             listing_agent_body = f"Your showing at {booking.address} (MLS # {booking.mls_number}) is confirmed on {buyer_selected_date} at {buyer_selected_time} . Your booking ID is {booking.id} for future reference. I'll send you a reminder the day before to ensure you're prepared."
+
         elif confirmation == "cancelled":
             logger.warning(f"booking Schudule is cancelled by buyer agent.")
+            if listing_user.queued_session_ids:
+                logger.info(f"listing user has queued sessions Found. active session id: {listing_user.active_session_id}, queued sessions: {listing_user.queued_session_ids}")
+                listing_user.completed_session_ids = listing_user.completed_session_ids + [listing_user.active_session_id]
+                listing_user.active_session_id = listing_user.queued_session_ids.pop(0)
+                flag_modified(listing_user, "queued_session_ids")
+                is_listing_user_session_queued = True
             booking.status = "cancelled"
-            buyer_agent_body = f"Thank you for letting us know. We will inform the listing Agent that the showing has been cancelled."
-            listing_agent_body = f"Unfortunately, the Buyer agent has cancelled the showing. We apologize for any inconvenience caused."
+            buyer_agent_body = f"Thank you for letting us know. I'll inform the listing agent that the showing has been canceled. The booking ID for this request is #{booking.id} for future reference."
+            listing_agent_body = f"Unfortunately, the Buyer agent, {booking.buyer_agent.full_name} has canceled the showing for the property at {booking.address} (MLS # {booking.mls_number}). We apologize for any inconvenience. Your booking ID is #{booking.id} for future reference."
+
         elif confirmation == "rescheduled":
             logger.warning(f"booking Schudule is rescheduled by Buyer agent. because of listing_selected_date: {buyer_selected_date}, listing_selected_time: {buyer_selected_time} and buyer_selected_date: {booking.buyer_selected_date}, buyer_selected_time: {booking.buyer_selected_time}")
             booking.status = "rescheduled"
             buyer_agent_body = f"paparaphrase this: Your reschedule request for the showing has been submitted and is waiting for confirmation from the buyer agent. I’ll update you once they confirm the showing appointment. Your booking ID is #{booking.id} for reference."
             listing_agent_body = f"The Buyer agent for the booking at {booking.address} (MLS # {booking.mls_number}) wants to rescheduled the showing on {buyer_selected_date} at {buyer_selected_time}. Your booking ID is #{booking.id} for future reference. Please confirm if this new time works for you."
+
         else:
             logger.warning(f"Invalid confirmation status")
             return "Invalid confirmation status"
 
+        if is_listing_user_session_queued:
+            countdown_second=60
+            queued_booking = crud.get_booking_by_session_id(db, listing_user.active_session_id)
+            logger.info(f"Queued sessions id: {queued_booking.listing_agent_session_id}")
+            buyer_user = queued_booking.buyer_agent
+            listing_agent_body = f"Hello {listing_user.full_name}, this is Alice assistant of {buyer_user.full_name}. We would like to schedule a showing for {queued_booking.address} (MLS # {queued_booking.mls_number}) at {queued_booking.buyer_selected_time} on {queued_booking.buyer_selected_date}. The booking ID for this request is #{queued_booking.id} for future reference. Can you make that work?."
+            logger.info(f"Sending message to listing agent, body: {listing_agent_body}, to: {listing_user.phone_number}")
+            result = schedule_n8n_workflow.apply_async(args = [listing_user.phone_number, listing_agent_body], countdown=countdown_second)
+            logger.info(f"Task scheduled to run in {countdown_second} seconds.")
+
         logger.info(f"Sending message to listing agent from: {ai_agnet_number}, body: {listing_agent_body}, to: {booking.listing_agent_phone_number}")
-        response = create_payload_for_whatsapp_message(from_number=booking.listing_agent_phone_number, to_number=ai_agnet_number, body=listing_agent_body)
+        response = create_payload_for_whatsapp_message(from_number=booking.listing_agent_phone_number, body=listing_agent_body)
         if response != "Message sent successfully":
             logger.error(f"Error in sending message to listing agent")
             return "Error in sending message to Listing agent"

app/task_scheduler/task.py

@@ -0,0 +1,47 @@
+import requests, os, json
+from celery import Celery
+from app.utils.log_utils import setup_logger
+
+logger = setup_logger("task")
+
+ai_agnet_number = os.getenv("AI_AGENT_NUMBER")
+n8n_webhook_url = os.getenv("N8N_WEBHOOK_URL")
+
+app = Celery(
+    "tasks",
+    broker="redis://localhost:6379/0",
+    backend="redis://localhost:6379/0",
+)
+app.conf.broker_connection_retry_on_startup = True
+
+@app.task
+def schedule_n8n_workflow(from_number, body):
+    payload = [
+        {
+            "SmsMessageSid": "SM0b07afa5f1ede39427957b4dc127cab3",
+            "NumMedia": "0",
+            "ProfileName": "",
+            "MessageType": "text",
+            "SmsSid": "SM0b07afa5f1ede39427957b4dc127cab3",
+            "WaId": from_number,
+            "SmsStatus": "received",
+            "Body": f"paparaphrase this: {body}",
+            "To": f"whatsapp:{ai_agnet_number}",
+            "NumSegments": "1",
+            "ReferralNumMedia": "0",
+            "MessageSid": "SM0b07afa5f1ede39427957b4dc127cab3",
+            "AccountSid": "",
+            "From": f"whatsapp:{from_number}",
+            "ApiVersion": "2010-04-01"
+        }
+    ]
+
+    try:
+        logger.info(f"Payload for whatsapp webhook: {payload}")
+        headers = {"Content-Type": "application/json"}
+        response = requests.post(n8n_webhook_url, headers=headers, data=json.dumps(payload))
+        logger.info(f"Response for whatsapp webhook: {response.text}")
+    except Exception as e:
+        logger.error(f"Error in sending message to agent: {e}")
+        return f"Error in sending message to agent: {str(e)}"
+    return "Message sent successfully"