From 226ba475aa67ed93dde1f7a83a62a46843510890 Mon Sep 17 00:00:00 2001
From: Azalea <22280294+hykilpikonna@users.noreply.github.com>
Date: Sun, 9 Mar 2025 10:19:39 -0400
Subject: [PATCH] [+] Verse: Generalize AI recommender

---
 tools/recommend.py | 16 ++++++++++++----
 1 file changed, 12 insertions(+), 4 deletions(-)

diff --git a/tools/recommend.py b/tools/recommend.py
index 10c475c6..7564894f 100644
--- a/tools/recommend.py
+++ b/tools/recommend.py
@@ -12,17 +12,16 @@ import implicit
 from hypy_utils.logging_utils import setup_logger
 
 BASE_URL = "https://aquadx.net/aqua/api/v2/game"
-GAME = "mai2"
 BOT_SECRET = "hunter2"
 
 log = setup_logger()
 
 
-if __name__ == '__main__':
+def main(game: str):
     # Load the CSV data
     log.info("Loading data...")
     # data = pd.read_csv("data.csv")
-    resp = requests.get(f"{BASE_URL}/{GAME}/recommender-fetch", params={"botSecret": BOT_SECRET})
+    resp = requests.get(f"{BASE_URL}/{game}/recommender-fetch", params={"botSecret": BOT_SECRET})
     assert resp.status_code == 200, f"Failed to fetch data: {resp.status_code} {resp.text}"
     data = pd.read_csv(StringIO(resp.text))
 
@@ -48,4 +47,13 @@ if __name__ == '__main__':
     # Save recommendations to a file
     log.info("Saving recommendations...")
     # Path("recommendations.json").write_text(json.dumps(recommendations))
-    resp = requests.post(f"{BASE_URL}/{GAME}/recommender-update", params={"botSecret": BOT_SECRET}, json=recommendations)
+    resp = requests.post(f"{BASE_URL}/{game}/recommender-update", params={"botSecret": BOT_SECRET}, json=recommendations)
+    if resp.status_code != 200:
+        log.error(f"Failed to update recommendations: {resp.status_code} {resp.text}")
+
+    log.info("Done!")
+
+
+if __name__ == '__main__':
+    main("mai2")
+    main("chu3")