Update app.py

app.py

@@ -18,7 +18,7 @@ from pydantic import BaseModel
 from vonage import Client, Sms
 from datetime import datetime, timedelta
 import schedule
-import threading
+import asyncio
 import time
 import vonage
 
@@ -31,7 +31,7 @@ class MessageRequest(BaseModel):
     text: str
     scheduled_time: str  # HH:MM format
 
-def send_scheduled_sms(message_request: MessageRequest):
+async def send_scheduled_sms(message_request: MessageRequest):
     try:
         response = sms.send_message(
             {
@@ -44,28 +44,29 @@ def send_scheduled_sms(message_request: MessageRequest):
     except Exception as e:
         print(f"Failed to send SMS to {message_request.phone_num}: {e}")
 
-def schedule_daily_sms(message_request: MessageRequest):
+
+async def schedule_sms(message_request: MessageRequest):
     scheduled_time = datetime.strptime(message_request.scheduled_time, "%H:%M")
-    schedule.every().day.at(message_request.scheduled_time).do(send_scheduled_sms, message_request)
-    print(f"Scheduled daily SMS for {message_request.phone_num} at {scheduled_time}")
+    now = datetime.now()
+    delta = scheduled_time - now
+    seconds_until_scheduled_time = delta.total_seconds()
+
+    if seconds_until_scheduled_time > 0:
+        print(f"Scheduling SMS for {message_request.phone_num} at {scheduled_time}")
+        await asyncio.sleep(seconds_until_scheduled_time)
+        await send_scheduled_sms(message_request)
 
-@app.post("/send_daily_sms")
-async def send_daily_sms_endpoint(message_request: MessageRequest):
-    schedule_daily_sms(message_request)
-    return {"message": f"Daily SMS scheduled for {message_request.phone_num} at {message_request.scheduled_time}"}
 
-def run_scheduler():
-    while True:
-        schedule.run_pending()
-        time.sleep(1)
+@app.post("/send_scheduled_sms")
+async def send_scheduled_sms_endpoint(message_request: MessageRequest):
+    asyncio.create_task(schedule_sms(message_request))
+    return {"message": f"SMS scheduled for {message_request.phone_num} at {message_request.scheduled_time}"}
+    
 
 if __name__ == "__main__":
-    scheduler_thread = threading.Thread(target=schedule_daily_sms)
-    scheduler_thread.start()
     import uvicorn
 
     uvicorn.run(app, host="127.0.0.1", port=8000)
 
 
-
 # uvicorn test:app --reload