Upload app.py

app.py

@@ -2,25 +2,31 @@ import os
 from fastapi import FastAPI, Header, HTTPException, Body
 from transformers import AutoTokenizer
 
-# 從 Hugging Face Space 的 Secret 環境變數取得 API_KEY
-EXPECTED_API_KEY = os.environ.get("apikey", "")
+# 從環境變數讀取 secret
+EXPECTED_API_KEY = os.environ.get("apikey")
 
-# 初始化 CKIP 模型 (使用 Hugging Face 提供的預訓練模型)
+# 初始化 tokenizer
 tokenizer = AutoTokenizer.from_pretrained("ckiplab/bert-base-chinese-ws")
 
 app = FastAPI(title="CKIP Word Segmentation API")
 
+# ✅ GET 方法：健康檢查
 @app.get("/")
-def root():
-    return {"message": "CKIP Tokenizer API is running!"}
+def health_check():
+    return {"status": "ok", "message": "API is running"}
 
+# ✅ POST 方法：斷詞
 @app.post("/tokenize")
-async def tokenize(input_text: TextInput, x_api_key: str = Header(default=None)):
-    # 驗證 API Key
+async def tokenize(
+    text: str = Body(..., embed=True),
+    x_api_key: str = Header(None)
+):
+    # 驗證 API Key 設定是否存在
     if not EXPECTED_API_KEY:
         raise HTTPException(status_code=500, detail="Server missing API_KEY config")
+    # 驗證 API Key 值是否正確
     if x_api_key != EXPECTED_API_KEY:
-        raise HTTPException(status_code=401, detail="Unauthorized: Invalid API Key")
+        raise HTTPException(status_code=401, detail="Invalid API Key")
 
     tokens = tokenizer.tokenize(text)
     return {"tokens": tokens}