Update main.py

main.py

@@ -145,28 +145,34 @@ def system_status():
     }
 
 @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
+async def play_noitu(word: str = Query(..., description="Từ 2 tiếng để nối")):
+    word = word.lower().strip()
+    
+    # Kiểm tra input phải là 2 tiếng
+    if len(word.split()) != 2:
+        raise HTTPException(status_code=400, detail="Vui lòng nhập từ có đúng 2 tiếng (ví dụ: 'học sinh').")
+
+    # Kiểm tra từ có trong từ điển không
     if not noitu_app.is_valid_word(word):
         return {
-            "ok": False,
-            "error": "Từ này không có trong từ điển tiếng Việt."
+            "ok": False, 
+            "error": "Từ này không có trong từ điển tiếng Việt hoặc không phải từ ghép hợp lệ."
         }
     
-    # 2. Tìm từ nối tiếp
-    bot_word = noitu_app.get_next_word(word)
+    # Bot tìm từ nối
+    bot_reply = noitu_app.get_next_word(word)
     
-    if not bot_word:
+    if not bot_reply:
         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."
+            "message": "Bạn thắng rồi! Mình không tìm được từ nào nữa."
         }
     
     return {
         "ok": True,
-        "bot_word": bot_word,
-        "input_word": word
+        "bot_word": bot_reply,
+        "input": word
     }