Update main.py

main.py

@@ -30,6 +30,7 @@ from app.gs_code import GSCodeApp
 from app.gm_crate import GmailLogic
 from app.gs_daily import GSDailyApp
 from app.invite_logic import DiscordInviteLogic
+from app.noi_tu import NoiTuApp
 
 xs_app = XoSoApp()
 lq_app = ClLqApp()
@@ -45,6 +46,7 @@ ios_app = IOSAppFetcher()
 gs_code_app = GSCodeApp()
 invite_app = DiscordInviteLogic()
 ht_random_app = HentaiRandomApp()
+noitu_app = NoiTuApp()
 
 API_KEY = os.getenv("API_KEY")
 START_TIME = time.time()
@@ -88,13 +90,13 @@ async def lifespan(app: FastAPI):
     await asyncio.gather(
         invite_app.start(), lq_app.start(), gs_app.start(),
         gs_code_app.start(), hentai_app.start(), ht_random_app.start(),
-        bypass_app.start(), ios_app.start(), xs_app.start(), tik_direct.start(), yt_cdn.start()
+        bypass_app.start(), ios_app.start(), xs_app.start(), tik_direct.start(), yt_cdn.start(), noitu_app.start()
     )
     yield
     await asyncio.gather(
         invite_app.stop(), lq_app.stop(), gs_app.stop(),
         gs_code_app.stop(), hentai_app.stop(), ht_random_app.stop(),
-        bypass_app.stop(), ios_app.stop(), xs_app.stop(), tik_direct.stop(), yt_cdn.stop()
+        bypass_app.stop(), ios_app.stop(), xs_app.stop(), tik_direct.stop(), yt_cdn.stop(), noitu_app.stop()
     )
 
 BASE_DIR = os.path.dirname(os.path.abspath(__file__))
@@ -142,45 +144,32 @@ def system_status():
         "total_ram_mb": round(memory.total / 1024 / 1024, 2)
     }
 
-@app.post("/api/v1/noitu/2-word")
-async def noi_tu_2_am_tiet(request: NoiTuRequest):
-    tudien = await load_noi_tu_words()
-    if not tudien:
-        raise HTTPException(status_code=503, detail="Không load được từ điển, thử lại sau")
-    
-    tu_truoc = request.tu_truoc.strip().lower()
-    if len(tu_truoc.split()) != 2:
-        raise HTTPException(status_code=400, detail="Từ trước phải đúng 2 âm tiết (cách nhau bằng space)")
-    
-    # Lấy âm tiết cuối
-    am_cuoi = tu_truoc.split()[-1]
-    
-    used_set = set(t.lower() for t in (request.used or []))
-    
-    candidates = []
-    for tu in tudien:
-        if tu not in used_set:
-            am_dau = tu.split()[0]
-            if am_dau == am_cuoi:
-                candidates.append(tu)
-    
-    if not candidates:
+@app.get("/api/v1/noitu/2-word")
+async def play_noi_tu(word: str = Query(..., description="Từ để nối (ví dụ: 'học sinh')")):
+    # 1. Kiểm tra từ đầu vào có hợp lệ trong từ điển không
+    if not noitu_app.is_valid_word(word):
         return {
             "ok": False,
-            "message": "Hết từ hợp lệ rồi! Bạn thắng hoặc... hết vốn từ :("
+            "error": "Từ này không có trong từ điển tiếng Việt."
         }
     
-    # Chọn random 1 từ (hoặc trả hết nếu muốn client random)
-    import random
-    tu_moi = random.choice(candidates)
+    # 2. Tìm từ nối tiếp
+    bot_word = noitu_app.get_next_word(word)
+    
+    if not bot_word:
+        return {
+            "ok": True,
+            "bot_word": None,
+            "message": "Bạn thắng rồi! Mình không tìm được từ nào để nối tiếp."
+        }
     
     return {
         "ok": True,
-        "tu_moi": tu_moi,
-        "am_tiet_noi": am_cuoi,
-        "so_tu_con_lai_uoc_tinh": len(candidates)  # bonus info
+        "bot_word": bot_word,
+        "input_word": word
     }
 
+
 @app.get("/api/v1/invites/discord")
 async def get_new_invite():
     res = await invite_app.get_invite()