Changed the rebuild method

app.py

@@ -45,14 +45,17 @@ if torch.cuda.is_available():
     torch.cuda.manual_seed_all(SEED)
 
 # Global Variables
-global G, features, user_nodes, post_nodes, node2idx, pyg_data, trained_model, post_embeddings
-G = features = user_nodes = post_nodes = node2idx = pyg_data = trained_model = post_embeddings = None
+global G, features, user_nodes, post_nodes, node2idx, pyg_data, trained_model, post_embeddings, supabase_client
+G = features = user_nodes = post_nodes = node2idx = pyg_data = trained_model = post_embeddings = supabase_client = None
 
 SUPABASE_URL = os.getenv('supabaseUrl')
 SUPABASE_KEY = os.getenv('supabaseAnonKey')
 
 def get_supabase_client():
-    return create_client(SUPABASE_URL, SUPABASE_KEY)
+    global supabase_client
+    if supabase_client is None:
+        supabase_client = create_client(SUPABASE_URL, SUPABASE_KEY)
+    return supabase_client
 
 async def get_record_count(table):
     supabase = get_supabase_client()
@@ -214,8 +217,8 @@ def get_recommendations(user_id, model, data, G, user_nodes, post_nodes, node2id
         return []
 
     user_idx = node2idx[user_id]
-    user_interacted = set(G[user_id])  # Posts the user has already interacted with
-    post_indices = [node2idx[p] for p in post_nodes if p not in user_interacted]  # Exclude interacted posts
+    user_interacted = set(G[user_id])
+    post_indices = [node2idx[p] for p in post_nodes if p not in user_interacted]
 
     if not post_indices:
         return []
@@ -227,7 +230,7 @@ def get_recommendations(user_id, model, data, G, user_nodes, post_nodes, node2id
 
     idx2node = {idx: node for node, idx in node2idx.items()}
     post_scores = [(idx2node[i], score.item()) for i, score in zip(post_indices, scores)]
-    post_scores = sorted(post_scores, key=lambda x: x[1], reverse=True)  # Sort all posts by score
+    post_scores = sorted(post_scores, key=lambda x: x[1], reverse=True)
 
     recommendations = [{"post_id": post_id, "score": score} for post_id, score in post_scores]
     logger.info(f"Generated {len(recommendations)} recommendations for user {user_id}")
@@ -298,18 +301,27 @@ async def health_check():
     return {"status": "success", "message": "Service operational"}
 
 scheduler = BackgroundScheduler(timezone="America/Los_Angeles")
+def sync_rebuild_model():
+    loop = asyncio.new_event_loop()
+    asyncio.set_event_loop(loop)
+    try:
+        loop.run_until_complete(rebuild_model())
+    finally:
+        loop.close()
+
 scheduler.add_job(
-    lambda: asyncio.run(rebuild_model()),
-    trigger=CronTrigger(hour=3, minute=30),
+    sync_rebuild_model,
+    trigger=CronTrigger(hour=3, minute=00),
     id='daily_model_rebuild',
-    replace_existing=True
+    replace_existing=True,
+    misfire_grace_time=None
 )
 
 @app.on_event("startup")
 async def startup_event():
     await rebuild_model()
     scheduler.start()
-    logger.info("Scheduler started, model will rebuild daily at 3:30 AM Pacific Time")
+    logger.info("Scheduler started, model will rebuild daily at 3:00 AM Pacific Time")
 
 @app.on_event("shutdown")
 async def shutdown_event():